Close webSocketClient when no longer needed

[lwin-kyaw]

Hi, I'm trying to close the websocket transport after I've done with the socket in the jest unittests.
However, I run into this error.

ReferenceError: You are trying to `import` a file after the Jest environment has been torn down. From src/tests.ts.

      at node_modules/viem/utils/rpc/webSocket.ts:28:31
      at getSocket (node_modules/viem/utils/rpc/webSocket.ts:28:25)
      at setup (node_modules/viem/utils/rpc/socket.ts:160:24)
      at Timeout._onTimeout (node_modules/viem/utils/rpc/socket.ts:147:13)
  console.error
    TypeError: WebSocket is not a constructor

Here's the minimal script to reproduce the issue

...
const url = 'wss://....`
 
// create websocket rpc client
const client = createPublicClient({
  transport: webSocket(url),
});
....
....

// close the socket when no longer needed
const socket = await client.transport?.getSocket();
if (socket) {
  socket.close();
}

I've also tried another way to close the socket with this

import { getWebSocketRpcClient } from 'viem/utils';

const socketClient = await getWebSocketRpcClient('wss://....`);
console.log('socketClient', socketClient);
socketClient.socket.close();

Still run into the same error.

May I know how can close the websocket transport after I've done with my test cases?

Node Version: v22.11.0
Viem Version: 2.33.1

Thank you very much.