node.js - runTransaction 中的模拟事务

Question

我想在runTransaction函数中模拟代码。

示例代码：

await admin.firestore().runTransaction(async transaction => {
  const hubDocument = admin.firestore().collection("Acme").doc('4');
  const hubData = (await transaction.get(hubDocument)).data();
  newData = {
    ...hubData,
    someAttribute: 'some new value'
  };
  transaction.update(hubDocument, newData);
})

我想模拟transaction，检查它是否用正确的数据 etc.pp 调用。

我设法模拟firestore()但不知道如何模拟transaction参数。

Answer 1

我没有对此进行测试，但我认为这样的事情应该可以解决问题：

import { Transaction } from '@google-cloud/firestore';

const origTransactionGet = Transaction.prototype.get
Transaction.prototype.get = function () {
   console.log(arguments, "< Intercepted Transaction.prototype.get")
   return origTransactionGet.apply(this, arguments)
}

// your code
await admin.firestore().runTransaction(async transaction => {
  const hubDocument = admin.firestore().collection("Acme").doc('4');
  const hubData = (await transaction.get(hubDocument)).data();
  newData = {
    ...hubData,
    someAttribute: 'some new value'
  };
  transaction.update(hubDocument, newData);
})

正如@FiodorovAndrei 评论的那样，如果您使用另一种可能更舒服的方法，那就是jest只使用firestore-jest-mock来模拟 Firestore 功能。