105 lines
2.7 KiB
TypeScript
105 lines
2.7 KiB
TypeScript
import {
|
|
WebsocketClient,
|
|
WSClientConfigurableOptions,
|
|
WS_KEY_MAP,
|
|
} from '../../src';
|
|
import {
|
|
silentLogger,
|
|
waitForSocketEvent,
|
|
WS_OPEN_EVENT_PARTIAL,
|
|
} from '../ws.util';
|
|
|
|
describe('Private Linear Perps Websocket Client', () => {
|
|
const API_KEY = process.env.API_KEY_COM;
|
|
const API_SECRET = process.env.API_SECRET_COM;
|
|
|
|
const wsClientOptions: WSClientConfigurableOptions = {
|
|
market: 'linear',
|
|
key: API_KEY,
|
|
secret: API_SECRET,
|
|
};
|
|
|
|
describe('with invalid credentials', () => {
|
|
it('should fail to open a connection if keys/signature are incorrect', async () => {
|
|
const badClient = new WebsocketClient(
|
|
{
|
|
...wsClientOptions,
|
|
key: 'bad',
|
|
secret: 'bad',
|
|
},
|
|
silentLogger
|
|
);
|
|
|
|
const wsOpenPromise = waitForSocketEvent(badClient, 'open', 2500);
|
|
|
|
badClient.connectPrivate();
|
|
|
|
try {
|
|
expect(await wsOpenPromise).toMatch('Failed to receive');
|
|
} catch (e) {
|
|
expect(e?.message).toStrictEqual('Unexpected server response: 401');
|
|
}
|
|
badClient.closeAll();
|
|
});
|
|
});
|
|
|
|
describe('with valid API credentails', () => {
|
|
let wsClient: WebsocketClient;
|
|
|
|
it('should have api credentials to test with', () => {
|
|
expect(API_KEY).toStrictEqual(expect.any(String));
|
|
expect(API_SECRET).toStrictEqual(expect.any(String));
|
|
});
|
|
|
|
const wsClientOptions: WSClientConfigurableOptions = {
|
|
market: 'linear',
|
|
key: API_KEY,
|
|
secret: API_SECRET,
|
|
};
|
|
|
|
beforeAll(() => {
|
|
wsClient = new WebsocketClient(wsClientOptions, silentLogger);
|
|
wsClient.connectPrivate();
|
|
});
|
|
|
|
afterAll(() => {
|
|
wsClient.closeAll();
|
|
});
|
|
|
|
it('should open a ws connection', async () => {
|
|
const wsOpenPromise = waitForSocketEvent(wsClient, 'open');
|
|
|
|
expect(wsOpenPromise).resolves.toMatchObject({
|
|
event: WS_OPEN_EVENT_PARTIAL,
|
|
wsKey: WS_KEY_MAP.linearPrivate,
|
|
});
|
|
|
|
await Promise.all([wsOpenPromise]);
|
|
});
|
|
|
|
it('should subscribe to private wallet events', async () => {
|
|
const wsResponsePromise = waitForSocketEvent(wsClient, 'response');
|
|
// const wsUpdatePromise = waitForSocketEvent(wsClient, 'update');
|
|
|
|
const wsTopic = 'wallet';
|
|
expect(wsResponsePromise).resolves.toMatchObject({
|
|
request: {
|
|
args: [wsTopic],
|
|
op: 'subscribe',
|
|
},
|
|
success: true,
|
|
});
|
|
|
|
// No easy way to trigger a private event (other than executing trades)
|
|
// expect(wsUpdatePromise).resolves.toMatchObject({
|
|
// topic: wsTopic,
|
|
// data: expect.any(Array),
|
|
// });
|
|
|
|
wsClient.subscribe(wsTopic);
|
|
|
|
await Promise.all([wsResponsePromise]);
|
|
});
|
|
});
|
|
});
|