chore(vendor): import pi 1.3.0
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
This commit is contained in:
alexion committed
1 parent
719370c8f7
commit
6081e3e325
1743 files changed
+205459
-131704
No files matched your search
@@ -0,0 +1,158 @@
|
||||
import assert from "node:assert/strict";
|
||||
import type { IncomingMessage } from "node:http";
|
||||
import { Type } from "typebox";
|
||||
import { describe, it } from "vitest";
|
||||
import { WebSocketServer } from "ws";
|
||||
import {
|
||||
acceptWebSockets,
|
||||
createHost,
|
||||
hostNamed,
|
||||
inflateMessage,
|
||||
isChordError,
|
||||
method,
|
||||
remote,
|
||||
type WebSocketLike,
|
||||
webSocketLink,
|
||||
} from "../src/index.ts";
|
||||
|
||||
const Board = remote("t.board", {
|
||||
version: 1,
|
||||
input: Type.Object({}),
|
||||
host: hostNamed("server"),
|
||||
variants: {
|
||||
main: {
|
||||
state: { notes: Type.Array(Type.String()) },
|
||||
methods: { add: method(Type.String()), who: method(undefined, Type.Unknown()) },
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
describe("WebSocket links", () => {
|
||||
it("serves a browser-style client over a real socket, stamped by the authenticator", async () => {
|
||||
const server = createHost({ self: { kind: "host" }, name: "server", onError: () => {} });
|
||||
server.serve(Board, (_input, principal, self) => {
|
||||
const notes = self.state("notes", Type.Array(Type.String()), []);
|
||||
return {
|
||||
variant: "main",
|
||||
notes,
|
||||
add: (text, ctx) => {
|
||||
notes.change(ctx, (draft) => {
|
||||
draft.push(text);
|
||||
});
|
||||
},
|
||||
who: () => principal,
|
||||
};
|
||||
});
|
||||
const wss = new WebSocketServer({ port: 0, host: "127.0.0.1" });
|
||||
await new Promise<void>((resolve) => wss.once("listening", () => resolve()));
|
||||
acceptWebSockets(wss as never, server, {
|
||||
principal: (request: IncomingMessage) => {
|
||||
const token = new URL(request.url ?? "/", "http://x").searchParams.get("token");
|
||||
return token === "ok" ? { guest: "web" } : undefined;
|
||||
},
|
||||
});
|
||||
const port = (wss.address() as { port: number }).port;
|
||||
const open = (token: string) =>
|
||||
createHost({
|
||||
self: { kind: "host" },
|
||||
name: "browser",
|
||||
kind: "client",
|
||||
upstream: {
|
||||
link: webSocketLink(new WebSocket(`ws://127.0.0.1:${port}/?token=${token}`) as unknown as WebSocketLike),
|
||||
principal: { kind: "server" },
|
||||
},
|
||||
onError: () => {},
|
||||
});
|
||||
try {
|
||||
const browser = open("ok");
|
||||
const board = browser.hold(Board, {});
|
||||
const seen: unknown[] = [];
|
||||
board.notes.subscribe((value) => seen.push(value));
|
||||
await board.add("hello");
|
||||
assert.deepEqual(await board.who(), { guest: "web" });
|
||||
for (let i = 0; i < 50 && board.notes.value?.length !== 1; i++) await new Promise((r) => setTimeout(r, 5));
|
||||
assert.deepEqual(board.notes.value, ["hello"]);
|
||||
assert.deepEqual(seen.at(-1), ["hello"]);
|
||||
await browser.close();
|
||||
|
||||
// WebSocket peers are always clients: one announcing itself as a host is cut off.
|
||||
const impostor = createHost({
|
||||
self: { kind: "host" },
|
||||
name: "session:S1",
|
||||
upstream: {
|
||||
link: webSocketLink(new WebSocket(`ws://127.0.0.1:${port}/?token=ok`) as unknown as WebSocketLike),
|
||||
principal: { kind: "server" },
|
||||
},
|
||||
onError: () => {},
|
||||
});
|
||||
await assert.rejects(
|
||||
impostor.root((ctx) => impostor.hold(Board, {}).add("x", ctx), {
|
||||
signal: AbortSignal.timeout(150),
|
||||
}),
|
||||
(error) => isChordError(error, "aborted"),
|
||||
);
|
||||
await impostor.close();
|
||||
|
||||
const refused = open("bad");
|
||||
const denied = refused.hold(Board, {});
|
||||
// The server closes unauthenticated sockets; calls wait for a reconnect until their ctx gives up.
|
||||
await assert.rejects(
|
||||
refused.root((ctx) => denied.add("x", ctx), { signal: AbortSignal.timeout(100) }),
|
||||
(error) => isChordError(error, "aborted"),
|
||||
);
|
||||
await refused.close();
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => wss.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
describe("WebSocket links: packed payloads", () => {
|
||||
it("large snapshots and results reach the client as binary messages; clients may not send them", async () => {
|
||||
const notes = Array.from({ length: 40 }, (_, i) => `${i}:${"x".repeat(4000)}`);
|
||||
const server = createHost({ self: { kind: "host" }, name: "server", onError: () => {} });
|
||||
server.serve(Board, (_input, _principal, self) => {
|
||||
const state = self.state("notes", Type.Array(Type.String()), notes);
|
||||
return { variant: "main", notes: state, add: () => {}, who: () => notes };
|
||||
});
|
||||
const wss = new WebSocketServer({ port: 0, host: "127.0.0.1" });
|
||||
await new Promise<void>((resolve) => wss.once("listening", () => resolve()));
|
||||
acceptWebSockets(wss as never, server, { principal: () => ({ guest: "web" }) });
|
||||
const url = `ws://127.0.0.1:${(wss.address() as { port: number }).port}/`;
|
||||
try {
|
||||
const binary: Uint8Array[] = [];
|
||||
const socket = new WebSocket(url);
|
||||
socket.addEventListener("message", (event) => {
|
||||
if (typeof event.data !== "string") binary.push(new Uint8Array(event.data as ArrayBuffer));
|
||||
});
|
||||
const browser = createHost({
|
||||
self: { kind: "host" },
|
||||
name: "browser",
|
||||
kind: "client",
|
||||
upstream: { link: webSocketLink(socket as unknown as WebSocketLike), principal: { kind: "server" } },
|
||||
onError: () => {},
|
||||
});
|
||||
const board = browser.hold(Board, {});
|
||||
assert.deepEqual(await board.who(), notes);
|
||||
board.notes.subscribe(() => {});
|
||||
for (let i = 0; i < 100 && board.notes.value === undefined; i++) await new Promise((r) => setTimeout(r, 5));
|
||||
assert.deepEqual(board.notes.value, notes);
|
||||
assert(binary.length >= 2, "the result and the snapshot came as binary messages");
|
||||
// A text-only transport (the TUI launcher) turns a packed message back into its text frame.
|
||||
const frames = binary.map((message) => JSON.parse(inflateMessage(message)) as Record<string, unknown>);
|
||||
assert(frames.some((frame) => frame.t === "result" && JSON.stringify(frame.value) === JSON.stringify(notes)));
|
||||
assert(frames.some((frame) => frame.t === "state" && JSON.stringify(frame.value) === JSON.stringify(notes)));
|
||||
await browser.close();
|
||||
|
||||
// A client's binary message is a protocol error: the server closes the socket.
|
||||
const raw = new WebSocket(url);
|
||||
raw.binaryType = "arraybuffer";
|
||||
await new Promise<void>((resolve) => raw.addEventListener("open", () => resolve()));
|
||||
const closed = new Promise<void>((resolve) => raw.addEventListener("close", () => resolve()));
|
||||
raw.send(new Uint8Array([0, 0, 0, 2, 123, 125]));
|
||||
await closed;
|
||||
} finally {
|
||||
await new Promise<void>((resolve) => wss.close(() => resolve()));
|
||||
}
|
||||
});
|
||||
});
|
||||
Reference in new issue
Block a user