deno.land / std@0.91.0 / ws / test.ts
نووسراو ببینە
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497// 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 { 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: new TextEncoder().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, new TextEncoder().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(new TextEncoder().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( new TextEncoder().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(new TextDecoder().decode(first.payload), "first"); assertEquals(first.opcode, OpCode.TextFrame); assertEquals(new TextDecoder().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", new TextEncoder().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(); }});
std
Version Info
3 years ago
External Dependencies
No external dependencies 🎉
