fix: unit tests

MetaMask · Aug 27, 2024 · 70f3b76 · 70f3b76
1 parent 34af96a
commit 70f3b76
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/...s/sdk-communication-layer/src/services/SocketService/ChannelManager/createChannel.test.ts b/...s/sdk-communication-layer/src/services/SocketService/ChannelManager/createChannel.test.ts
@@ -7,7 +7,9 @@ import { setupChannelListeners } from './setupChannelListeners';
 jest.mock('./setupChannelListeners');
 
 const mockConnect = jest.fn();
-const mockEmit = jest.fn();
+const mockEmit = jest.fn((_event, _payload, callback) => {
+  callback(null, { ready: true }); // Simulate successful emit
+});
 const mockGetMyPublicKey = jest.fn();
 
 describe('createChannel', () => {
@@ -67,11 +69,15 @@ describe('createChannel', () => {
   it('should emit JOIN_CHANNEL event with correct parameters', async () => {
     const result = await createChannel(instance);
 
-    expect(instance.state.socket?.emit).toHaveBeenCalledWith('join_channel', {
-      channelId: instance.state.channelId,
-      clientType: 'dapp',
-      context: 'testContextcreateChannel',
-    });
+    expect(instance.state.socket?.emit).toHaveBeenCalledWith(
+      'join_channel',
+      {
+        channelId: instance.state.channelId,
+        clientType: 'dapp',
+        context: 'testContextcreateChannel',
+      },
+      expect.any(Function), // Match the callback function
+    );
 
     expect(result.pubKey).toBe('');
   });