UniqAI/hexabot
1
0
Fork 0
mirror of https://github.com/hexastack/hexabot synced 2025-06-26 18:27:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(api): enhance logic

Browse Source
This commit is contained in:
yassinedorbozgithub 2025-06-19 06:50:28 +01:00
parent 6f0668c9f1
commit 200c256abd
1 changed files with 19 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
api/src/utils/test/utils.ts
View File

@ -195,6 +195,19 @@ const getModels = (models: TModel[]): ModelDefinition[] =>
typeof model === 'string' ? getModel(model, 'Model') : model,
);
const defaultProviders = [
LoggerService,
EventEmitter2,
{
provide: CACHE_MANAGER,
useValue: {
del: jest.fn(),
set: jest.fn(),
get: jest.fn(),
},
},
];
export const buildTestingMocks = async ({
models = [],
imports = [],
@ -204,6 +217,7 @@ export const buildTestingMocks = async ({
...rest
}: buildTestingMocksProps) => {
const extendedProviders = new Set<Provider>();
const dynamicProviders = new Set<Provider>();
const injectionFrom = autoInjectFrom as ToUnionArray<typeof autoInjectFrom>;
if (injectionFrom?.includes('providers')) {
@ -219,6 +233,9 @@ export const buildTestingMocks = async ({
}
providers.forEach((provider) => extendedProviders.add(provider));
[...defaultProviders, ...extendedProviders].forEach((provider) => {
dynamicProviders.add(provider);
});
const module = await Test.createTestingModule({
imports: [
@ -227,26 +244,14 @@ export const buildTestingMocks = async ({
MongooseModule.forFeature([
...getModels(models),
...(autoInjectFrom
? getNestedModels([...extendedProviders], 'Repository')
? getNestedModels([...dynamicProviders], 'Repository')
: []),
]),
]
: []),
...imports,
],
providers: [
LoggerService,
EventEmitter2,
{
provide: CACHE_MANAGER,
useValue: {
del: jest.fn(),
set: jest.fn(),
get: jest.fn(),
},
},
...extendedProviders,
],
providers: [...dynamicProviders],
controllers,
...rest,
}).compile();