fix: mocks

lidofinance · Dec 6, 2023 · 4e73aad · 4e73aad
1 parent c14bc1e
commit 4e73aad
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 11 deletions.
diff --git a/src/contracts/deposit/deposit.service.ts b/src/contracts/deposit/deposit.service.ts
@@ -16,7 +16,6 @@ import {
   VerifiedDepositEventsCacheHeaders,
   VerifiedDepositEventGroup,
 } from './interfaces';
-import { OneAtTime } from 'common/decorators';
 import { RepositoryService } from 'contracts/repository';
 import { CacheService } from 'cache';
 import { BlockTag } from 'provider';
@@ -335,7 +334,6 @@ export class DepositService {
 
     const mergedEvents = cachedEvents.data.concat(freshEvents);
 
-    // TODO: Why we don't cache freshEvents?
     return {
       events: mergedEvents,
       startBlock: cachedEvents.headers.startBlock,

diff --git a/src/guardian/guardian.service.spec.ts b/src/guardian/guardian.service.spec.ts
@@ -179,7 +179,6 @@ describe('GuardianService', () => {
       guardianService.handleNewBlock(),
     ]);
 
-    // expect(getStakingModulesMock).toBeCalledTimes(1);
     expect(getBlockGuardServiceMock).toBeCalledTimes(1);
   });
 });
diff --git a/test/helpers/mockKeysApi.ts b/test/helpers/mockKeysApi.ts
@@ -3,6 +3,7 @@ import { toHexString } from '@chainsafe/ssz';
 
 import { KeysApiService } from '../../src/keys-api/keys-api.service';
 import { NOP_REGISTRY, pk } from './../constants';
+import { RegistryOperator } from 'keys-api/interfaces/RegistryOperator';
 
 export const mockKeysApi = (
   sig: Uint8Array[],
@@ -36,20 +37,34 @@ export const mockKeysApi = (
     operatorIndex: 0,
     used,
     index: 0,
+    moduleAddress: NOP_REGISTRY,
   }));
 
-  jest.spyOn(keysApiService, 'getModulesList').mockImplementation(async () => ({
-    data: [mockedModule],
-    elBlockSnapshot: mockedMeta,
+  const mockedOperators: RegistryOperator[] = [
+    {
+      name: 'Dev team',
+      rewardAddress: '0x6D725DAe055287f913661ee0b79dE6B21F12A459',
+      stakingLimit: 11,
+      stoppedValidators: 0,
+      totalSigningKeys: 10,
+      usedSigningKeys: 10,
+      index: 0,
+      active: true,
+      moduleAddress: '0x595F64Ddc3856a3b5Ff4f4CC1d1fb4B46cFd2bAC',
+    },
+  ];
+
+  jest.spyOn(keysApiService, 'getUnusedKeys').mockImplementation(async () => ({
+    data: mockedKeys,
+    meta: {
+      elBlockSnapshot: mockedMeta,
+    },
   }));
 
   jest
-    .spyOn(keysApiService, 'getUnusedModuleKeys')
+    .spyOn(keysApiService, 'getOperatorListWithModule')
     .mockImplementation(async () => ({
-      data: {
-        keys: mockedKeys,
-        module: mockedModule,
-      },
+      data: [{ operators: mockedOperators, module: mockedModule }],
       meta: {
         elBlockSnapshot: mockedMeta,
       },