summaryrefslogtreecommitdiff
path: root/std/ws/test.ts
diff options
context:
space:
mode:
Diffstat (limited to 'std/ws/test.ts')
-rw-r--r--std/ws/test.ts495
1 files changed, 0 insertions, 495 deletions
diff --git a/std/ws/test.ts b/std/ws/test.ts
deleted file mode 100644
index c437d8c28..000000000
--- a/std/ws/test.ts
+++ /dev/null
@@ -1,495 +0,0 @@
-// Copyright 2018-2021 the Deno authors. All rights reserved. MIT license.
-import { BufReader, BufWriter } from "../io/bufio.ts";
-import {
- assert,
- assertEquals,
- assertThrowsAsync,
- fail,
-} from "../testing/asserts.ts";
-import { TextProtoReader } from "../textproto/mod.ts";
-import * as bytes from "../bytes/mod.ts";
-import {
- acceptable,
- acceptWebSocket,
- createSecAccept,
- createSecKey,
- createWebSocket,
- handshake,
- OpCode,
- readFrame,
- unmask,
- writeFrame,
-} from "./mod.ts";
-import { decode, encode } from "../encoding/utf8.ts";
-import { delay } from "../async/delay.ts";
-import { serve } from "../http/server.ts";
-import { deferred } from "../async/deferred.ts";
-
-Deno.test("[ws] read unmasked text frame", async () => {
- // unmasked single text frame with payload "Hello"
- const buf = new BufReader(
- new Deno.Buffer(new Uint8Array([0x81, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f])),
- );
- const frame = await readFrame(buf);
- assertEquals(frame.opcode, OpCode.TextFrame);
- assertEquals(frame.mask, undefined);
- const actual = new TextDecoder().decode(
- new Deno.Buffer(frame.payload).bytes(),
- );
- assertEquals(actual, "Hello");
- assertEquals(frame.isLastFrame, true);
-});
-
-Deno.test("[ws] read masked text frame", async () => {
- // a masked single text frame with payload "Hello"
- const buf = new BufReader(
- new Deno.Buffer(
- new Uint8Array([
- 0x81,
- 0x85,
- 0x37,
- 0xfa,
- 0x21,
- 0x3d,
- 0x7f,
- 0x9f,
- 0x4d,
- 0x51,
- 0x58,
- ]),
- ),
- );
- const frame = await readFrame(buf);
- assertEquals(frame.opcode, OpCode.TextFrame);
- unmask(frame.payload, frame.mask);
- const actual = new TextDecoder().decode(
- new Deno.Buffer(frame.payload).bytes(),
- );
- assertEquals(actual, "Hello");
- assertEquals(frame.isLastFrame, true);
-});
-
-Deno.test("[ws] read unmasked split text frames", async () => {
- const buf1 = new BufReader(
- new Deno.Buffer(new Uint8Array([0x01, 0x03, 0x48, 0x65, 0x6c])),
- );
- const buf2 = new BufReader(
- new Deno.Buffer(new Uint8Array([0x80, 0x02, 0x6c, 0x6f])),
- );
- const [f1, f2] = await Promise.all([readFrame(buf1), readFrame(buf2)]);
- assertEquals(f1.isLastFrame, false);
- assertEquals(f1.mask, undefined);
- assertEquals(f1.opcode, OpCode.TextFrame);
- const actual1 = new TextDecoder().decode(new Deno.Buffer(f1.payload).bytes());
- assertEquals(actual1, "Hel");
-
- assertEquals(f2.isLastFrame, true);
- assertEquals(f2.mask, undefined);
- assertEquals(f2.opcode, OpCode.Continue);
- const actual2 = new TextDecoder().decode(new Deno.Buffer(f2.payload).bytes());
- assertEquals(actual2, "lo");
-});
-
-Deno.test("[ws] read unmasked ping / pong frame", async () => {
- // unmasked ping with payload "Hello"
- const buf = new BufReader(
- new Deno.Buffer(new Uint8Array([0x89, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f])),
- );
- const ping = await readFrame(buf);
- assertEquals(ping.opcode, OpCode.Ping);
- const actual1 = new TextDecoder().decode(
- new Deno.Buffer(ping.payload).bytes(),
- );
- assertEquals(actual1, "Hello");
- // deno-fmt-ignore
- const pongFrame = [0x8a, 0x85, 0x37, 0xfa, 0x21, 0x3d, 0x7f, 0x9f, 0x4d, 0x51, 0x58]
- const buf2 = new BufReader(new Deno.Buffer(new Uint8Array(pongFrame)));
- const pong = await readFrame(buf2);
- assertEquals(pong.opcode, OpCode.Pong);
- assert(pong.mask !== undefined);
- unmask(pong.payload, pong.mask);
- const actual2 = new TextDecoder().decode(
- new Deno.Buffer(pong.payload).bytes(),
- );
- assertEquals(actual2, "Hello");
-});
-
-Deno.test("[ws] read unmasked big binary frame", async () => {
- const payloadLength = 0x100;
- const a = [0x82, 0x7e, 0x01, 0x00];
- for (let i = 0; i < payloadLength; i++) {
- a.push(i);
- }
- const buf = new BufReader(new Deno.Buffer(new Uint8Array(a)));
- const bin = await readFrame(buf);
- assertEquals(bin.opcode, OpCode.BinaryFrame);
- assertEquals(bin.isLastFrame, true);
- assertEquals(bin.mask, undefined);
- assertEquals(bin.payload.length, payloadLength);
-});
-
-Deno.test("[ws] read unmasked bigger binary frame", async () => {
- const payloadLength = 0x10000;
- const a = [0x82, 0x7f, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00];
- for (let i = 0; i < payloadLength; i++) {
- a.push(i);
- }
- const buf = new BufReader(new Deno.Buffer(new Uint8Array(a)));
- const bin = await readFrame(buf);
- assertEquals(bin.opcode, OpCode.BinaryFrame);
- assertEquals(bin.isLastFrame, true);
- assertEquals(bin.mask, undefined);
- assertEquals(bin.payload.length, payloadLength);
-});
-
-Deno.test("[ws] createSecAccept", () => {
- const nonce = "dGhlIHNhbXBsZSBub25jZQ==";
- const d = createSecAccept(nonce);
- assertEquals(d, "s3pPLMBiTxaQ9kYGzzhZRbK+xOo=");
-});
-
-Deno.test("[ws] acceptable", () => {
- const ret = acceptable({
- headers: new Headers({
- upgrade: "websocket",
- "sec-websocket-key": "aaa",
- }),
- });
- assertEquals(ret, true);
-
- assert(
- acceptable({
- headers: new Headers([
- ["connection", "Upgrade"],
- ["host", "127.0.0.1:9229"],
- [
- "sec-websocket-extensions",
- "permessage-deflate; client_max_window_bits",
- ],
- ["sec-websocket-key", "dGhlIHNhbXBsZSBub25jZQ=="],
- ["sec-websocket-version", "13"],
- ["upgrade", "WebSocket"],
- ]),
- }),
- );
-});
-
-Deno.test("[ws] acceptable should return false when headers invalid", () => {
- assertEquals(
- acceptable({
- headers: new Headers({ "sec-websocket-key": "aaa" }),
- }),
- false,
- );
- assertEquals(
- acceptable({
- headers: new Headers({ upgrade: "websocket" }),
- }),
- false,
- );
- assertEquals(
- acceptable({
- headers: new Headers({ upgrade: "invalid", "sec-websocket-key": "aaa" }),
- }),
- false,
- );
- assertEquals(
- acceptable({
- headers: new Headers({ upgrade: "websocket", "sec-websocket-ky": "" }),
- }),
- false,
- );
-});
-
-Deno.test("[ws] write and read masked frame", async () => {
- const mask = new Uint8Array([0, 1, 2, 3]);
- const msg = "hello";
- const buf = new Deno.Buffer();
- const r = new BufReader(buf);
- await writeFrame(
- {
- isLastFrame: true,
- mask,
- opcode: OpCode.TextFrame,
- payload: encode(msg),
- },
- buf,
- );
- const frame = await readFrame(r);
- assertEquals(frame.opcode, OpCode.TextFrame);
- assertEquals(frame.isLastFrame, true);
- assertEquals(frame.mask, mask);
- unmask(frame.payload, frame.mask);
- assertEquals(frame.payload, encode(msg));
-});
-
-Deno.test("[ws] handshake should not send search when it's empty", async () => {
- const writer = new Deno.Buffer();
- const reader = new Deno.Buffer(encode("HTTP/1.1 400\r\n"));
-
- await assertThrowsAsync(
- async (): Promise<void> => {
- await handshake(
- new URL("ws://example.com"),
- new Headers(),
- new BufReader(reader),
- new BufWriter(writer),
- );
- },
- );
-
- const tpReader = new TextProtoReader(new BufReader(writer));
- const statusLine = await tpReader.readLine();
-
- assertEquals(statusLine, "GET / HTTP/1.1");
-});
-
-Deno.test(
- "[ws] handshake should send search correctly",
- async function wsHandshakeWithSearch(): Promise<void> {
- const writer = new Deno.Buffer();
- const reader = new Deno.Buffer(encode("HTTP/1.1 400\r\n"));
-
- await assertThrowsAsync(
- async (): Promise<void> => {
- await handshake(
- new URL("ws://example.com?a=1"),
- new Headers(),
- new BufReader(reader),
- new BufWriter(writer),
- );
- },
- );
-
- const tpReader = new TextProtoReader(new BufReader(writer));
- const statusLine = await tpReader.readLine();
-
- assertEquals(statusLine, "GET /?a=1 HTTP/1.1");
- },
-);
-
-Deno.test("[ws] ws.close() should use 1000 as close code", async () => {
- const buf = new Deno.Buffer();
- const bufr = new BufReader(buf);
- const conn = dummyConn(buf, buf);
- const ws = createWebSocket({ conn });
- await ws.close();
- const frame = await readFrame(bufr);
- assertEquals(frame.opcode, OpCode.Close);
- const code = (frame.payload[0] << 8) | frame.payload[1];
- assertEquals(code, 1000);
-});
-
-function dummyConn(r: Deno.Reader, w: Deno.Writer): Deno.Conn {
- return {
- rid: -1,
- closeWrite: (): Promise<void> => Promise.resolve(),
- read: (x: Uint8Array): Promise<number | null> => r.read(x),
- write: (x: Uint8Array): Promise<number> => w.write(x),
- close: (): void => {},
- localAddr: { transport: "tcp", hostname: "0.0.0.0", port: 0 },
- remoteAddr: { transport: "tcp", hostname: "0.0.0.0", port: 0 },
- };
-}
-
-function delayedWriter(ms: number, dest: Deno.Writer): Deno.Writer {
- return {
- write(p: Uint8Array): Promise<number> {
- return new Promise<number>((resolve) => {
- setTimeout(async (): Promise<void> => {
- resolve(await dest.write(p));
- }, ms);
- });
- },
- };
-}
-Deno.test({
- name: "[ws] WebSocket.send(), WebSocket.ping() should be exclusive",
- fn: async (): Promise<void> => {
- const buf = new Deno.Buffer();
- const conn = dummyConn(new Deno.Buffer(), delayedWriter(1, buf));
- const sock = createWebSocket({ conn });
- // Ensure send call
- await Promise.all([
- sock.send("first"),
- sock.send("second"),
- sock.ping(),
- sock.send(new Uint8Array([3])),
- ]);
- const bufr = new BufReader(buf);
- const first = await readFrame(bufr);
- const second = await readFrame(bufr);
- const ping = await readFrame(bufr);
- const third = await readFrame(bufr);
- assertEquals(first.opcode, OpCode.TextFrame);
- assertEquals(decode(first.payload), "first");
- assertEquals(first.opcode, OpCode.TextFrame);
- assertEquals(decode(second.payload), "second");
- assertEquals(ping.opcode, OpCode.Ping);
- assertEquals(third.opcode, OpCode.BinaryFrame);
- assertEquals(bytes.equals(third.payload, new Uint8Array([3])), true);
- },
-});
-
-Deno.test("[ws] createSecKeyHasCorrectLength", () => {
- // Note: relies on --seed=86 being passed to deno to reproduce failure in
- // #4063.
- const secKey = createSecKey();
- assertEquals(atob(secKey).length, 16);
-});
-
-Deno.test(
- "[ws] WebSocket should throw `Deno.errors.ConnectionReset` when peer closed connection without close frame",
- async () => {
- const buf = new Deno.Buffer();
- const eofReader: Deno.Reader = {
- read(_: Uint8Array): Promise<number | null> {
- return Promise.resolve(null);
- },
- };
- const conn = dummyConn(eofReader, buf);
- const sock = createWebSocket({ conn });
- sock.closeForce();
- await assertThrowsAsync(
- () => sock.send("hello"),
- Deno.errors.ConnectionReset,
- );
- await assertThrowsAsync(() => sock.ping(), Deno.errors.ConnectionReset);
- await assertThrowsAsync(() => sock.close(0), Deno.errors.ConnectionReset);
- },
-);
-
-Deno.test(
- "[ws] WebSocket shouldn't throw `Deno.errors.UnexpectedEof`",
- async () => {
- const buf = new Deno.Buffer();
- const eofReader: Deno.Reader = {
- read(_: Uint8Array): Promise<number | null> {
- return Promise.resolve(null);
- },
- };
- const conn = dummyConn(eofReader, buf);
- const sock = createWebSocket({ conn });
- const it = sock[Symbol.asyncIterator]();
- const { value, done } = await it.next();
- assertEquals(value, undefined);
- assertEquals(done, true);
- },
-);
-
-Deno.test({
- name:
- "[ws] WebSocket should reject sending promise when connection reset forcely",
- fn: async () => {
- const buf = new Deno.Buffer();
- let timer: number | undefined;
- const lazyWriter: Deno.Writer = {
- write(_: Uint8Array): Promise<number> {
- return new Promise((resolve) => {
- timer = setTimeout(() => resolve(0), 1000);
- });
- },
- };
- const conn = dummyConn(buf, lazyWriter);
- const sock = createWebSocket({ conn });
- const onError = (e: unknown): unknown => e;
- const p = Promise.all([
- sock.send("hello").catch(onError),
- sock.send(new Uint8Array([1, 2])).catch(onError),
- sock.ping().catch(onError),
- ]);
- sock.closeForce();
- assertEquals(sock.isClosed, true);
- const [a, b, c] = await p;
- assert(a instanceof Deno.errors.ConnectionReset);
- assert(b instanceof Deno.errors.ConnectionReset);
- assert(c instanceof Deno.errors.ConnectionReset);
- clearTimeout(timer);
- // Wait for another event loop turn for `timeout` op promise
- // to resolve, otherwise we'll get "op leak".
- await delay(10);
- },
-});
-
-Deno.test("[ws] WebSocket should act as asyncIterator", async () => {
- const pingHello = new Uint8Array([0x89, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]);
- const hello = new Uint8Array([0x81, 0x05, 0x48, 0x65, 0x6c, 0x6c, 0x6f]);
- const close = new Uint8Array([0x88, 0x04, 0x03, 0xf3, 0x34, 0x32]);
-
- enum Frames {
- ping,
- hello,
- close,
- end,
- }
-
- let frame = Frames.ping;
-
- const reader: Deno.Reader = {
- read(p: Uint8Array): Promise<number | null> {
- if (frame === Frames.ping) {
- frame = Frames.hello;
- p.set(pingHello);
- return Promise.resolve(pingHello.byteLength);
- }
-
- if (frame === Frames.hello) {
- frame = Frames.close;
- p.set(hello);
- return Promise.resolve(hello.byteLength);
- }
-
- if (frame === Frames.close) {
- frame = Frames.end;
- p.set(close);
- return Promise.resolve(close.byteLength);
- }
-
- return Promise.resolve(null);
- },
- };
-
- const conn = dummyConn(reader, new Deno.Buffer());
- const sock = createWebSocket({ conn });
-
- const events = [];
- for await (const wsEvent of sock) {
- events.push(wsEvent);
- }
-
- assertEquals(events.length, 3);
- assertEquals(events[0], ["ping", encode("Hello")]);
- assertEquals(events[1], "Hello");
- assertEquals(events[2], { code: 1011, reason: "42" });
-});
-
-Deno.test("[ws] WebSocket protocol", async () => {
- const promise = deferred();
- const server = serve({ port: 5839 });
-
- const ws = new WebSocket("ws://localhost:5839", ["foo", "bar"]);
- ws.onopen = () => {
- assertEquals(ws.protocol, "foo, bar");
- ws.close();
- };
- ws.onerror = () => fail();
- ws.onclose = () => {
- server.close();
- promise.resolve();
- };
-
- const x = await server[Symbol.asyncIterator]().next();
- if (!x.done) {
- const { conn, r: bufReader, w: bufWriter, headers } = x.value;
- await acceptWebSocket({
- conn,
- bufReader,
- bufWriter,
- headers,
- });
-
- await promise;
- } else {
- fail();
- }
-});
generated by cgit v1.2.3 (git 2.25.1) at 2025-06-27 08:54:44 +0000