fix(dapi): fix fetchProofForStateTransitionFactory.js unit test

dashpay · Feb 13, 2025 · 7b40975 · 7b40975
1 parent f719e7d
commit 7b40975
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/packages/dapi/lib/externalApis/drive/fetchProofForStateTransitionFactory.js b/packages/dapi/lib/externalApis/drive/fetchProofForStateTransitionFactory.js
@@ -30,7 +30,7 @@ function fetchProofForStateTransitionFactory(driveClient, dpp) {
 
     const requestV0 = new GetProofsRequestV0();
 
-    let dataContractsCache = {};
+    const dataContractsCache = {};
 
     if (stateTransition.isDocumentStateTransition()) {
       const {

diff --git a/packages/dapi/test/unit/externalApis/drive/fetchProofForStateTransitionFactory.spec.js b/packages/dapi/test/unit/externalApis/drive/fetchProofForStateTransitionFactory.spec.js
@@ -9,7 +9,10 @@ const {
 } = require('@dashevo/dapi-grpc');
 
 const generateRandomIdentifierAsync = require('@dashevo/wasm-dpp/lib/test/utils/generateRandomIdentifierAsync');
-const { StateTransitionTypes } = require('@dashevo/wasm-dpp');
+const {
+  StateTransitionTypes,
+  DocumentTransition,
+} = require('@dashevo/wasm-dpp');
 
 const fetchProofForStateTransitionFactory = require('../../../../lib/externalApis/drive/fetchProofForStateTransitionFactory');
 
@@ -81,14 +84,15 @@ describe('fetchProofForStateTransition', () => {
 
   it('should fetch documents proofs', async function it() {
     stateTransitionFixture.isDocumentStateTransition.returns(true);
-    stateTransitionFixture.getTransitions = this.sinon.stub().returns([
-      {
-        getDataContractId: this.sinon.stub().returns(await generateRandomIdentifierAsync()),
-        getType: this.sinon.stub().returns('niceDocument'),
-        getId: this.sinon.stub().returns(await generateRandomIdentifierAsync()),
-        hasPrefundedBalance: this.sinon.stub().returns(true),
-      },
-    ]);
+    const mockDocumentTransition = Object.create(DocumentTransition.prototype);
+
+    mockDocumentTransition.getDataContractId = this.sinon.stub()
+      .returns(await generateRandomIdentifierAsync());
+    mockDocumentTransition.getType = this.sinon.stub().returns('niceDocument');
+    mockDocumentTransition.getId = this.sinon.stub().returns(await generateRandomIdentifierAsync());
+    mockDocumentTransition.hasPrefundedBalance = this.sinon.stub().returns(true);
+
+    stateTransitionFixture.getTransitions = this.sinon.stub().returns([mockDocumentTransition]);
 
     const result = await fetchProofForStateTransition(stateTransitionFixture);
     expect(result.serializeBinary()).to.deep