Update web.socket controller spec

multiversx · Jan 3, 2025 · 471468b · 471468b
1 parent 0a0fe48
commit 471468b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/test/unit/controllers/web.socket.publiser.controller.spec.ts b/src/test/unit/controllers/web.socket.publiser.controller.spec.ts
@@ -1,11 +1,14 @@
 import { ShardTransaction } from "@elrondnetwork/transaction-processor";
 import { TestingModule, Test } from "@nestjs/testing";
+import { EventEmitter2 } from "@nestjs/event-emitter";
 import { WebSocketPublisherController } from "src/common/websockets/web-socket-publisher-controller";
 import { WebSocketPublisherService } from "src/common/websockets/web-socket-publisher-service";
+import { MetricsEvents } from "src/utils/metrics-events.constants";
 
 describe('WebSocketPublisherController', () => {
   let controller: WebSocketPublisherController;
   let webSocketPublisherService: WebSocketPublisherService;
+  let eventEmitter: EventEmitter2;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
@@ -19,29 +22,39 @@ describe('WebSocketPublisherController', () => {
             onBatchUpdated: jest.fn(),
           },
         },
+        {
+          provide: EventEmitter2,
+          useValue: {
+            emit: jest.fn(),
+          },
+        },
       ],
     }).compile();
 
     controller = module.get<WebSocketPublisherController>(WebSocketPublisherController);
     webSocketPublisherService = module.get<WebSocketPublisherService>(WebSocketPublisherService);
+    eventEmitter = module.get<EventEmitter2>(EventEmitter2);
   });
 
   it('should handle transactionsCompleted event', async () => {
     const mockTransactions = [{}, {}] as ShardTransaction[];
     await controller.transactionsCompleted(mockTransactions);
     expect(webSocketPublisherService.onTransactionCompleted).toHaveBeenCalledTimes(mockTransactions.length);
+    expect(eventEmitter.emit).toHaveBeenCalledWith(MetricsEvents.SetTransactionsCompleted, { transactions: mockTransactions });
   });
 
   it('should handle transactionsPendingResults event', async () => {
     const mockTransactions = [{}, {}] as ShardTransaction[];
     await controller.transactionsPendingResults(mockTransactions);
     expect(webSocketPublisherService.onTransactionPendingResults).toHaveBeenCalledTimes(mockTransactions.length);
+    expect(eventEmitter.emit).toHaveBeenCalledWith(MetricsEvents.SetTransactionsPendingResults, { transactions: mockTransactions });
   });
 
   it('should handle onBatchUpdated event', () => {
     const mockPayload = { address: 'testAddress', batchId: 'testBatchId', txHashes: ['hash1', 'hash2'] };
     controller.onBatchUpdated(mockPayload);
     expect(webSocketPublisherService.onBatchUpdated).toHaveBeenCalledWith(mockPayload.address, mockPayload.batchId, mockPayload.txHashes);
     expect(webSocketPublisherService.onBatchUpdated).toHaveBeenCalledTimes(1);
+    expect(eventEmitter.emit).toHaveBeenCalledWith(MetricsEvents.SetBatchUpdated);
   });
 });