runTransaction 中的模擬事務

Question

賞金將在 19 小時后到期。 此問題的答案有資格獲得+100聲望賞金。 Andresch Serj希望引起對這個問題的更多關注。

我想在runTransaction function 中模擬代碼。

示例代碼：

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 功能。