Merge pull request #92 from hubert-deriv/useWS_test_cov

mohsen-deriv · web-flow · commit 9312d63caeb8 · 2023-04-05T16:23:01.000+03:30
Hubert / useWS test coverage
diff --git a/src/hooks/useWs/__tests__/useWs.test.tsx b/src/hooks/useWs/__tests__/useWs.test.tsx
@@ -205,4 +205,39 @@ describe('Use WS', () => {
       req_id: 1,
     });
   });
+
+  it('Should be able to clear the websocket call', async () => {
+    const { result, waitForNextUpdate } = renderHook(() => useWS('ping'));
+
+    expect(result.current.is_loading).toBeFalsy();
+
+    act(() => {
+      result.current.send();
+    });
+    expect(result.current.is_loading).toBeTruthy();
+
+    await expect(wsServer).toReceiveMessage({ ping: 1, req_id: 1 });
+
+    wsServer.send({
+      echo_req: {
+        ping: 1,
+        req_id: 1,
+      },
+      msg_type: 'ping',
+      ping: 'pong',
+      req_id: 1,
+    });
+
+    await waitForNextUpdate();
+
+    expect(result.current.is_loading).toBeFalsy();
+
+    expect(result.current.data).toBe('pong');
+
+    act(() => {
+      result.current.clear();
+    });
+
+    expect(result.current.data).toBe(null);
+  });
 });
