@openparachute/agent 0.2.3-rc.2 → 0.2.3-rc.3

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (54) hide show
  1. package/package.json +4 -1
  2. package/src/transports/vault.ts +19 -1
  3. package/src/_parked/interactive-spawn.test.ts +0 -324
  4. package/src/_parked/interactive-spawn.ts +0 -701
  5. package/src/agent-defs.test.ts +0 -1504
  6. package/src/agent-mcp-config.test.ts +0 -115
  7. package/src/agents.test.ts +0 -360
  8. package/src/auth.test.ts +0 -46
  9. package/src/backends/attached-queue.test.ts +0 -376
  10. package/src/backends/programmatic.test.ts +0 -1715
  11. package/src/backends/registry.test.ts +0 -1494
  12. package/src/backends/stream-json.test.ts +0 -570
  13. package/src/channel-backend-wiring.test.ts +0 -237
  14. package/src/credentials.test.ts +0 -274
  15. package/src/cron.test.ts +0 -342
  16. package/src/daemon-agent-def-api.test.ts +0 -166
  17. package/src/daemon-agent-defs-api.test.ts +0 -953
  18. package/src/daemon-agent-env-api.test.ts +0 -338
  19. package/src/daemon-attached-queue-store.test.ts +0 -65
  20. package/src/daemon-config-api.test.ts +0 -962
  21. package/src/daemon-jobs-api.test.ts +0 -271
  22. package/src/daemon-vault-chat.test.ts +0 -250
  23. package/src/daemon.test.ts +0 -746
  24. package/src/def-vaults.test.ts +0 -136
  25. package/src/delivery-state.test.ts +0 -110
  26. package/src/effective-env.test.ts +0 -114
  27. package/src/grants.test.ts +0 -638
  28. package/src/hub-jwt.test.ts +0 -161
  29. package/src/jobs.test.ts +0 -245
  30. package/src/mcp-http.test.ts +0 -265
  31. package/src/mint-token.test.ts +0 -152
  32. package/src/module-manifest.test.ts +0 -158
  33. package/src/programmatic-wiring.test.ts +0 -838
  34. package/src/registry.test.ts +0 -227
  35. package/src/resolve-port.test.ts +0 -64
  36. package/src/routing.test.ts +0 -184
  37. package/src/runner.test.ts +0 -506
  38. package/src/sandbox/config.test.ts +0 -150
  39. package/src/sandbox/egress.test.ts +0 -113
  40. package/src/sandbox/live-seatbelt.test.ts +0 -277
  41. package/src/sandbox/mounts.test.ts +0 -154
  42. package/src/sandbox/sandbox.test.ts +0 -168
  43. package/src/services-manifest.test.ts +0 -106
  44. package/src/spa-serve.test.ts +0 -116
  45. package/src/spawn-agent-cli.test.ts +0 -172
  46. package/src/spawn-agent.test.ts +0 -1218
  47. package/src/spawn-deps.test.ts +0 -54
  48. package/src/terminal-assets.test.ts +0 -50
  49. package/src/terminal.test.ts +0 -530
  50. package/src/transports/http-ui.test.ts +0 -455
  51. package/src/transports/telegram.test.ts +0 -174
  52. package/src/transports/vault.test.ts +0 -2012
  53. package/src/ui-kit.test.ts +0 -178
  54. package/web/ui/tsconfig.json +0 -21
