mirrors/hedgedoc
1
0
Fork 0
mirror of https://github.com/hedgedoc/hedgedoc.git synced 2025-05-09 13:51:57 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: improve select query mock builder

Browse source 
Signed-off-by: Tilman Vatteroth <git@tilmanvatteroth.de>
This commit is contained in:
Tilman Vatteroth 2023-06-11 12:58:14 +02:00
parent f95f5406be
commit eb986b1504
1 changed files with 38 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

46
backend/src/utils/test-utils/mockSelectQueryBuilder.ts
View file

@ -13,18 +13,48 @@ import { ObjectLiteral, Repository, SelectQueryBuilder } from 'typeorm';
* @return The mocked query builder * @return The mocked query builder
*/ */
export function mockSelectQueryBuilder<T extends ObjectLiteral>( export function mockSelectQueryBuilder<T extends ObjectLiteral>(
returnValue: T | null, returnValue: T | T[] | null,
): SelectQueryBuilder<T> { ): SelectQueryBuilder<T> {
const mockedQueryBuilder: SelectQueryBuilder<T> = Mock.of< const mockedQueryBuilder: SelectQueryBuilder<T> = Mock.of<
SelectQueryBuilder<T> SelectQueryBuilder<T>
>({ >({
where: () => mockedQueryBuilder, where: (where) => {
andWhere: () => mockedQueryBuilder, if (typeof where === 'function') {
where(mockedQueryBuilder);
}
return mockedQueryBuilder;
},
andWhere: (where) => {
if (typeof where === 'function') {
where(mockedQueryBuilder);
}
return mockedQueryBuilder;
},
subQuery: () => mockedQueryBuilder,
select: () => mockedQueryBuilder,
from: <M extends ObjectLiteral>() => mockSelectQueryBuilder<M>(null),
innerJoin: () => mockedQueryBuilder,
leftJoinAndSelect: () => mockedQueryBuilder, leftJoinAndSelect: () => mockedQueryBuilder,
getOne: () => Promise.resolve(returnValue), getQuery: () => '',
orWhere: () => mockedQueryBuilder, getOne: () =>
Promise.resolve(
Array.isArray(returnValue) ? returnValue[0] : returnValue,
),
orWhere: (where) => {
if (typeof where === 'function') {
where(mockedQueryBuilder);
}
return mockedQueryBuilder;
},
setParameter: () => mockedQueryBuilder, setParameter: () => mockedQueryBuilder,
getMany: () => Promise.resolve(returnValue ? [returnValue] : []), getMany: () => {
if (!returnValue) {
return Promise.resolve([]);
}
return Promise.resolve(
Array.isArray(returnValue) ? returnValue : [returnValue],
);
},
}); });
return mockedQueryBuilder; return mockedQueryBuilder;
} }
@ -39,9 +69,9 @@ export function mockSelectQueryBuilder<T extends ObjectLiteral>(
*/ */
export function mockSelectQueryBuilderInRepo<T extends ObjectLiteral>( export function mockSelectQueryBuilderInRepo<T extends ObjectLiteral>(
repository: Repository<T>, repository: Repository<T>,
returnValue: T | null, returnValue: T | T[] | null,
): SelectQueryBuilder<T> { ): SelectQueryBuilder<T> {
const selectQueryBuilder = mockSelectQueryBuilder(returnValue); const selectQueryBuilder = mockSelectQueryBuilder<T>(returnValue);
jest jest
.spyOn(repository, 'createQueryBuilder') .spyOn(repository, 'createQueryBuilder')
.mockImplementation(() => selectQueryBuilder); .mockImplementation(() => selectQueryBuilder);