Squashed 'src/pi/' content from commit d981de122

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

packages/agent/test/harness/jsonl-storage-conformance.test.ts

@@ -0,0 +1,48 @@
+import { describe, it } from "vitest";
+import { BACKGROUND_CONTEXT } from "../../src/harness/context.ts";
+import { NodeExecutionEnv } from "../../src/harness/env/nodejs.ts";
+import { JSONL_FORMAT_VERSION, JsonlStorage } from "../../src/harness/session/jsonl/index.ts";
+import {
+	type ConformanceCase,
+	createStorageConformance,
+	type StorageFixture,
+} from "../../src/harness/session/testing/index.ts";
+import { createTempDir } from "./session-test-utils.ts";
+
+const NOW = 1_700_000_000_000;
+
+function registerConformance(name: string, cases: readonly ConformanceCase[]): void {
+	describe(name, () => {
+		for (const group of new Set(cases.map((testCase) => testCase.group))) {
+			describe(group, () => {
+				for (const testCase of cases.filter((candidate) => candidate.group === group)) {
+					it(testCase.name, () => testCase.run());
+				}
+			});
+		}
+	});
+}
+
+registerConformance(
+	"JsonlStorage conformance",
+	createStorageConformance(async () => {
+		const fileSystem = new NodeExecutionEnv({ cwd: createTempDir() });
+		const storage = await JsonlStorage.create(
+			{ fileSystem, path: "session.jsonl", now: () => NOW },
+			{
+				v: JSONL_FORMAT_VERSION,
+				kind: "header",
+				id: "session",
+				storageVersion: 1,
+				createdAt: NOW,
+				cwd: "/workspace",
+			},
+			[],
+			BACKGROUND_CONTEXT,
+		);
+		return {
+			storage,
+			[Symbol.asyncDispose]: () => storage.close(BACKGROUND_CONTEXT),
+		} satisfies StorageFixture;
+	}),
+);