fix forceful close workflow

test/spot/ws.private.v1.test.ts

@@ -9,6 +9,7 @@ import {
   getSilentLogger,
   waitForSocketEvent,
   WS_OPEN_EVENT_PARTIAL,
+  fullLogger,
 } from '../ws.util';
 
 describe('Private Spot V1 Websocket Client', () => {
@@ -30,6 +31,7 @@ describe('Private Spot V1 Websocket Client', () => {
   beforeAll(() => {
     wsClient = new WebsocketClient(
       wsClientOptions,
+      // fullLogger
       getSilentLogger('expectSuccess')
     );
     logAllEvents(wsClient);

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