@@ -1,54 +0,0 @@
1
- /**
2
- * Tests for `resolveSpawnDeps` (`src/spawn-deps.ts`) — the real-dep resolver
3
- * shared by the CLI and the web spawn endpoint.
4
- *
5
- * The load-bearing regression guard here is the claude config binding: the
6
- * sandboxed `claude` MUST get `~/.claude.json` bound read-only, or it runs
7
- * first-run onboarding whose connectivity check is FATAL under the restricted
8
- * egress proxy and the tmux session dies instantly ("An unknown error occurred").
9
- * That bug shipped once; this test ensures the binding stays.
10
- */
11
-
12
- import { describe, test, expect, afterEach } from "bun:test";
13
- import { mkdtempSync, writeFileSync, rmSync } from "node:fs";
14
- import { tmpdir, homedir } from "node:os";
15
- import { join, resolve } from "node:path";
16
- import { resolveSpawnDeps, SpawnDepsError } from "./spawn-deps.ts";
17
-
18
- const savedHome = process.env.PARACHUTE_HOME;
19
- let tmp: string | undefined;
20
-
21
- afterEach(() => {
22
- if (savedHome === undefined) delete process.env.PARACHUTE_HOME;
23
- else process.env.PARACHUTE_HOME = savedHome;
24
- if (tmp) {
25
- try { rmSync(tmp, { recursive: true, force: true }); } catch {}
26
- tmp = undefined;
27
- }
28
- });
29
-
30
- describe("resolveSpawnDeps", () => {
31
- test("throws SpawnDepsError when there's no operator token", () => {
32
- tmp = mkdtempSync(join(tmpdir(), "spawn-deps-empty-"));
33
- process.env.PARACHUTE_HOME = tmp; // no operator.token inside
34
- expect(() => resolveSpawnDeps()).toThrow(SpawnDepsError);
35
- });
36
-
37
- test("binds the claude binary (confined reads) but NOT the operator's ~/.claude", () => {
38
- tmp = mkdtempSync(join(tmpdir(), "spawn-deps-"));
39
- process.env.PARACHUTE_HOME = tmp;
40
- writeFileSync(join(tmp, "operator.token"), "fake-operator-bearer");
41
- const deps = resolveSpawnDeps();
42
- // The agent's config/onboarding now lives in its own per-session HOME
43
- // (seedAgentHome), so we no longer expose the operator's real config.
44
- expect(deps.runtimeReadOnly).not.toContain(resolve(homedir(), ".claude.json"));
45
- expect(deps.runtimeReadOnly).not.toContain(resolve(homedir(), ".claude"));
46
- // The claude BINARY is still bound (needed under confined/scoped reads) when
47
- // resolvable on PATH — and claudeBin is set to its absolute path.
48
- const bin = Bun.which("claude");
49
- if (bin) {
50
- expect(deps.claudeBin).toBe(bin);
51
- expect(deps.runtimeReadOnly).toContain(bin);
52
- }
53
- });
54
- });
@@ -1,50 +0,0 @@
1
- /**
2
- * Tests for same-origin terminal asset serving (`src/terminal-assets.ts`) — the
3
- * fix for "xterm.js failed to load (CDN blocked?)". Proves the vendored xterm JS +
4
- * CSS resolve from the installed packages and serve with the right content-type,
5
- * and that an unknown asset name is a clean miss (the daemon 404s it).
6
- */
7
-
8
- import { describe, test, expect } from "bun:test";
9
- import { serveTerminalAsset, TERMINAL_ASSET_NAMES } from "./terminal-assets.ts";
10
-
11
- describe("serveTerminalAsset", () => {
12
- test("serves xterm.js as JavaScript with a real body", async () => {
13
- const res = serveTerminalAsset("xterm.js");
14
- expect(res).not.toBeNull();
15
- expect(res!.status).toBe(200);
16
- expect(res!.headers.get("content-type")).toContain("javascript");
17
- const body = await res!.text();
18
- // The UMD bundle defines the Terminal global — a non-trivial body proves the
19
- // file resolved from the package (not an empty/placeholder response).
20
- expect(body.length).toBeGreaterThan(1000);
21
- expect(body).toContain("Terminal");
22
- });
23
-
24
- test("serves xterm.css as CSS", async () => {
25
- const res = serveTerminalAsset("xterm.css");
26
- expect(res!.status).toBe(200);
27
- expect(res!.headers.get("content-type")).toContain("css");
28
- expect((await res!.text()).length).toBeGreaterThan(100);
29
- });
30
-
31
- test("serves addon-fit.js", async () => {
32
- const res = serveTerminalAsset("addon-fit.js");
33
- expect(res!.status).toBe(200);
34
- expect(res!.headers.get("content-type")).toContain("javascript");
35
- });
36
-
37
- test("sets an immutable cache header (version-pinned)", () => {
38
- const res = serveTerminalAsset("xterm.js");
39
- expect(res!.headers.get("cache-control")).toContain("immutable");
40
- });
41
-
42
- test("an unknown asset name → null (caller 404s)", () => {
43
- expect(serveTerminalAsset("evil.js")).toBeNull();
44
- expect(serveTerminalAsset("../../etc/passwd")).toBeNull();
45
- });
46
-
47
- test("the known names are the three xterm assets", () => {
48
- expect(TERMINAL_ASSET_NAMES.sort()).toEqual(["addon-fit.js", "xterm.css", "xterm.js"]);
49
- });
50
- });
@@ -1,530 +0,0 @@
1
- /**
2
- * In-page terminal tests — the WS↔pty relay (`createTerminalWsHandlers`,
3
- * `parseControlFrame`, src/terminal.ts) + the daemon-side terminal upgrade gate
4
- * (`authorizeTerminalUpgrade`, src/daemon.ts).
5
- *
6
- * The relay tests drive the handler set against a FAKE `ServerWebSocket` and the
7
- * injectable `spawnTerminal` seam — no real tmux, no real pty, no `Bun.serve`.
8
- * The fake socket exposes a settable `getBufferedAmount()` so we can simulate
9
- * the hub's send-buffer filling under a flood and assert the flow-control
10
- * contract (the load-bearing item): a flood PARKS in the daemon-side coalesce
11
- * queue, the socket is NOT closed, and output resumes (drains) once the client
12
- * catches up. The hub's blunt 8 MiB cap therefore never has to fire.
13
- *
14
- * The auth tests call the pure `authorizeTerminalUpgrade` directly, using the
15
- * same sentinel-token `mock.module("./hub-jwt.ts")` harness daemon-config-api.
16
- * test.ts uses — accept paths run without a live hub/JWKS; the no-token reject
17
- * still hits the real short-circuit.
18
- */
19
- import { describe, test, expect, mock } from "bun:test";
20
- import type { ServerWebSocket } from "bun";
21
-
22
- // Sentinel tokens → fixed scope sets. Mirrors daemon-config-api.test.ts so this
23
- // process-wide mock stays compatible. ADMIN_TOKEN carries agent:admin (==
24
- // SCOPE_TERMINAL); READ_TOKEN is under-scoped; LEGACY_ADMIN_TOKEN carries the
25
- // pre-rename channel:admin scope (dual-accept back-compat); anything else throws.
26
- const ADMIN_TOKEN = "test-admin-token"; // agent:admin (== SCOPE_TERMINAL)
27
- const READ_TOKEN = "test-read-token"; // agent:read only (insufficient)
28
- const LEGACY_ADMIN_TOKEN = "test-legacy-admin-token"; // legacy channel:admin (still accepted)
29
- import { HubJwtError, looksLikeJwt } from "@openparachute/scope-guard";
30
- mock.module("./hub-jwt.ts", () => ({
31
- AGENT_AUDIENCE: "agent",
32
- CHANNEL_AUDIENCE: "channel", // deprecated alias kept for back-compat
33
- async validateHubJwt(token: string) {
34
- // New tokens carry aud:"agent" + agent:* scopes (the daemon mints/validates
35
- // agent now). A legacy token carries aud:"channel" + channel:* scopes — the
36
- // resource server dual-accepts both during the rename window.
37
- const base = { sub: "test", aud: "agent", jti: undefined, clientId: undefined, vaultScope: undefined };
38
- if (token === ADMIN_TOKEN) return { ...base, scopes: ["agent:read", "agent:send", "agent:admin"] };
39
- if (token === READ_TOKEN) return { ...base, scopes: ["agent:read"] };
40
- // Pre-rename token: legacy audience + legacy channel:* scopes. requireScope's
41
- // dual-accept (hasScope) must still authorize agent:admin from channel:admin.
42
- if (token === LEGACY_ADMIN_TOKEN)
43
- return { ...base, aud: "channel", scopes: ["channel:read", "channel:send", "channel:admin"] };
44
- throw new HubJwtError("issuer", "invalid token");
45
- },
46
- HubJwtError,
47
- looksLikeJwt,
48
- resetJwksCache() {},
49
- resetRevocationCache() {},
50
- }));
51
-
52
- import {
53
- createTerminalWsHandlers,
54
- tmuxAttachEnv,
55
- parseControlFrame,
56
- HUB_WS_CAP_BYTES,
57
- PAUSE_FRAC,
58
- RESUME_FRAC,
59
- type TerminalWsData,
60
- type SpawnTerminalFn,
61
- type SpawnedTerminal,
62
- type BunTerminal,
63
- } from "./terminal.ts";
64
- import { authorizeTerminalUpgrade } from "./daemon.ts";
65
- import type { Channel } from "./registry.ts";
66
-
67
- // ===========================================================================
68
- // Fakes — a recording pty + a controllable ServerWebSocket.
69
- // ===========================================================================
70
-
71
- /** A fake pty that records writes/resizes and lets the test push pty output. */
72
- class FakePty implements BunTerminal {
73
- writes: Uint8Array[] = [];
74
- writeStrings: string[] = [];
75
- resizes: Array<{ cols: number; rows: number }> = [];
76
- closed = false;
77
- /** Captured callbacks from the spawn opts — the test drives pty output via these. */
78
- onData!: (bytes: Uint8Array) => void;
79
- onExit!: () => void;
80
-
81
- write(data: string | ArrayBufferView | ArrayBufferLike): number {
82
- if (typeof data === "string") {
83
- this.writeStrings.push(data);
84
- return data.length;
85
- }
86
- const u8 = data instanceof Uint8Array ? data : new Uint8Array(data as ArrayBufferLike);
87
- this.writes.push(u8);
88
- return u8.byteLength;
89
- }
90
- resize(cols: number, rows: number): void {
91
- this.resizes.push({ cols, rows });
92
- }
93
- close(): void {
94
- this.closed = true;
95
- }
96
- }
97
-
98
- interface FakeProc {
99
- readonly exited: Promise<number>;
100
- killed: boolean;
101
- kill(signal?: number | string): void;
102
- }
103
-
104
- /** A fake ServerWebSocket: records sends/close, settable buffered depth. */
105
- class FakeWs {
106
- data: TerminalWsData;
107
- sent: Uint8Array[] = [];
108
- closeCalls: Array<{ code?: number; reason?: string }> = [];
109
- /** Test sets this to simulate the hub send-buffer depth. */
110
- buffered = 0;
111
-
112
- constructor(data: TerminalWsData) {
113
- this.data = data;
114
- }
115
- send(bytes: Uint8Array | string): number {
116
- if (typeof bytes === "string") {
117
- this.sent.push(new TextEncoder().encode(bytes));
118
- return bytes.length;
119
- }
120
- this.sent.push(bytes);
121
- return bytes.byteLength;
122
- }
123
- getBufferedAmount(): number {
124
- return this.buffered;
125
- }
126
- close(code?: number, reason?: string): void {
127
- this.closeCalls.push({ code, reason });
128
- }
129
- get totalSentBytes(): number {
130
- return this.sent.reduce((n, b) => n + b.byteLength, 0);
131
- }
132
- }
133
-
134
- /** Build a fake-pty spawn seam; returns the seam + a handle on the created pty. */
135
- function makeSpawn(): { spawn: SpawnTerminalFn; ptyRef: { current?: FakePty }; procRef: { current?: FakeProc } } {
136
- const ptyRef: { current?: FakePty } = {};
137
- const procRef: { current?: FakeProc } = {};
138
- const spawn: SpawnTerminalFn = (_session, opts) => {
139
- const pty = new FakePty();
140
- pty.onData = opts.onData;
141
- pty.onExit = opts.onExit;
142
- const proc: FakeProc = { exited: Promise.resolve(0), killed: false, kill() { this.killed = true; } };
143
- ptyRef.current = pty;
144
- procRef.current = proc;
145
- return { terminal: pty, proc } as unknown as SpawnedTerminal;
146
- };
147
- return { spawn, ptyRef, procRef };
148
- }
149
-
150
- function wsData(over: Partial<TerminalWsData> = {}): TerminalWsData {
151
- return { session: "eng-agent", channel: "eng", cols: 80, rows: 24, ...over };
152
- }
153
-
154
- /** Open a handler set + fake ws + fake pty, returning the lot for a test. */
155
- function setup(opts: { capBytes?: number } = {}) {
156
- const { spawn, ptyRef, procRef } = makeSpawn();
157
- const warnings: string[] = [];
158
- const handlers = createTerminalWsHandlers({
159
- spawnTerminal: spawn,
160
- capBytes: opts.capBytes,
161
- logger: { warn: (...a: unknown[]) => warnings.push(a.map(String).join(" ")) },
162
- });
163
- const ws = new FakeWs(wsData());
164
- handlers.open(ws as unknown as ServerWebSocket<TerminalWsData>);
165
- return { handlers, ws, pty: () => ptyRef.current!, proc: () => procRef.current!, warnings };
166
- }
167
-
168
- // Cast helper — drive a handler with the fake ws.
169
- const asWs = (ws: FakeWs) => ws as unknown as ServerWebSocket<TerminalWsData>;
170
-
171
- // ===========================================================================
172
- // parseControlFrame — unit cases
173
- // ===========================================================================
174
- describe("parseControlFrame", () => {
175
- test("a valid resize frame parses + clamps", () => {
176
- expect(parseControlFrame(JSON.stringify({ type: "resize", cols: 120, rows: 40 }))).toEqual({
177
- type: "resize",
178
- cols: 120,
179
- rows: 40,
180
- });
181
- });
182
-
183
- test("out-of-range dims clamp to [1, 9999]", () => {
184
- expect(parseControlFrame(JSON.stringify({ type: "resize", cols: 0, rows: 999999 }))).toEqual({
185
- type: "resize",
186
- cols: 1,
187
- rows: 9999,
188
- });
189
- });
190
-
191
- test("malformed JSON → null (forwarded as input by the caller)", () => {
192
- expect(parseControlFrame("{not json")).toBeNull();
193
- });
194
-
195
- test("a non-resize control type → null", () => {
196
- expect(parseControlFrame(JSON.stringify({ type: "evil", cmd: "rm -rf /" }))).toBeNull();
197
- });
198
-
199
- test("resize with non-finite dims → null", () => {
200
- expect(parseControlFrame(JSON.stringify({ type: "resize", cols: "80", rows: NaN }))).toBeNull();
201
- expect(parseControlFrame(JSON.stringify({ type: "resize" }))).toBeNull();
202
- });
203
-
204
- test("a bare JSON scalar / array (not an object) → null", () => {
205
- expect(parseControlFrame("42")).toBeNull();
206
- expect(parseControlFrame("null")).toBeNull();
207
- expect(parseControlFrame("[1,2,3]")).toBeNull();
208
- });
209
- });
210
-
211
- // ===========================================================================
212
- // Relay — both directions
213
- // ===========================================================================
214
- describe("relay — pty ↔ ws", () => {
215
- test("pty output → ws.send (binary), bytes preserved", () => {
216
- const { ws, pty } = setup();
217
- const chunk = new Uint8Array([0x68, 0x69]); // "hi"
218
- pty().onData(chunk);
219
- expect(ws.sent).toHaveLength(1);
220
- expect(Array.from(ws.sent[0]!)).toEqual([0x68, 0x69]);
221
- });
222
-
223
- test("ws binary message → terminal.write (keystrokes to the pty)", () => {
224
- const { handlers, ws, pty } = setup();
225
- const keys = Buffer.from("ls\n");
226
- handlers.message(asWs(ws), keys);
227
- expect(pty().writes).toHaveLength(1);
228
- expect(Buffer.from(pty().writes[0]!).toString()).toBe("ls\n");
229
- });
230
-
231
- test("a text frame that ISN'T a control frame is forwarded to the pty as input (fail-safe)", () => {
232
- const { handlers, ws, pty } = setup();
233
- handlers.message(asWs(ws), "plain text");
234
- expect(pty().writeStrings).toEqual(["plain text"]);
235
- expect(pty().resizes).toHaveLength(0);
236
- });
237
- });
238
-
239
- // ===========================================================================
240
- // Control frame — resize
241
- // ===========================================================================
242
- describe("control frame — resize", () => {
243
- test("a {type:resize} text frame → terminal.resize(cols, rows), NOT written as input", () => {
244
- const { handlers, ws, pty } = setup();
245
- handlers.message(asWs(ws), JSON.stringify({ type: "resize", cols: 100, rows: 30 }));
246
- expect(pty().resizes).toEqual([{ cols: 100, rows: 30 }]);
247
- expect(pty().writes).toHaveLength(0);
248
- expect(pty().writeStrings).toHaveLength(0);
249
- });
250
- });
251
-
252
- // ===========================================================================
253
- // THE load-bearing test — backpressure flood stays alive
254
- // ===========================================================================
255
- describe("backpressure — a flood parks in the queue, never closes the socket", () => {
256
- // A tiny cap so the test floods it cheaply. pause @ 50% (500), resume @ 25% (250).
257
- const CAP = 1000;
258
-
259
- test("a huge burst while the socket buffer is high → paused + coalesced, socket NOT closed, hub cap never approached", () => {
260
- const { ws, pty, proc } = setup({ capBytes: CAP });
261
-
262
- // Simulate the hub's send buffer already full (over the pause mark) and never
263
- // draining during the burst — exactly the "a build log floods xterm" case.
264
- ws.buffered = CAP; // 1000 ≥ pauseAt(500)
265
-
266
- // First chunk goes out, then the post-send buffered check trips pause.
267
- pty().onData(new Uint8Array(64));
268
- // Subsequent chunks while paused MUST queue, not send.
269
- const FLOOD_CHUNK = new Uint8Array(64 * 1024); // 64 KiB each
270
- for (let i = 0; i < 200; i++) pty().onData(FLOOD_CHUNK); // ~12.8 MiB of pty output
271
-
272
- // The socket is ALIVE — no close (no 1011, no 1013), pty alive, proc alive.
273
- expect(ws.closeCalls).toHaveLength(0);
274
- expect(pty().closed).toBe(false);
275
- expect(proc().killed).toBe(false);
276
-
277
- // Our OWN buffered bytes (what the hub sees) never grew past the one chunk we
278
- // sent before pausing — the flood parked daemon-side, NOT in the hub buffer.
279
- // i.e. we never let the socket approach the 8 MiB hub cap.
280
- expect(ws.totalSentBytes).toBeLessThanOrEqual(64);
281
- expect(ws.totalSentBytes).toBeLessThan(HUB_WS_CAP_BYTES);
282
- });
283
-
284
- test("when the client drains, the queued flood is eventually delivered (resume)", () => {
285
- const { handlers, ws, pty } = setup({ capBytes: CAP });
286
-
287
- ws.buffered = CAP; // over pause mark
288
- pty().onData(new Uint8Array(64)); // sent → then pause
289
- // Queue a bounded flood (well under MAX_QUEUE_BYTES so it parks, not closes).
290
- const N = 50;
291
- const CHUNK = 64; // bytes
292
- for (let i = 0; i < N; i++) pty().onData(new Uint8Array(CHUNK).fill(i));
293
- expect(ws.closeCalls).toHaveLength(0);
294
- const sentBeforeDrain = ws.sent.length;
295
-
296
- // Client catches up: buffer empties. Bun fires drain repeatedly as it clears;
297
- // drain flushes the queue while buffered stays under the pause mark.
298
- ws.buffered = 0;
299
- // A few drain cycles to flush the whole queue (flushQueue drains until empty
300
- // or the pause mark; with buffered=0 it empties in one pass, but loop to be safe).
301
- for (let i = 0; i < 5 && ws.sent.length < sentBeforeDrain + N; i++) {
302
- handlers.drain(asWs(ws));
303
- }
304
-
305
- // Everything queued got delivered — the flood was buffered, not dropped.
306
- expect(ws.sent.length).toBe(sentBeforeDrain + N);
307
- expect(ws.closeCalls).toHaveLength(0);
308
- // And the very last queued chunk arrived intact (content preserved through the queue).
309
- const last = ws.sent[ws.sent.length - 1]!;
310
- expect(last.byteLength).toBe(CHUNK);
311
- expect(last[0]).toBe(N - 1);
312
- });
313
- });
314
-
315
- // ===========================================================================
316
- // Hysteresis — pause @ 50%, resume only under 25%
317
- // ===========================================================================
318
- describe("hysteresis — pause @ PAUSE_FRAC, resume only under RESUME_FRAC", () => {
319
- const CAP = 1000; // pauseAt = 500, resumeAt = 250
320
-
321
- test("constants are 0.5 / 0.25 (the documented margins)", () => {
322
- expect(PAUSE_FRAC).toBe(0.5);
323
- expect(RESUME_FRAC).toBe(0.25);
324
- });
325
-
326
- test("does NOT pause while buffered stays under the pause mark", () => {
327
- const { ws, pty } = setup({ capBytes: CAP });
328
- ws.buffered = 400; // < pauseAt(500)
329
- pty().onData(new Uint8Array(10));
330
- pty().onData(new Uint8Array(10)); // still forwarded, not queued
331
- expect(ws.sent).toHaveLength(2);
332
- expect(ws.closeCalls).toHaveLength(0);
333
- });
334
-
335
- test("pauses at the 50% mark; a drain to BETWEEN resume(25%) and pause(50%) does NOT resume", () => {
336
- const { handlers, ws, pty } = setup({ capBytes: CAP });
337
- ws.buffered = 500; // == pauseAt → pause after the send
338
- pty().onData(new Uint8Array(10));
339
- // Now paused: a new pty chunk queues rather than sends.
340
- pty().onData(new Uint8Array(10));
341
- expect(ws.sent).toHaveLength(1); // only the first went out
342
-
343
- // Drain partway — buffer at 300 (still > resumeAt 250). Queue flushes only if
344
- // under the pause mark (300 < 500 so it flushes), but we stay PAUSED for live
345
- // forwarding because we're not yet under the resume mark.
346
- ws.buffered = 300;
347
- handlers.drain(asWs(ws));
348
- // A fresh live chunk should STILL queue (paused), proving no premature resume.
349
- const sentAfterDrain = ws.sent.length;
350
- pty().onData(new Uint8Array(10));
351
- expect(ws.sent.length).toBe(sentAfterDrain); // still paused → queued, not sent
352
- });
353
-
354
- test("resumes only once buffered falls under the 25% mark (and the queue is empty)", () => {
355
- const { handlers, ws, pty } = setup({ capBytes: CAP });
356
- ws.buffered = 600; // pause
357
- pty().onData(new Uint8Array(10));
358
- pty().onData(new Uint8Array(10)); // queued
359
-
360
- // Drain fully under the resume mark → drain flushes the queue + resumes.
361
- ws.buffered = 100; // < resumeAt(250)
362
- handlers.drain(asWs(ws));
363
- // Now resumed: a new live chunk forwards immediately.
364
- const before = ws.sent.length;
365
- pty().onData(new Uint8Array(10));
366
- expect(ws.sent.length).toBe(before + 1);
367
- });
368
- });
369
-
370
- // ===========================================================================
371
- // 1013 — a hopelessly-stuck client is shed
372
- // ===========================================================================
373
- describe("stuck client — closed 1013 when the queue blows past MAX_QUEUE_BYTES", () => {
374
- // Keep the hub cap tiny so we pause immediately; the queue cap (16 MiB) is the
375
- // line under test. Flood past 16 MiB while the socket never drains.
376
- const CAP = 1000;
377
-
378
- test(">16 MiB queued while the socket stays full → close(1013), pty + viewer torn down", () => {
379
- const { ws, pty, proc, warnings } = setup({ capBytes: CAP });
380
- ws.buffered = CAP; // perpetually over the pause mark → everything queues
381
- pty().onData(new Uint8Array(64)); // first send trips pause
382
-
383
- // 17 MiB in 1 MiB chunks — crosses the 16 MiB MAX_QUEUE_BYTES line.
384
- const MiB = new Uint8Array(1024 * 1024);
385
- for (let i = 0; i < 17; i++) pty().onData(MiB);
386
-
387
- const closed1013 = ws.closeCalls.find((c) => c.code === 1013);
388
- expect(closed1013).toBeDefined();
389
- expect(pty().closed).toBe(true); // pty closed
390
- expect(proc().killed).toBe(true); // viewer process killed
391
- expect(warnings.some((w) => w.includes("too far behind") || w.includes("too slow"))).toBe(true);
392
- });
393
- });
394
-
395
- // ===========================================================================
396
- // Lifecycle — pty exit + client close tear down the viewer (session lives on)
397
- // ===========================================================================
398
- describe("lifecycle — teardown", () => {
399
- test("pty exit closes the socket with a clean 1000 (session ended)", () => {
400
- const { ws, pty, proc } = setup();
401
- pty().onExit();
402
- expect(ws.closeCalls.some((c) => c.code === 1000)).toBe(true);
403
- expect(pty().closed).toBe(true);
404
- expect(proc().killed).toBe(true);
405
- });
406
-
407
- test("client close releases the pty + kills the viewer proc (idempotent)", () => {
408
- const { handlers, ws, pty, proc } = setup();
409
- handlers.close(asWs(ws));
410
- expect(pty().closed).toBe(true);
411
- expect(proc().killed).toBe(true);
412
- // Idempotent — a second close is a no-op (no throw, no double behavior change).
413
- expect(() => handlers.close(asWs(ws))).not.toThrow();
414
- });
415
- });
416
-
417
- // ===========================================================================
418
- // tmuxAttachEnv — TERM wiring for the `tmux attach` viewer
419
- // ===========================================================================
420
- describe("tmuxAttachEnv — forces a real TERM so `tmux attach` finds terminfo", () => {
421
- test("sets TERM=xterm-256color (else tmux aborts 'does not support clear')", () => {
422
- expect(tmuxAttachEnv({ PATH: "/usr/bin" }).TERM).toBe("xterm-256color");
423
- });
424
- test("preserves the inherited env + overrides a stale/empty TERM", () => {
425
- const env = tmuxAttachEnv({ PATH: "/usr/bin", HOME: "/home/op", TERM: "" });
426
- expect(env.PATH).toBe("/usr/bin");
427
- expect(env.HOME).toBe("/home/op");
428
- expect(env.TERM).toBe("xterm-256color"); // empty/missing TERM is the daemon's reality
429
- });
430
- });
431
-
432
- // ===========================================================================
433
- // Daemon-side terminal auth — authorizeTerminalUpgrade (pure fn)
434
- // ===========================================================================
435
- describe("authorizeTerminalUpgrade — operator-gated agent:admin", () => {
436
- function channels(names: string[] = ["eng"]): Map<string, Channel> {
437
- const m = new Map<string, Channel>();
438
- for (const name of names) {
439
- m.set(name, { name, transport: {} as Channel["transport"], entry: { name, transport: "vault" } });
440
- }
441
- return m;
442
- }
443
- function req(query = ""): { req: Request; url: URL } {
444
- const url = new URL(`http://127.0.0.1/terminal/eng${query}`);
445
- return { req: new Request(url, { headers: { upgrade: "websocket" } }), url };
446
- }
447
-
448
- test("missing token → reject (401), no session leaked", async () => {
449
- const { req: r, url } = req();
450
- const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
451
- expect(d.ok).toBe(false);
452
- if (!d.ok) expect(d.response.status).toBe(401);
453
- });
454
-
455
- test("bad/expired token → reject (401)", async () => {
456
- const { req: r, url } = req("?token=garbage-not-a-real-jwt");
457
- const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
458
- expect(d.ok).toBe(false);
459
- if (!d.ok) expect(d.response.status).toBe(401);
460
- });
461
-
462
- test("under-scoped token (agent:read, not agent:admin) → reject (403)", async () => {
463
- const { req: r, url } = req(`?token=${READ_TOKEN}`);
464
- const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
465
- expect(d.ok).toBe(false);
466
- if (!d.ok) expect(d.response.status).toBe(403);
467
- });
468
-
469
- test("valid agent:admin token → ok, with the right tmux session + geometry", async () => {
470
- const { req: r, url } = req(`?token=${ADMIN_TOKEN}&cols=120&rows=40`);
471
- const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
472
- expect(d.ok).toBe(true);
473
- if (d.ok) {
474
- expect(d.channel).toBe("eng");
475
- expect(d.session).toBe("eng-agent"); // <name>-agent (launch-session.sh:38)
476
- expect(d.cols).toBe(120);
477
- expect(d.rows).toBe(40);
478
- }
479
- });
480
-
481
- test("legacy channel:admin token still authorizes (dual-accept back-compat)", async () => {
482
- // A token minted before the channel→agent rename carries channel:admin (and
483
- // aud:"channel"). requireScope's dual-accept (hasScope) must still authorize
484
- // the agent:admin-gated terminal until live tokens are re-minted.
485
- const { req: r, url } = req(`?token=${LEGACY_ADMIN_TOKEN}&cols=100&rows=30`);
486
- const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
487
- expect(d.ok).toBe(true);
488
- if (d.ok) {
489
- expect(d.session).toBe("eng-agent");
490
- expect(d.cols).toBe(100);
491
- expect(d.rows).toBe(30);
492
- }
493
- });
494
-
495
- test("geometry defaults (80×24) when cols/rows absent or out-of-range", async () => {
496
- const { req: r, url } = req(`?token=${ADMIN_TOKEN}&cols=0&rows=abc`);
497
- const d = await authorizeTerminalUpgrade(r, url, channels(), "eng");
498
- expect(d.ok).toBe(true);
499
- if (d.ok) {
500
- expect(d.cols).toBe(80);
501
- expect(d.rows).toBe(24);
502
- }
503
- });
504
-
505
- test("an agent name that isn't a configured channel is ACCEPTED → its session (agents have own names)", async () => {
506
- // The terminal attaches to an AGENT (tmux <name>-agent), not a channel — so a
507
- // valid-slug name that isn't in the channel map is accepted (a non-existent
508
- // session just fails to attach downstream with a clean 1000, no 404 here).
509
- const url = new URL(`http://127.0.0.1/terminal/weaver?token=${ADMIN_TOKEN}`);
510
- const r = new Request(url, { headers: { upgrade: "websocket" } });
511
- const d = await authorizeTerminalUpgrade(r, url, channels(["eng"]), "weaver");
512
- expect(d.ok).toBe(true);
513
- if (d.ok) expect(d.session).toBe("weaver-agent");
514
- });
515
-
516
- test("a path-traversal-shaped name is rejected by the slug guard → 400 (no escape into tmux -t)", async () => {
517
- const weird = "../../etc";
518
- const url = new URL(`http://127.0.0.1/terminal/${encodeURIComponent(weird)}?token=${ADMIN_TOKEN}`);
519
- const r = new Request(url, { headers: { upgrade: "websocket" } });
520
- const d = await authorizeTerminalUpgrade(r, url, channels(["eng"]), weird);
521
- expect(d.ok).toBe(false);
522
- if (!d.ok) expect(d.response.status).toBe(400);
523
- });
524
- });
525
-
526
- // Touch HUB_WS_CAP_BYTES so the import is load-bearing if the relationship to the
527
- // hub cap ever drifts (the constant mirrors the hub's DEFAULT_MAX_BUFFERED_BYTES).
528
- test("HUB_WS_CAP_BYTES mirrors the hub's 8 MiB cap", () => {
529
- expect(HUB_WS_CAP_BYTES).toBe(8 * 1024 * 1024);
530
- });