49 lines
2.0 KiB
TypeScript
49 lines
2.0 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
import { z } from 'zod';
|
|
import type { WebSocketServer } from 'ws';
|
|
|
|
import { ApplicationRouter } from '../../../source/services/router.ts';
|
|
import { WsTransportRouter } from '../../../source/services/transport/ws-transport.ts';
|
|
import { Logger } from '../../../source/utils/logger.ts';
|
|
import { toExtendedJson } from '@xo-cash/utils';
|
|
|
|
describe('WebSocket request decoding', (): void => {
|
|
it('decodes the minimal route-agnostic envelope and Extended JSON body', async (): Promise<void> => {
|
|
await expect(WsTransportRouter.decodeWebSocketRequest(toExtendedJson({
|
|
id: 'request-1',
|
|
path: '/data/write',
|
|
body: { value: new Uint8Array([ 1, 2, 3 ]) },
|
|
}))).resolves.toEqual({
|
|
requestId: 'request-1',
|
|
path: '/data/write',
|
|
body: { value: new Uint8Array([ 1, 2, 3 ]) },
|
|
});
|
|
});
|
|
|
|
it.each([ '{}', '{"path":42}', '{"path":"/data/get","id":1}', '{"path":"/data/get","method":"POST"}' ])(
|
|
'rejects an invalid envelope: %s',
|
|
async (payload) => {
|
|
await expect(WsTransportRouter.decodeWebSocketRequest(payload)).rejects.toBeInstanceOf(z.ZodError);
|
|
},
|
|
);
|
|
|
|
it('rejects malformed JSON', async (): Promise<void> => {
|
|
await expect(WsTransportRouter.decodeWebSocketRequest('{')).rejects.toMatchObject({
|
|
statusCode: 400,
|
|
message: 'Invalid JSON in WebSocket message',
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('WsTransportRouter payload limits', (): void => {
|
|
it("configures Hono's ws server with the requested maxPayload", async () => {
|
|
const debug = new Logger('ws-transport-test');
|
|
const router = await ApplicationRouter.create([]);
|
|
const transport = new WsTransportRouter(router, debug, 1024);
|
|
const wsServer = transport.websocketServer as unknown as WebSocketServer;
|
|
|
|
expect(wsServer.options.maxPayload).toBe(1024);
|
|
await transport.stop();
|
|
});
|
|
});
|