Skip to content

Commit a4b40fd

Browse files
committed
test(workers): add ToolsService mock to WorkersService tests
1 parent 02c2497 commit a4b40fd

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

core-api/src/modules/workers/workers.service.spec.ts

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import { ApiKeysService } from '../apikeys/apikeys.service';
77
import { Asset } from '../assets/entities/assets.entity';
88
import { JobsRegistryService } from '../jobs-registry/jobs-registry.service';
99
import { WorkspaceTool } from '../tools/entities/workspace_tools.entity';
10+
import { ToolsService } from '../tools/tools.service';
1011
import { WorkerInstance } from './entities/worker.entity';
1112
import { WorkersService } from './workers.service';
1213

@@ -18,6 +19,7 @@ describe('WorkersService', () => {
1819
let mockJobsRegistryService: Partial<JobsRegistryService>;
1920
let mockApiKeysService: Partial<ApiKeysService>;
2021
let mockConfigService: Partial<ConfigService>;
22+
let mockToolsService: Partial<ToolsService>;
2123

2224
beforeEach(async () => {
2325
mockWorkerInstanceRepository = {
@@ -69,6 +71,10 @@ describe('WorkersService', () => {
6971
get: jest.fn(),
7072
} as any;
7173

74+
mockToolsService = {
75+
getBuiltInTools: jest.fn().mockResolvedValue({ data: [] }),
76+
} as any;
77+
7278
const module: TestingModule = await Test.createTestingModule({
7379
providers: [
7480
WorkersService,
@@ -96,6 +102,10 @@ describe('WorkersService', () => {
96102
provide: ConfigService,
97103
useValue: mockConfigService,
98104
},
105+
{
106+
provide: ToolsService,
107+
useValue: mockToolsService,
108+
},
99109
],
100110
}).compile();
101111

0 commit comments

Comments
 (0)