Skip to content

Commit 009d6e7

Browse files
committed
feat: updated tests
1 parent 8b52137 commit 009d6e7

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

src/waiting-time/waiting-time.service.spec.ts

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import { RewardsService } from 'events/rewards/rewards.service';
1414
import { SECONDS_PER_SLOT, SLOTS_PER_EPOCH } from 'common/genesis-time';
1515

1616
import { WaitingTimeCalculationType } from './waiting-time.types';
17+
import { SimpleFallbackJsonRpcBatchProvider } from '@lido-nestjs/execution';
1718

1819
jest.mock('common/config', () => ({}));
1920

@@ -24,6 +25,7 @@ describe('WaitingTimeService', () => {
2425
let contractConfig: ContractConfigStorageService;
2526
let genesisTimeService: GenesisTimeService;
2627
let validatorsStorage: ValidatorsStorageService;
28+
let rpcBatchProvider: SimpleFallbackJsonRpcBatchProvider;
2729

2830
// constants
2931
const genesisTime = 1606824023;
@@ -108,6 +110,12 @@ describe('WaitingTimeService', () => {
108110
getFrameBalances: jest.fn(),
109111
},
110112
},
113+
{
114+
provide: SimpleFallbackJsonRpcBatchProvider,
115+
useValue: {
116+
getBlock: jest.fn(),
117+
},
118+
},
111119
{
112120
provide: GenesisTimeService,
113121
useValue: {
@@ -130,6 +138,7 @@ describe('WaitingTimeService', () => {
130138
contractConfig = moduleRef.get<ContractConfigStorageService>(ContractConfigStorageService);
131139
genesisTimeService = moduleRef.get<GenesisTimeService>(GenesisTimeService);
132140
validatorsStorage = moduleRef.get<ValidatorsStorageService>(ValidatorsStorageService);
141+
rpcBatchProvider = moduleRef.get<SimpleFallbackJsonRpcBatchProvider>(SimpleFallbackJsonRpcBatchProvider);
133142

134143
// mocks
135144
jest.spyOn(contractConfig, 'getInitialEpoch').mockReturnValue(initialEpoch);
@@ -145,6 +154,8 @@ describe('WaitingTimeService', () => {
145154
jest.spyOn(validatorsStorage, 'getActiveValidatorsCount').mockReturnValue(10000);
146155
jest.spyOn(validatorsStorage, 'getFrameBalances').mockReturnValue({});
147156
jest.spyOn(service, 'getFrameIsBunker').mockReturnValue(null);
157+
// needed for mock only block number
158+
jest.spyOn(rpcBatchProvider, 'getBlock').mockResolvedValue({ number: 21367114 } as any);
148159
});
149160

150161
afterEach(async () => {

0 commit comments

Comments
 (0)