Squashed 'third_party/pi/' content from commit d981de122

git-subtree-dir: third_party/pi
git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477

packages/coding-agent/test/session-share.test.ts

@@ -0,0 +1,88 @@
+import { EventEmitter } from "node:events";
+import { readFileSync, writeFileSync } from "node:fs";
+import { PassThrough } from "node:stream";
+import { beforeAll, describe, expect, it, vi } from "vitest";
+
+const childProcessMocks = vi.hoisted(() => ({
+	spawn: vi.fn(),
+	spawnSync: vi.fn(() => ({ status: 0 })),
+}));
+
+vi.mock("node:child_process", () => childProcessMocks);
+
+import { shareSession } from "../src/modes/interactive/session-share.ts";
+import { initTheme } from "../src/modes/interactive/theme/theme.ts";
+
+function deferred(): { promise: Promise<void>; resolve: () => void } {
+	let resolve!: () => void;
+	const promise = new Promise<void>((done) => {
+		resolve = done;
+	});
+	return { promise, resolve };
+}
+
+describe("shareSession", () => {
+	beforeAll(() => initTheme("dark"));
+
+	it("keeps concurrent session exports isolated", async () => {
+		const uploads: string[] = [];
+		childProcessMocks.spawn.mockImplementation((_command, args: string[]) => {
+			uploads.push(readFileSync(args.at(-1)!, "utf8"));
+			const child = Object.assign(new EventEmitter(), {
+				stdout: new PassThrough(),
+				stderr: new PassThrough(),
+				kill: vi.fn(),
+			});
+			queueMicrotask(() => {
+				child.stdout.end(`https://gist.github.com/test/${uploads.length}\n`);
+				child.stderr.end();
+				child.emit("close", 0);
+			});
+			return child;
+		});
+
+		const aWritten = deferred();
+		const bWritten = deferred();
+		const releaseB = deferred();
+		const errors: string[] = [];
+		const context = (name: "A" | "B") => ({
+			session: {
+				sessionManager: {
+					getSessionId: () => name,
+					getCwd: () => "/tmp",
+					getBranch: () => [],
+				},
+				state: { systemPrompt: name, tools: [] },
+				modelRuntime: { getProvider: () => undefined },
+				exportToHtml: async (filePath: string) => {
+					writeFileSync(filePath, name);
+					if (name === "A") {
+						aWritten.resolve();
+						await bWritten.promise;
+					} else {
+						bWritten.resolve();
+						await releaseB.promise;
+					}
+				},
+			},
+			ui: { setFocus() {}, requestRender() {} },
+			editorContainer: { clear() {}, addChild() {} },
+			editor: {},
+			showStatus() {},
+			showError(message: string) {
+				errors.push(message);
+			},
+		});
+
+		const shareA = shareSession(context("A") as never);
+		await aWritten.promise;
+		const shareB = shareSession(context("B") as never);
+		await bWritten.promise;
+		await shareA;
+		releaseB.resolve();
+		await shareB;
+
+		expect(uploads).toEqual(["A", "B"]);
+		expect(errors).toEqual([]);
+	});
+});