fix api service kyve test

subquery · Apr 17, 2024 · 3ba4fa1 · 3ba4fa1
1 parent f37102a
commit 3ba4fa1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/packages/node/src/indexer/api.service.test.ts b/packages/node/src/indexer/api.service.test.ts
@@ -17,7 +17,7 @@ import {
   NodeConfig,
 } from '@subql/node-core';
 import { GraphQLSchema } from 'graphql';
-import { CosmosNodeConfig } from '../configure/NodeConfig';
+import Pino from 'pino';
 import { SubqueryProject } from '../configure/SubqueryProject';
 import { LazyBlockContent } from '../utils/cosmos';
 import { ApiService } from './api.service';
@@ -99,9 +99,7 @@ describe('ApiService', () => {
     });
 
     it('Able to fetch with cached promises and remove cached bundle files', async () => {
-      jest
-        .spyOn(apiService as any, 'retryFetch')
-        .mockRejectedValue('Disabled rpc for kyve testing');
+      const rpcFetchSpy = jest.spyOn(apiService as any, 'retryFetch');
 
       const heights_1 = [150, 300, 1, 301, 450, 550];
       const heights_2 = [498, 600, 801, 1100];
@@ -117,6 +115,8 @@ describe('ApiService', () => {
       });
 
       const files = await fs.promises.readdir(tmpPath);
+
+      expect(rpcFetchSpy).toHaveBeenCalledTimes(0);
       expect(files).not.toContain('bundle_0.json');
       expect(files).not.toContain('bundle_1.json');
     });

diff --git a/packages/node/src/utils/kyve/kyve.spec.ts b/packages/node/src/utils/kyve/kyve.spec.ts
@@ -308,7 +308,7 @@ describe('KyveApi', () => {
     const files = await fs.promises.readdir(tmpPath);
     expect(files.length).toBe(0);
   });
-  it('ensure to remove logic', () => {
+  it('ensure to remove logic', async () => {
     const cachedBundleDetails = [
       { id: '0', from_key: '1', to_key: '150' },
       { id: '1', from_key: '151', to_key: '300' },
@@ -320,7 +320,7 @@ describe('KyveApi', () => {
     const height = 650;
     const bufferSize = 300;
 
-    const toRemoveBundles = (kyveApi as any).getToRemoveBundles(
+    const toRemoveBundles = await (kyveApi as any).getToRemoveBundles(
       cachedBundleDetails,
       height,
       bufferSize,