test: fix repository tests

dashpay · Mar 17, 2022 · 413bdae · 413bdae
1 parent ac0577d
commit 413bdae
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 10 deletions.
diff --git a/packages/js-drive/test/integration/dataContract/DataContractStoreRepository.spec.js b/packages/js-drive/test/integration/dataContract/DataContractStoreRepository.spec.js
@@ -108,7 +108,7 @@ describe('DataContractStoreRepository', () => {
     });
 
     it('should fetch Data Contract', async () => {
-      await store.getDrive().applyContract(dataContract, false);
+      await store.getDrive().applyContract(dataContract, new Date(), false);
 
       const storedDataContract = await repository.fetch(dataContract.getId());
 
@@ -119,7 +119,7 @@ describe('DataContractStoreRepository', () => {
     it('should fetch Data Contract using transaction', async () => {
       await store.startTransaction();
 
-      await store.getDrive().applyContract(dataContract, true);
+      await store.getDrive().applyContract(dataContract, new Date(), true);
 
       const notFoundDataContract = await repository.fetch(dataContract.getId(), false);
 

diff --git a/packages/js-drive/test/unit/document/DocumentRepository.spec.js b/packages/js-drive/test/unit/document/DocumentRepository.spec.js
@@ -72,10 +72,11 @@ describe('DocumentRepository', () => {
         { useTransaction: false },
       );
 
-      expect(driveMock.createDocument).to.be.calledOnceWithExactly(
-        document,
-        false,
-      );
+      expect(driveMock.createDocument.getCall(0).args).to.have.lengthOf(3);
+
+      expect(driveMock.createDocument.getCall(0).args[0]).to.equals(document);
+      expect(driveMock.createDocument.getCall(0).args[1]).to.be.instanceOf(Date);
+      expect(driveMock.createDocument.getCall(0).args[2]).to.equals(false);
 
       expect(loggerMock.info).to.be.calledOnce();
     });
@@ -100,10 +101,11 @@ describe('DocumentRepository', () => {
         { useTransaction: true },
       );
 
-      expect(driveMock.updateDocument).to.be.calledOnceWithExactly(
-        document,
-        true,
-      );
+      expect(driveMock.updateDocument.getCall(0).args).to.have.lengthOf(3);
+
+      expect(driveMock.updateDocument.getCall(0).args[0]).to.equals(document);
+      expect(driveMock.updateDocument.getCall(0).args[1]).to.be.instanceOf(Date);
+      expect(driveMock.updateDocument.getCall(0).args[2]).to.equals(true);
     });
   });