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

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

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