mirror of
https://github.com/toeverything/AFFiNE.git
synced 2026-08-03 18:40:00 +08:00
test(server): add MockJobQueue (#11032)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user