c5cefd622b
Record genuine Code Mode use and full model usage rather than response-only estimates. Describe return shapes, permit byte-budgeted internal C reads, separate guest CPU from analysis deadlines, and move subsequent model profiles outside the source tree. Verify lease, admission, cleanup-retry and cancellation semantics. Co-authored-by: gensokyo-dev <342580529+gensokyo-dev@users.noreply.github.com> Co-authored-by: Kaoru0822 <208281322+kaoru0822-kitauji@users.noreply.github.com>
87 lines
2.9 KiB
JavaScript
87 lines
2.9 KiB
JavaScript
import assert from "node:assert/strict";
|
|
import test from "node:test";
|
|
import Ajv2020 from "ajv/dist/2020.js";
|
|
import { AtomicSession, reference } from "./api.mjs";
|
|
|
|
const factory = () => {
|
|
let sequence = 0;
|
|
const calls = [];
|
|
const create = async (path) => ({
|
|
context: { lease: `lease-${++sequence}`, selected_path: path },
|
|
close: async () => {
|
|
calls.push("close");
|
|
},
|
|
cancel: () => {
|
|
calls.push("cancel");
|
|
},
|
|
strings: async (input) => {
|
|
calls.push(input);
|
|
return { items: [], source: { evidence_id: "ev_real-owner" } };
|
|
},
|
|
});
|
|
return { create, calls };
|
|
};
|
|
test("named contracts reject extra keys and stale leases before provider admission", async () => {
|
|
const provider = factory();
|
|
const session = new AtomicSession(provider.create);
|
|
await session.open("first");
|
|
const lease = session.context.lease;
|
|
const invalid = await session.call("strings", { query: "x", guessed: true });
|
|
assert.equal(invalid.error.code, "invalid_input");
|
|
const wrongIdentity = await session.call("relations", {
|
|
snapshot: "snap_test",
|
|
kind: "incoming",
|
|
seeds: ["0"],
|
|
});
|
|
assert.equal(wrongIdentity.error.code, "invalid_input");
|
|
assert.equal(provider.calls.length, 0);
|
|
await session.open("second");
|
|
const stale = await session.call("strings", { query: "x" }, undefined, lease);
|
|
assert.equal(stale.error.code, "stale_context");
|
|
assert.deepEqual(provider.calls, ["close"]);
|
|
await session.close();
|
|
});
|
|
test("close drains admitted reads and a failed cleanup retains ownership for retry", async () => {
|
|
let release;
|
|
let reads = 0;
|
|
let closes = 0;
|
|
const session = new AtomicSession(async () => ({
|
|
context: { lease: "owned" },
|
|
strings: () => {
|
|
reads++;
|
|
return new Promise((resolve) => {
|
|
release = resolve;
|
|
});
|
|
},
|
|
close: () => {
|
|
if (++closes === 1) throw new Error("cleanup failed");
|
|
},
|
|
cancel() {},
|
|
}));
|
|
await session.open("target");
|
|
const read = session.call("strings", { query: "x" });
|
|
await new Promise((resolve) => setImmediate(resolve));
|
|
const close = session.close();
|
|
assert.equal(closes, 0);
|
|
assert.equal(reads, 1);
|
|
release({ items: [] });
|
|
assert.equal((await read).ok, true);
|
|
await assert.rejects(close, /cleanup failed/u);
|
|
assert.equal(session.context.lease, "owned");
|
|
await session.close();
|
|
assert.equal(session.context, null);
|
|
});
|
|
test("cancellation before admission makes no provider query; advertised reference schemas compile", async () => {
|
|
const provider = factory();
|
|
const session = new AtomicSession(provider.create);
|
|
await session.open("target");
|
|
const stop = new AbortController();
|
|
stop.abort();
|
|
const cancelled = await session.call("strings", { query: "x" }, stop.signal);
|
|
assert.equal(cancelled.error.code, "cancelled");
|
|
assert.deepEqual(provider.calls, []);
|
|
const ajv = new Ajv2020({ strict: false });
|
|
for (const method of reference().methods) ajv.compile(method.input_schema);
|
|
await session.close();
|
|
});
|