fix forceful close workflow

2022-11-12 17:56:52 +00:00
parent ba7099d11c
commit d3b2c65c22
3 changed files with 21 additions and 11 deletions
--- a/test/unified-margin/ws.public.perp.usdc.test.ts
+++ b/test/unified-margin/ws.public.perp.usdc.test.ts
@@ -28,10 +28,6 @@ describe('Public Unified Margin Websocket Client (Perps - USDC)', () => {
    wsClient.connectPublic();
  });

-  afterAll(() => {
-    wsClient.closeAll(true);
-  });
-
  it('should open a public ws connection', async () => {
    const wsOpenPromise = waitForSocketEvent(wsClient, 'open');
    try {
@@ -42,6 +38,8 @@ describe('Public Unified Margin Websocket Client (Perps - USDC)', () => {
    } catch (e) {
      expect(e).toBeFalsy();
    }
+
+    wsClient.closeAll(true);
  });

  // TODO: are there USDC topics? This doesn't seem to work