Add global class middleware in nest application for testing

Author: nicohrubec

dev-packages/e2e-tests/test-applications/nestjs-basic/src/app.module.ts

@@ -1,12 +1,20 @@
-import { Module } from '@nestjs/common';
+import {MiddlewareConsumer, Module} from '@nestjs/common';
 import { ScheduleModule } from '@nestjs/schedule';
 import { SentryModule } from '@sentry/nestjs/setup';
 import { AppController } from './app.controller';
 import { AppService } from './app.service';
+import { LoggerMiddleware } from "./logger.middleware";
 
 @Module({
   imports: [SentryModule.forRoot(), ScheduleModule.forRoot()],
   controllers: [AppController],
   providers: [AppService],
 })
-export class AppModule {}
+export class AppModule {
+  configure(consumer: MiddlewareConsumer): void {
+    console.log('Add class middleware.');
+    consumer
+      .apply(LoggerMiddleware)
+      .forRoutes(AppController);
+  }
+}

dev-packages/e2e-tests/test-applications/nestjs-basic/src/logger.middleware.ts

@@ -0,0 +1,18 @@
+import { Injectable, NestMiddleware } from '@nestjs/common';
+import { Request, Response, NextFunction } from 'express';
+
+function delay(ms) {
+  const start = Date.now();
+  while (Date.now() - start < ms) {
+    // Do nothing
+  }
+}
+
+@Injectable()
+export class LoggerMiddleware implements NestMiddleware {
+  use(req: Request, res: Response, next: NextFunction) {
+    console.log('Class middleware...');
+    delay(500);
+    next();
+  }
+}