Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions packages/safe-bash-playground/src/engine/kernel.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ describe("real safe-bash browser kernel", () => {
const activeWorkers = new Set<{ terminate(): void }>();

beforeAll(async () => {
vi.stubGlobal("navigator", { language: "en-US" });
vi.stubGlobal(
"Worker",
class extends EventTarget {
Expand All @@ -28,6 +29,7 @@ describe("real safe-bash browser kernel", () => {
const worker = new NodeWorker(
`
const { parentPort } = require('node:worker_threads');
Object.defineProperty(globalThis, 'navigator', { configurable: true, value: { language: 'en-US' } });
globalThis.addEventListener = (event, handler) => parentPort.on(event, data => handler({ data }));
globalThis.postMessage = (value, transfer) => parentPort.postMessage(value, transfer);
${code}
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,11 @@
import { describe, expect, it, vi } from "vitest";
import { afterAll, describe, expect, it, vi } from "vitest";
import { createMemoryFileSystem, FsError } from "./engine/index.js";
import type { FileSystem } from "./engine/index.js";
import { decodeError, encodeError, hostFileSystem, remoteFileSystem } from "./execution-filesystem.js";

vi.hoisted(() => vi.stubGlobal("navigator", { language: "en-US" }));
afterAll(() => vi.unstubAllGlobals());

vi.mock("./engine/index.js", async () => {
const { buildBrowserEngine } = await import("./engine/build-plugin.mjs");
const built = await buildBrowserEngine();
Expand Down
2 changes: 2 additions & 0 deletions packages/safe-bash-playground/src/execution.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ import type { ExecutionMessage } from "./execution-protocol.js";
import { browserWorkerFixture } from "../test/browser-worker.js";
import { setTimeout as delay } from "node:timers/promises";

vi.hoisted(() => vi.stubGlobal("navigator", { language: "en-US" }));

vi.mock("virtual:safe-bash-worker-sources", async () => {
const { buildBrowserEngine } = await import("./engine/build-plugin.mjs");
return { sources: (await buildBrowserEngine({ workersOnly: true })).workerSources };
Expand Down
2 changes: 2 additions & 0 deletions packages/safe-bash-playground/src/session.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@ import { createSession, SESSION_LIMITS } from "./session.js";
import { sampleFiles } from "./samples.js";
import { browserWorkerFixture } from "../test/browser-worker.js";

vi.hoisted(() => vi.stubGlobal("navigator", { language: "en-US" }));

vi.mock("virtual:safe-bash-worker-sources", async () => {
const { buildBrowserEngine } = await import("./engine/build-plugin.mjs");
return { sources: (await buildBrowserEngine({ workersOnly: true })).workerSources };
Expand Down
1 change: 1 addition & 0 deletions packages/safe-bash-playground/test/browser-worker.ts
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ export function browserWorkerFixture(executionSource: string) {
this.worker = source.then((code) => {
const worker = new NodeWorker(`
const { parentPort } = require("node:worker_threads");
Object.defineProperty(globalThis, "navigator", { configurable: true, value: { language: "en-US" } });
globalThis.addEventListener = (event, handler) => parentPort.on(event, data => handler({ data }));
globalThis.postMessage = value => parentPort.postMessage(value);
(() => { ${code} })();
Expand Down
12 changes: 9 additions & 3 deletions packages/toolcraft/src/stream.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,12 @@ import {
type ToolcraftStream
} from "./index.js";

async function collect<Value>(stream: AsyncIterable<Value>): Promise<Value[]> {
const values: Value[] = [];
for await (const value of stream) values.push(value);
return values;
}

describe("defineStreamCommand SDK lifecycle", () => {
it("starts lazily and advances only when the consumer pulls", async () => {
const produced: number[] = [];
Expand Down Expand Up @@ -78,7 +84,7 @@ describe("defineStreamCommand SDK lifecycle", () => {

const stream = sdk.watch({}, { onStatus: (event) => statuses.push(event) });

await expect(Array.fromAsync(stream)).resolves.toEqual([{ state: "fresh-token" }]);
await expect(collect(stream)).resolves.toEqual([{ state: "fresh-token" }]);
expect(statuses).toEqual([
{ type: "reconnecting", message: "Refreshing credentials" }
]);
Expand All @@ -95,7 +101,7 @@ describe("defineStreamCommand SDK lifecycle", () => {
});
const sdk = createSDK(defineGroup({ name: "devices", children: [watch] }));

await expect(Array.fromAsync(sdk.watch({}))).rejects.toThrow("state");
await expect(collect(sdk.watch({}))).rejects.toThrow("state");
});

it("propagates terminal errors and releases resources once", async () => {
Expand All @@ -115,7 +121,7 @@ describe("defineStreamCommand SDK lifecycle", () => {
});
const sdk = createSDK(defineGroup({ name: "devices", children: [watch] }));

await expect(Array.fromAsync(sdk.watch({}))).rejects.toThrow("connection lost");
await expect(collect(sdk.watch({}))).rejects.toThrow("connection lost");
expect(cleanup).toHaveBeenCalledOnce();
});
});
Loading