Merge pull request #181 from sutcalag/main

enable server side test case

server/src/app.controller.spec.ts

@@ -1,6 +1,20 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { AppController } from './app.controller';
-import { AppService } from './app.service';
+import { AuthService } from './auth/auth.service';
+
+const testUser = {
+  userId: 1,
+  username: 'milvus',
+  password: 'milvus-admin',
+}
+class AuthServiceMock {
+  login = () => testUser;
+}
+
+const AuthServiceProvider = {
+  provide: AuthService,
+  useClass: AuthServiceMock,
+};
 
 describe('AppController', () => {
   let appController: AppController;
@@ -8,15 +22,20 @@ describe('AppController', () => {
   beforeEach(async () => {
     const app: TestingModule = await Test.createTestingModule({
       controllers: [AppController],
-      providers: [AppService],
+      providers: [AuthServiceProvider],
     }).compile();
 
     appController = app.get<AppController>(AppController);
   });
 
   describe('root', () => {
-    it('should return "Hello World!"', () => {
-      expect(appController.getHello()).toBe('Hello World!');
+    it('should defined', () => {
+      expect(appController).toBeDefined();
+    });
+
+    it('should return result', async () => {
+      const data = await appController.login(1);
+      expect(data).toBe(testUser);
     });
   });
 });

server/src/collections/collections.controller.spec.ts

@@ -1,14 +1,15 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { CollectionsController } from './collections.controller';
+import { CacheModule } from '@nestjs/common';
+import { CollectionsModule } from './collections.module';
 
 describe('CollectionsController', () => {
   let controller: CollectionsController;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
-      controllers: [CollectionsController],
+      imports: [CacheModule.register({}), CollectionsModule],
     }).compile();
-
     controller = module.get<CollectionsController>(CollectionsController);
   });
 

server/src/collections/collections.service.spec.ts

@@ -1,12 +1,13 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { CollectionsService } from './collections.service';
+import { MilvusServiceProvider } from '../milvus/milvus.service.mock';
 
 describe('CollectionsService', () => {
   let service: CollectionsService;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
-      providers: [CollectionsService],
+      providers: [CollectionsService, MilvusServiceProvider],
     }).compile();
 
     service = module.get<CollectionsService>(CollectionsService);

server/src/milvus/milvus.controller.spec.ts

@@ -1,5 +1,6 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { MilvusController } from './milvus.controller';
+import { MilvusService } from './milvus.service';
 
 describe('MilvusController', () => {
   let controller: MilvusController;
@@ -7,6 +8,7 @@ describe('MilvusController', () => {
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
       controllers: [MilvusController],
+      providers: [MilvusService]
     }).compile();
 
     controller = module.get<MilvusController>(MilvusController);

server/src/milvus/milvus.controller.ts

@@ -8,14 +8,14 @@ import {
   UsePipes,
 } from '@nestjs/common';
 import { ApiTags } from '@nestjs/swagger';
-import { ValidationPipe } from 'src/pipe/validation.pipe';
+import { ValidationPipe } from '../pipe/validation.pipe';
 import { CheckMilvus, ConnectMilvus, Flush } from './dto';
 import { MilvusService } from './milvus.service';
 
 @ApiTags('milvus')
 @Controller('milvus')
 export class MilvusController {
-  constructor(private milvusService: MilvusService) {}
+  constructor(private milvusService: MilvusService) { }
 
   @Post('connect')
   @UsePipes(new ValidationPipe())

server/src/milvus/milvus.service.mock.ts

@@ -0,0 +1,13 @@
+import { MilvusService } from './milvus.service';
+
+class MilvusServiceMock {
+  milvusClientGetter() {
+    return {};
+  }
+}
+
+export const MilvusServiceProvider = {
+  provide: MilvusService,
+  useClass: MilvusServiceMock,
+};
+

server/src/partitions/partitions.controller.spec.ts

@@ -1,5 +1,7 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { PartitionsController } from './partitions.controller';
+import { PartitionsService } from './partitions.service';
+import { MilvusServiceProvider } from '../milvus/milvus.service.mock';
 
 describe('PartitionsController', () => {
   let controller: PartitionsController;
@@ -7,6 +9,7 @@ describe('PartitionsController', () => {
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
       controllers: [PartitionsController],
+      providers: [PartitionsService, MilvusServiceProvider],
     }).compile();
 
     controller = module.get<PartitionsController>(PartitionsController);

server/src/partitions/partitions.service.spec.ts

@@ -1,12 +1,13 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { PartitionsService } from './partitions.service';
+import { MilvusServiceProvider } from '../milvus/milvus.service.mock';
 
 describe('PartitionsService', () => {
   let service: PartitionsService;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
-      providers: [PartitionsService],
+      providers: [PartitionsService, MilvusServiceProvider],
     }).compile();
 
     service = module.get<PartitionsService>(PartitionsService);

server/src/partitions/partitions.service.ts

@@ -8,13 +8,13 @@ import {
   ReleasePartitionsReq,
   ShowPartitionsReq,
 } from '@zilliz/milvus2-sdk-node/dist/milvus/types'; // todo: need improve like export types in root file.
-import { throwErrorFromSDK } from 'src/utils/Error';
+import { throwErrorFromSDK } from '../utils/Error';
 import { findKeyValue } from '../utils/Helper';
 import { ROW_COUNT } from '../utils/Const';
 
 @Injectable()
 export class PartitionsService {
-  constructor(private milvusService: MilvusService) {}
+  constructor(private milvusService: MilvusService) { }
 
   get milvusClient() {
     return this.milvusService.milvusClientGetter;

server/src/schema/schema.controller.spec.ts

@@ -1,5 +1,7 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { SchemaController } from './schema.controller';
+import { SchemaService } from './schema.service';
+import { MilvusServiceProvider } from '../milvus/milvus.service.mock';
 
 describe('SchemaController', () => {
   let controller: SchemaController;
@@ -7,6 +9,7 @@ describe('SchemaController', () => {
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
       controllers: [SchemaController],
+      providers: [SchemaService, MilvusServiceProvider],
     }).compile();
 
     controller = module.get<SchemaController>(SchemaController);

server/src/schema/schema.service.spec.ts

@@ -1,12 +1,13 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { SchemaService } from './schema.service';
+import { MilvusServiceProvider } from '../milvus/milvus.service.mock';
 
 describe('SchemaService', () => {
   let service: SchemaService;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
-      providers: [SchemaService],
+      providers: [SchemaService, MilvusServiceProvider],
     }).compile();
 
     service = module.get<SchemaService>(SchemaService);

server/src/schema/schema.service.ts

@@ -6,12 +6,12 @@ import {
   GetIndexBuildProgressReq,
   GetIndexStateReq,
 } from '@zilliz/milvus2-sdk-node/dist/milvus/types';
-import { throwErrorFromSDK } from 'src/utils/Error';
+import { throwErrorFromSDK } from '../utils/Error';
 import { MilvusService } from '../milvus/milvus.service';
 
 @Injectable()
 export class SchemaService {
-  constructor(private milvusService: MilvusService) {}
+  constructor(private milvusService: MilvusService) { }
 
   get milvusClient() {
     return this.milvusService.milvusClientGetter;

server/src/users/users.service.spec.ts

@@ -26,7 +26,7 @@ describe('UsersService', () => {
     expect(res).toEqual({
       userId: 1,
       username: 'milvus',
-      password: 'milvus-em',
+      password: 'milvus-admin',
     });
   });
 });