|
@@ -24,6 +24,7 @@ import {
|
|
ShowCollectionsReq,
|
|
ShowCollectionsReq,
|
|
ShowPartitionsReq,
|
|
ShowPartitionsReq,
|
|
} from '@zilliz/milvus2-sdk-node/dist/milvus/types';
|
|
} from '@zilliz/milvus2-sdk-node/dist/milvus/types';
|
|
|
|
+import { DeleteEntitiesReq } from '@zilliz/milvus2-sdk-node/dist/milvus/types/Data';
|
|
import { QueryDto } from '../../../collections/dto';
|
|
import { QueryDto } from '../../../collections/dto';
|
|
import {
|
|
import {
|
|
CodeEnum,
|
|
CodeEnum,
|
|
@@ -335,6 +336,16 @@ const mockMilvusClient = jest.fn().mockImplementation((address: string) => {
|
|
data: param,
|
|
data: param,
|
|
};
|
|
};
|
|
},
|
|
},
|
|
|
|
+ deleteEntities: (param: DeleteEntitiesReq) => {
|
|
|
|
+ const { collection_name } = param;
|
|
|
|
+ if (!collection_name) {
|
|
|
|
+ return { status: mockStatusInfo(CodeEnum.error, ERR_NO_COLLECTION) };
|
|
|
|
+ }
|
|
|
|
+ return {
|
|
|
|
+ status: mockStatusInfo(CodeEnum.success),
|
|
|
|
+ data: param,
|
|
|
|
+ };
|
|
|
|
+ },
|
|
},
|
|
},
|
|
};
|
|
};
|
|
});
|
|
});
|