fix: failed tests

janhq · Jun 16, 2024 · 0295d94 · 0295d94
1 parent 28d59dd
commit 0295d94
Show file tree

Hide file tree

Showing 8 changed files with 27 additions and 2 deletions.
diff --git a/cortex-js/package.json b/cortex-js/package.json
@@ -114,7 +114,8 @@
     "coverageDirectory": "../coverage",
     "testEnvironment": "node",
     "moduleNameMapper": {
-      "@/(.*)$": "<rootDir>/$1"
+      "@/(.*)$": "<rootDir>/$1",
+      "@commanders/(.*)$": "<rootDir>/../src/infrastructure/commanders/$1"
     }
   }
 }
diff --git a/cortex-js/src/download-manager/download-manager.service.spec.ts b/cortex-js/src/download-manager/download-manager.service.spec.ts
@@ -1,11 +1,14 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { DownloadManagerService } from './download-manager.service';
+import { HttpModule } from '@nestjs/axios';
+import { EventEmitterModule } from '@nestjs/event-emitter';
 
 describe('DownloadManagerService', () => {
   let service: DownloadManagerService;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
+      imports: [HttpModule, EventEmitterModule.forRoot()],
       providers: [DownloadManagerService],
     }).compile();
 

diff --git a/cortex-js/src/infrastructure/controllers/chat.controller.spec.ts b/cortex-js/src/infrastructure/controllers/chat.controller.spec.ts
@@ -5,6 +5,8 @@ import { DatabaseModule } from '../database/database.module';
 import { ExtensionModule } from '../repositories/extensions/extension.module';
 import { ModelRepositoryModule } from '../repositories/models/model.module';
 import { HttpModule } from '@nestjs/axios';
+import { DownloadManagerModule } from '@/download-manager/download-manager.module';
+import { EventEmitterModule } from '@nestjs/event-emitter';
 
 describe('ChatController', () => {
   let controller: ChatController;
@@ -16,6 +18,8 @@ describe('ChatController', () => {
         ExtensionModule,
         ModelRepositoryModule,
         HttpModule,
+        DownloadManagerModule,
+        EventEmitterModule.forRoot(),
       ],
       controllers: [ChatController],
       providers: [ChatUsecases],

diff --git a/cortex-js/src/infrastructure/controllers/embeddings.controller.spec.ts b/cortex-js/src/infrastructure/controllers/embeddings.controller.spec.ts
@@ -5,6 +5,8 @@ import { DatabaseModule } from '../database/database.module';
 import { ModelRepositoryModule } from '../repositories/models/model.module';
 import { ExtensionModule } from '../repositories/extensions/extension.module';
 import { HttpModule } from '@nestjs/axios';
+import { DownloadManagerModule } from '@/download-manager/download-manager.module';
+import { EventEmitterModule } from '@nestjs/event-emitter';
 
 describe('EmbeddingsController', () => {
   let controller: EmbeddingsController;
@@ -16,6 +18,8 @@ describe('EmbeddingsController', () => {
         ModelRepositoryModule,
         ExtensionModule,
         HttpModule,
+        DownloadManagerModule,
+        EventEmitterModule.forRoot(),
       ],
       controllers: [EmbeddingsController],
       providers: [ChatUsecases],

diff --git a/cortex-js/src/infrastructure/controllers/models.controller.spec.ts b/cortex-js/src/infrastructure/controllers/models.controller.spec.ts
@@ -7,6 +7,8 @@ import { FileManagerModule } from '@/infrastructure/services/file-manager/file-m
 import { HttpModule } from '@nestjs/axios';
 import { CortexUsecases } from '@/usecases/cortex/cortex.usecases';
 import { ModelRepositoryModule } from '../repositories/models/model.module';
+import { DownloadManagerModule } from '@/download-manager/download-manager.module';
+import { EventEmitterModule } from '@nestjs/event-emitter';
 
 describe('ModelsController', () => {
   let controller: ModelsController;
@@ -19,6 +21,8 @@ describe('ModelsController', () => {
         FileManagerModule,
         HttpModule,
         ModelRepositoryModule,
+        DownloadManagerModule,
+        EventEmitterModule.forRoot(),
       ],
       controllers: [ModelsController],
       providers: [ModelsUsecases, CortexUsecases],

diff --git a/cortex-js/src/usecases/chat/chat.usecases.spec.ts b/cortex-js/src/usecases/chat/chat.usecases.spec.ts
@@ -4,6 +4,8 @@ import { DatabaseModule } from '@/infrastructure/database/database.module';
 import { ExtensionModule } from '@/infrastructure/repositories/extensions/extension.module';
 import { ModelRepositoryModule } from '@/infrastructure/repositories/models/model.module';
 import { HttpModule } from '@nestjs/axios';
+import { DownloadManagerModule } from '@/download-manager/download-manager.module';
+import { EventEmitterModule } from '@nestjs/event-emitter';
 
 describe('ChatService', () => {
   let service: ChatUsecases;
@@ -15,6 +17,8 @@ describe('ChatService', () => {
         ExtensionModule,
         ModelRepositoryModule,
         HttpModule,
+        DownloadManagerModule,
+        EventEmitterModule.forRoot(),
       ],
       providers: [ChatUsecases],
       exports: [ChatUsecases],

diff --git a/cortex-js/src/usecases/models/models.usecases.spec.ts b/cortex-js/src/usecases/models/models.usecases.spec.ts
@@ -6,6 +6,8 @@ import { ExtensionModule } from '@/infrastructure/repositories/extensions/extens
 import { FileManagerModule } from '@/infrastructure/services/file-manager/file-manager.module';
 import { HttpModule } from '@nestjs/axios';
 import { ModelRepositoryModule } from '@/infrastructure/repositories/models/model.module';
+import { DownloadManagerModule } from '@/download-manager/download-manager.module';
+import { EventEmitterModule } from '@nestjs/event-emitter';
 
 describe('ModelsService', () => {
   let service: ModelsUsecases;
@@ -19,6 +21,8 @@ describe('ModelsService', () => {
         FileManagerModule,
         HttpModule,
         ModelRepositoryModule,
+        DownloadManagerModule,
+        EventEmitterModule.forRoot(),
       ],
       providers: [ModelsUsecases],
       exports: [ModelsUsecases],

diff --git a/cortex-js/test/jest-e2e.json b/cortex-js/test/jest-e2e.json
@@ -7,6 +7,7 @@
     "^.+\\.(t|j)s$": "ts-jest"
   },
   "moduleNameMapper": {
-    "@/(.*)$": "<rootDir>/../src/$1"
+    "@/(.*)$": "<rootDir>/../src/$1",
+    "@commanders/(.*)$": "<rootDir>/../src/infrastructure/commanders/$1"
   }
 }