Add websockets e2e for nestjs

Author: nicohrubec

dev-packages/e2e-tests/test-applications/nestjs-websockets/nest-cli.json

@@ -0,0 +1,8 @@
+{
+  "$schema": "https://json.schemastore.org/nest-cli",
+  "collection": "@nestjs/schematics",
+  "sourceRoot": "src",
+  "compilerOptions": {
+    "deleteOutDir": true
+  }
+}

dev-packages/e2e-tests/test-applications/nestjs-websockets/package.json

@@ -0,0 +1,33 @@
+{
+  "name": "nestjs-websockets",
+  "version": "0.0.1",
+  "private": true,
+  "scripts": {
+    "build": "nest build",
+    "start": "nest start",
+    "test": "playwright test",
+    "test:build": "pnpm install && pnpm build",
+    "test:assert": "pnpm test"
+  },
+  "dependencies": {
+    "@nestjs/common": "^10.0.0",
+    "@nestjs/core": "^10.0.0",
+    "@nestjs/platform-express": "^10.0.0",
+    "@nestjs/websockets": "^10.0.0",
+    "@nestjs/platform-socket.io": "^10.0.0",
+    "@sentry/nestjs": "latest || *",
+    "reflect-metadata": "^0.2.0",
+    "rxjs": "^7.8.1"
+  },
+  "devDependencies": {
+    "@playwright/test": "~1.56.0",
+    "@sentry-internal/test-utils": "link:../../../test-utils",
+    "@nestjs/cli": "^10.0.0",
+    "@types/node": "^18.19.1",
+    "socket.io-client": "^4.0.0",
+    "typescript": "~5.0.0"
+  },
+  "volta": {
+    "extends": "../../package.json"
+  }
+}

dev-packages/e2e-tests/test-applications/nestjs-websockets/playwright.config.mjs

@@ -0,0 +1,7 @@
+import { getPlaywrightConfig } from '@sentry-internal/test-utils';
+
+const config = getPlaywrightConfig({
+  startCommand: `pnpm start`,
+});
+
+export default config;

dev-packages/e2e-tests/test-applications/nestjs-websockets/src/app.controller.ts

@@ -0,0 +1,11 @@
+import { Controller, Get } from '@nestjs/common';
+import { flush } from '@sentry/nestjs';
+
+@Controller()
+export class AppController {
+  @Get('/flush')
+  async flush() {
+    await flush();
+    return 'ok';
+  }
+}

dev-packages/e2e-tests/test-applications/nestjs-websockets/src/app.gateway.ts

@@ -0,0 +1,25 @@
+import { SubscribeMessage, WebSocketGateway, MessageBody } from '@nestjs/websockets';
+import * as Sentry from '@sentry/nestjs';
+
+@WebSocketGateway({ cors: true })
+export class AppGateway {
+  @SubscribeMessage('test-message')
+  handleTestMessage(@MessageBody() data: { message: string }) {
+    return { event: 'test-response', data: { message: data.message } };
+  }
+
+  @SubscribeMessage('test-exception')
+  handleTestException() {
+    throw new Error('This is an exception in a WebSocket handler');
+  }
+
+  @SubscribeMessage('test-manual-capture')
+  handleManualCapture() {
+    try {
+      throw new Error('Manually captured WebSocket error');
+    } catch (e) {
+      Sentry.captureException(e);
+    }
+    return { event: 'capture-response', data: { success: true } };
+  }
+}

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

@@ -0,0 +1,18 @@
+import { Module } from '@nestjs/common';
+import { APP_FILTER } from '@nestjs/core';
+import { SentryGlobalFilter, SentryModule } from '@sentry/nestjs/setup';
+import { AppController } from './app.controller';
+import { AppGateway } from './app.gateway';
+
+@Module({
+  imports: [SentryModule.forRoot()],
+  controllers: [AppController],
+  providers: [
+    {
+      provide: APP_FILTER,
+      useClass: SentryGlobalFilter,
+    },
+    AppGateway,
+  ],
+})
+export class AppModule {}

dev-packages/e2e-tests/test-applications/nestjs-websockets/src/instrument.ts

@@ -0,0 +1,11 @@
+import * as Sentry from '@sentry/nestjs';
+
+Sentry.init({
+  environment: 'qa',
+  dsn: process.env.E2E_TEST_DSN,
+  tunnel: `http://localhost:3031/`,
+  tracesSampleRate: 1,
+  transportOptions: {
+    bufferSize: 1000,
+  },
+});

dev-packages/e2e-tests/test-applications/nestjs-websockets/src/main.ts

@@ -0,0 +1,15 @@
+// Import this first
+import './instrument';
+
+// Import other modules
+import { NestFactory } from '@nestjs/core';
+import { AppModule } from './app.module';
+
+const PORT = 3030;
+
+async function bootstrap() {
+  const app = await NestFactory.create(AppModule);
+  await app.listen(PORT);
+}
+
+bootstrap();

dev-packages/e2e-tests/test-applications/nestjs-websockets/start-event-proxy.mjs

@@ -0,0 +1,6 @@
+import { startEventProxyServer } from '@sentry-internal/test-utils';
+
+startEventProxyServer({
+  port: 3031,
+  proxyServerName: 'nestjs-websockets',
+});

dev-packages/e2e-tests/test-applications/nestjs-websockets/tests/errors.test.ts

@@ -0,0 +1,23 @@
+import { expect, test } from '@playwright/test';
+import { waitForError } from '@sentry-internal/test-utils';
+import { io } from 'socket.io-client';
+
+test('Captures manually reported error in WebSocket gateway handler', async ({ baseURL }) => {
+  const errorPromise = waitForError('nestjs-websockets', event => {
+    return event.exception?.values?.[0]?.value === 'Manually captured WebSocket error';
+  });
+
+  const socket = io(baseURL!);
+  await new Promise<void>(resolve => socket.on('connect', resolve));
+
+  socket.emit('test-manual-capture', {});
+
+  const error = await errorPromise;
+
+  expect(error.exception?.values?.[0]).toMatchObject({
+    type: 'Error',
+    value: 'Manually captured WebSocket error',
+  });
+
+  socket.disconnect();
+});