alexion
67b361f0e9
git-subtree-dir: third_party/pi git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477
62 lines
1.8 KiB
TypeScript
62 lines
1.8 KiB
TypeScript
import { BACKGROUND_CONTEXT } from "@earendil-works/pi-agent-core";
|
|
import {
|
|
type ConformanceCase,
|
|
createStorageConformance,
|
|
type StorageFixture,
|
|
} from "@earendil-works/pi-agent-core/harness/session/testing";
|
|
import { describe, it } from "vitest";
|
|
import { createNodeSqliteFactory, SQLITE_STORAGE_VERSION, SqliteStorage, sql } from "../src/index.ts";
|
|
import { applyInitialSchema } from "../src/sqlite/migrations.ts";
|
|
|
|
const SESSION_ID = "session";
|
|
const NOW = 1_700_000_000_000;
|
|
const EMPTY_USAGE = {
|
|
input: 0,
|
|
output: 0,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
totalTokens: 0,
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
};
|
|
|
|
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(
|
|
"SqliteStorage conformance",
|
|
createStorageConformance(async () => {
|
|
const db = await createNodeSqliteFactory().open(":memory:");
|
|
try {
|
|
await applyInitialSchema(db);
|
|
sql`INSERT INTO sessions
|
|
(id, created_at, parent_session_id, storage_version, metadata, message_count, usage_payload, next_seq)
|
|
VALUES (${SESSION_ID}, ${NOW}, ${null}, ${SQLITE_STORAGE_VERSION}, ${null}, ${0}, ${JSON.stringify(EMPTY_USAGE)}, ${1})`.run(
|
|
db,
|
|
);
|
|
const storage = new SqliteStorage(db, { sessionId: SESSION_ID, now: () => NOW });
|
|
return {
|
|
storage,
|
|
async [Symbol.asyncDispose]() {
|
|
try {
|
|
await storage.close(BACKGROUND_CONTEXT);
|
|
} finally {
|
|
db.close();
|
|
}
|
|
},
|
|
} satisfies StorageFixture;
|
|
} catch (error) {
|
|
db.close();
|
|
throw error;
|
|
}
|
|
}),
|
|
);
|