Mirrors/trezor-suite
2
0
Fork 0
mirror of https://github.com/trezor/trezor-suite.git synced 2026-03-23 15:47:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

test(websocket-client): use WebsocketClient class directly

Browse Source
This commit is contained in:
Martin Varmuza
2025-03-24 06:46:47 +01:00
committed by martin
parent fb0052142d
commit 9731de127c
1 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
packages/websocket-client/tests/client.test.ts
View File

@@ -2,14 +2,7 @@ import { ServerOptions, WebSocket } from 'ws';
import { WebsocketClient } from '../src/client';
class Client extends WebsocketClient<{ 'foo-event': 'bar-event' }> {
ping() {
return this.sendMessage({ method: 'ping' });
}
sendMessage(message: Record<string, any>) {
return super.sendMessage(message);
}
}
class Client extends WebsocketClient<{ 'foo-event': 'bar-event' }> {}
class Server extends WebSocket.Server {
private _url: string;
@@ -95,6 +88,7 @@ describe('WebsocketClient', () => {
jest.useFakeTimers();
const cli = new Client({ url: server.getUrl(), pingTimeout: 5000 });
// @ts-expect-error ping is protected
const pingSpy = jest.spyOn(cli, 'ping');
await cli.connect();
@@ -112,7 +106,7 @@ describe('WebsocketClient', () => {
});
it('reconnect with sync disconnect()', async () => {
const cli = new Client({ url: server.getUrl() });
const cli = new WebsocketClient({ url: server.getUrl() });
await cli.connect();
cli.disconnect(); // NOTE: intentionally not awaited
await cli.connect();