test(server): add MockJobQueue (#11032)

This commit is contained in:
fengmk2
2025-03-21 04:08:29 +00:00
parent 4ed2c9bd81
commit 55a60906a5
6 changed files with 61 additions and 8 deletions
@@ -8,11 +8,22 @@ import cookieParser from 'cookie-parser';
import graphqlUploadExpress from 'graphql-upload/graphqlUploadExpress.mjs';
import supertest from 'supertest';
import { AFFiNELogger, ApplyType, GlobalExceptionFilter } from '../../base';
import {
AFFiNELogger,
ApplyType,
GlobalExceptionFilter,
JobQueue,
} from '../../base';
import { AuthService } from '../../core/auth';
import { Mailer } from '../../core/mail';
import { UserModel } from '../../models';
import { createFactory, MockedUser, MockUser, MockUserInput } from '../mocks';
import {
createFactory,
MockedUser,
MockJobQueue,
MockUser,
MockUserInput,
} from '../mocks';
import { MockMailer } from '../mocks/mailer.mock';
import { createTestingModule } from './testing-module';
import { initTestingDB, TEST_LOG_LEVEL } from './utils';
@@ -85,6 +96,7 @@ export class TestingApp extends ApplyType<INestApplication>() {
readonly create!: ReturnType<typeof createFactory>;
readonly mails!: MockMailer;
readonly queue!: MockJobQueue;
[Symbol.asyncDispose](): Promise<void> {
return this.close();
@@ -285,6 +297,8 @@ function makeTestingApp(app: INestApplication): TestingApp {
testingApp.create = createFactory(app.get(PrismaClient, { strict: false }));
// @ts-expect-error allow
testingApp.mails = app.get(Mailer, { strict: false }) as MockMailer;
// @ts-expect-error allow
testingApp.queue = app.get(JobQueue, { strict: false }) as MockJobQueue;
return new Proxy(testingApp, {
get(target, prop) {
@@ -9,7 +9,7 @@ import {
import { PrismaClient } from '@prisma/client';
import { AppModule, FunctionalityModules } from '../../app.module';
import { AFFiNELogger, Runtime } from '../../base';
import { AFFiNELogger, JobQueue, Runtime } from '../../base';
import { GqlModule } from '../../base/graphql';
import { AuthGuard, AuthModule } from '../../core/auth';
import { Mailer, MailModule } from '../../core/mail';
@@ -17,11 +17,11 @@ import { ModelsModule } from '../../models';
// for jsdoc inference
// oxlint-disable-next-line no-unused-vars
import type { createModule } from '../create-module';
import { createFactory } from '../mocks';
import { createFactory, MockJobQueue } from '../mocks';
import { MockMailer } from '../mocks/mailer.mock';
import { initTestingDB, TEST_LOG_LEVEL } from './utils';
interface TestingModuleMeatdata extends ModuleMetadata {
interface TestingModuleMetadata extends ModuleMetadata {
tapModule?(m: TestingModuleBuilder): void;
}
@@ -29,6 +29,7 @@ export interface TestingModule extends BaseTestingModule {
initTestingDB(): Promise<void>;
create: ReturnType<typeof createFactory>;
mails: MockMailer;
queue: MockJobQueue;
[Symbol.asyncDispose](): Promise<void>;
}
@@ -58,7 +59,7 @@ class MockResolver {
* @deprecated use {@link createModule} instead
*/
export async function createTestingModule(
moduleDef: TestingModuleMeatdata = {},
moduleDef: TestingModuleMetadata = {},
autoInitialize = true
): Promise<TestingModule> {
// setting up
@@ -88,10 +89,11 @@ export async function createTestingModule(
controllers: moduleDef.controllers,
});
builder.overrideProvider(Mailer).useClass(MockMailer);
builder.overrideProvider(JobQueue).useClass(MockJobQueue);
if (moduleDef.tapModule) {
moduleDef.tapModule(builder);
}
builder.overrideProvider(Mailer).useClass(MockMailer);
const module = await builder.compile();
@@ -114,6 +116,7 @@ export async function createTestingModule(
};
testingModule.mails = module.get(Mailer, { strict: false }) as MockMailer;
testingModule.queue = module.get(JobQueue, { strict: false }) as MockJobQueue;
const logger = new AFFiNELogger();
// we got a lot smoking tests try to break nestjs