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

feat(api): enhance typing

Browse Source
This commit is contained in:
yassinedorbozgithub 2025-06-16 15:52:15 +01:00
parent 0de472e5c8
commit 027d610578
1 changed files with 34 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -23,12 +23,36 @@ type TTypeOrToken = [
...(new (...args: any[]) => any[]),
];
type model = ModelDefinition | `${string}Model`;
type TModel = ModelDefinition | `${string}Model`;
interface buildTestingMocksProps extends ModuleMetadata {
models?: model[];
autoInjectFrom?: ('all' | 'controllers' | 'providers')[];
}
type ToUnionArray<T> = (NonNullable<T> extends (infer U)[] ? U : never)[];
type buildTestingMocksProps<
P extends ModuleMetadata['providers'] = ModuleMetadata['providers'],
C extends ModuleMetadata['controllers'] = ModuleMetadata['controllers'],
> = ModuleMetadata & {
models?: TModel[];
} & (
| {
providers: NonNullable<P>;
controllers: NonNullable<C>;
autoInjectFrom?: 'all'[];
}
| {
providers: NonNullable<P>;
controllers?: undefined;
autoInjectFrom?: 'providers'[];
}
| {
controllers: NonNullable<C>;
autoInjectFrom?: 'controllers'[];
}
| {
providers?: never;
controllers?: never;
autoInjectFrom?: never;
}
);
const findInstances = async <T extends TTypeOrToken>(
type: keyof TestingModule,
@ -133,7 +157,7 @@ const canInjectModels = (imports: buildTestingMocksProps['imports']): boolean =>
dynamicModule.module.name === 'MongooseModule',
) > -1;
const getModels = (models: model[]): ModelDefinition[] =>
const getModels = (models: TModel[]): ModelDefinition[] =>
models.map((model) =>
typeof model === 'string' ? getModel(model, 'Model') : model,
);
@ -147,13 +171,14 @@ export const buildTestingMocks = async ({
...rest
}: buildTestingMocksProps) => {
const nestedProviders: Provider[] = [];
const canAutoInjectFromAll = autoInjectFrom?.includes('all');
const injectionFrom = autoInjectFrom as ToUnionArray<typeof autoInjectFrom>;
const canAutoInjectFromAll = injectionFrom?.includes('all');
if (canAutoInjectFromAll || autoInjectFrom?.includes('providers')) {
if (canAutoInjectFromAll || injectionFrom?.includes('providers')) {
nestedProviders.push(...providers, ...getNestedDependencies(providers));
}
if (canAutoInjectFromAll || autoInjectFrom?.includes('controllers')) {
if (canAutoInjectFromAll || injectionFrom?.includes('controllers')) {
nestedProviders.push(...controllers, ...getNestedDependencies(controllers));
}