fix server test

server/src/__tests__/__mocks__/consts.ts

@@ -171,3 +171,46 @@ export const mockGetPartitionsInfoData = [
     rowCount: 7,
   },
 ];
+
+export const mockGetReplicasData = [
+  {
+    partition_ids: ['1'],
+    shard_replicas: [
+      {
+        node_ids: ['2'],
+        leaderID: '2',
+        leader_addr: '172.20.0.11:21123',
+        dm_channel_name: 'by-dev-rootcoord-dml_2_438207169419481300v0',
+      },
+      {
+        node_ids: ['2'],
+        leaderID: '2',
+        leader_addr: '172.20.0.11:21123',
+        dm_channel_name: 'by-dev-rootcoord-dml_3_438207169419481300v1',
+      },
+    ],
+    node_ids: ['2'],
+    replicaID: '438207185122754599',
+    collectionID: '438207169419481300',
+  },
+  {
+    partition_ids: [],
+    shard_replicas: [
+      {
+        node_ids: ['4'],
+        leaderID: '4',
+        leader_addr: '172.20.0.10:21123',
+        dm_channel_name: 'by-dev-rootcoord-dml_2_438207169419481300v0',
+      },
+      {
+        node_ids: ['4'],
+        leaderID: '4',
+        leader_addr: '172.20.0.10:21123',
+        dm_channel_name: 'by-dev-rootcoord-dml_3_438207169419481300v1',
+      },
+    ],
+    node_ids: ['4'],
+    replicaID: '438207185122754598',
+    collectionID: '438207169419481300',
+  },
+];

server/src/__tests__/collections/collections.service.test.ts

@@ -14,6 +14,7 @@ import {
   mockGetAllCollectionsData,
   mockLoadedCollections,
   mockLoadedCollectionsData,
+  mockGetReplicasData,
 } from '../__mocks__/consts';
 import { MilvusClient } from '@zilliz/milvus2-sdk-node/dist/milvus';
 
@@ -39,6 +40,12 @@ describe('Test collections service', () => {
       insightCacheForTest
     );
     service = new CollectionsService(milvusService);
+    // mock getReplicas
+    service.getReplicas = async () => {
+      return new Promise<any>(resolve => {
+        resolve(mockGetReplicasData);
+      });
+    };
   });
 
   afterAll(() => {
@@ -174,7 +181,7 @@ describe('Test collections service', () => {
   test('test importSample method', async () => {
     const mockParam = {
       collection_name: 'c1',
-      size: 2
+      size: 2,
     };
     const res = await service.importSample(mockParam);
     expect(res.data.fields_data.length).toEqual(2);
@@ -182,7 +189,7 @@ describe('Test collections service', () => {
     try {
       await service.importSample({
         collection_name: '',
-        size: 20
+        size: 20,
       });
     } catch (err) {
       expect(err).toBe(ERR_NO_COLLECTION);

server/src/__tests__/crons/crons.service.test.ts

@@ -4,7 +4,11 @@ import { CollectionsService } from '../../collections/collections.service';
 import { CronsService, SchedulerRegistry } from '../../crons/crons.service';
 import { MilvusService } from '../../milvus/milvus.service';
 import { WS_EVENTS, WS_EVENTS_TYPE } from '../../utils/Const';
-import { insightCacheForTest, mockAddress } from '../__mocks__/consts';
+import {
+  insightCacheForTest,
+  mockAddress,
+  mockGetReplicasData,
+} from '../__mocks__/consts';
 import { MilvusClient } from '@zilliz/milvus2-sdk-node/dist/milvus';
 
 // mock Milvus client
@@ -51,6 +55,11 @@ describe('test crons service', () => {
     );
 
     collectionService = new CollectionsService(milvusService);
+    collectionService.getReplicas = async () => {
+      return new Promise<any>(resolve => {
+        resolve(mockGetReplicasData);
+      });
+    };
 
     schedulerRegistry = new SchedulerRegistry([]);
     cronsService = new CronsService(collectionService, schedulerRegistry);
@@ -153,6 +162,12 @@ describe('test crons service', () => {
     const newCollectionService = new CollectionsService(milvusService);
     const newSchedulerRegistry = new SchedulerRegistry([]);
 
+    newCollectionService.getReplicas = async () => {
+      return new Promise<any>(resolve => {
+        resolve(mockGetReplicasData);
+      });
+    };
+
     const newCronsService = new CronsService(
       newCollectionService,
       newSchedulerRegistry