test(): add test for error handling in controller

nestjs · Aug 3, 2023 · 5fd35cd · 5fd35cd
1 parent 4daba90
commit 5fd35cd
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 3 deletions.
diff --git a/tests/e2e/module-e2e.spec.ts b/tests/e2e/module-e2e.spec.ts
@@ -122,12 +122,23 @@ describe('EventEmitterModule - e2e', () => {
     expect(customConsumer.isEmitted).toBeTruthy();
   });
 
-  it('should be able to gracefully recover when an unexpected error occurs', async () => {
+  it('should be able to gracefully recover when an unexpected error occurs from provider', async () => {
     const eventsConsumerRef = app.get(EventsProviderConsumer);
     await app.init();
 
     const emitter = app.get(EventEmitter2);
-    const result = emitter.emit('error-handling.event');
+    const result = emitter.emit('error-handling.provider');
+
+    expect(eventsConsumerRef.errorHandlingCalls).toEqual(1);
+    expect(result).toBeTruthy();
+  });
+
+  it('should be able to gracefully recover when an unexpected error occurs from controller', async () => {
+    const eventsConsumerRef = app.get(EventsControllerConsumer);
+    await app.init();
+
+    const emitter = app.get(EventEmitter2);
+    const result = emitter.emit('error-handling.controller');
 
     expect(eventsConsumerRef.errorHandlingCalls).toEqual(1);
     expect(result).toBeTruthy();

diff --git a/tests/src/events-controller.consumer.ts b/tests/src/events-controller.consumer.ts
@@ -4,9 +4,16 @@ import { OnEvent } from '../../lib';
 @Controller()
 export class EventsControllerConsumer {
   public eventPayload = {};
+  public errorHandlingCalls = 0;
 
   @OnEvent('test.*')
   onTestEvent(payload: Record<string, any>) {
     this.eventPayload = payload;
   }
+
+  @OnEvent('error-handling.controller')
+  onErrorHandlingEvent() {
+    this.errorHandlingCalls++;
+    throw new Error('This is a test error');
+  }
 }
diff --git a/tests/src/events-provider.consumer.ts b/tests/src/events-provider.consumer.ts
@@ -18,7 +18,7 @@ export class EventsProviderConsumer {
     this.stackedEventCalls++;
   }
 
-  @OnEvent('error-handling.*')
+  @OnEvent('error-handling.provider')
   onErrorHandlingEvent() {
     this.errorHandlingCalls++;
     throw new Error('This is a test error');