fix: tests

Greenstand · Jul 7, 2023 · d2543a2 · d2543a2
1 parent 51ce44a
commit d2543a2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/server/models/Wallet.spec.js b/server/models/Wallet.spec.js
@@ -137,10 +137,14 @@ describe('Wallet Model', () => {
     });
 
     expect(result).eql([{ id: walletId, wallet }]);
-    expect(walletRepositoryStub.getAllWallets).calledOnceWithExactly(walletId, {
-      limit: 1,
-      offset: 1,
-    });
+    expect(walletRepositoryStub.getAllWallets).calledOnceWithExactly(
+      walletId,
+      {
+        limit: 1,
+        offset: 1,
+      },
+      '',
+    );
   });
 
   describe('hasControlOver function', async () => {

diff --git a/server/repositories/WalletRepository.spec.js b/server/repositories/WalletRepository.spec.js
@@ -80,4 +80,17 @@ describe('WalletRepository', () => {
     const entity = await walletRepository.getAllWallets(uuid.v4());
     expect(entity).to.be.a('array');
   });
+
+  it('getAllWalletsCount', async () => {
+    tracker.uninstall();
+    tracker.install();
+    tracker.on('query', (query) => {
+      expect(query.sql).match(
+        /select.*wallet.*where.*actor_wallet_id.*request_type.*/,
+      );
+      query.response([{ id: 1 }]);
+    });
+    const entity = await walletRepository.getAllWallets(uuid.v4());
+    expect(entity).to.be.a('array');
+  });
 });