91 lines
3.7 KiB
JavaScript
91 lines
3.7 KiB
JavaScript
|
"use strict";
|
||
|
|
|
||
|
|
const test = require("node:test");
|
||
|
|
const assert = require("node:assert/strict");
|
||
|
|
const { createSessionOwnershipRegistry } = require("./sessionOwnership.cjs");
|
||
|
|
|
||
|
|
test("session ownership is isolated by AI scope", () => {
|
||
|
|
const ownership = createSessionOwnershipRegistry();
|
||
|
|
ownership.register("chat-a", "session-1");
|
||
|
|
|
||
|
|
assert.equal(ownership.validate("chat-a", "session-1").ok, true);
|
||
|
|
assert.equal(ownership.validate("chat-b", "session-1").ok, false);
|
||
|
|
assert.match(ownership.validate("chat-b", "session-1").error, /not opened/i);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("listOwned returns sessions registered for a chat scope", () => {
|
||
|
|
const ownership = createSessionOwnershipRegistry();
|
||
|
|
ownership.register("chat-a", "session-1");
|
||
|
|
ownership.register("chat-a", "session-2");
|
||
|
|
ownership.register("chat-b", "session-3");
|
||
|
|
|
||
|
|
assert.deepEqual(ownership.listOwned("chat-a").sort(), ["session-1", "session-2"]);
|
||
|
|
assert.deepEqual(ownership.listOwned("chat-b"), ["session-3"]);
|
||
|
|
assert.deepEqual(ownership.listOwned("missing"), []);
|
||
|
|
assert.deepEqual(ownership.listOwned(""), []);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("forgetSession revokes ownership from every scope", () => {
|
||
|
|
const ownership = createSessionOwnershipRegistry();
|
||
|
|
ownership.register("chat-a", "session-1");
|
||
|
|
ownership.register("chat-b", "session-1");
|
||
|
|
ownership.forgetSession("session-1");
|
||
|
|
|
||
|
|
assert.equal(ownership.validate("chat-a", "session-1").ok, false);
|
||
|
|
assert.equal(ownership.validate("chat-b", "session-1").ok, false);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("releaseScopeOwnership clears retained ids without revoking generations", () => {
|
||
|
|
const ownership = createSessionOwnershipRegistry();
|
||
|
|
const generation = ownership.captureGeneration("chat-a");
|
||
|
|
ownership.register("chat-a", "session-1", generation);
|
||
|
|
ownership.register("chat-b", "session-2");
|
||
|
|
|
||
|
|
ownership.releaseScopeOwnership("chat-a");
|
||
|
|
|
||
|
|
assert.deepEqual(ownership.listOwned("chat-a"), []);
|
||
|
|
assert.equal(ownership.validate("chat-a", "session-1").ok, false);
|
||
|
|
assert.equal(ownership.validate("chat-b", "session-2").ok, true);
|
||
|
|
// Late host_open for the same generation must still be allowed.
|
||
|
|
assert.equal(ownership.register("chat-a", "session-3", generation), true);
|
||
|
|
assert.deepEqual(ownership.listOwned("chat-a"), ["session-3"]);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("clearScope only revokes the deleted chat scope", () => {
|
||
|
|
const ownership = createSessionOwnershipRegistry();
|
||
|
|
ownership.register("chat-a", "session-1");
|
||
|
|
ownership.register("chat-b", "session-2");
|
||
|
|
ownership.clearScope("chat-a");
|
||
|
|
|
||
|
|
assert.equal(ownership.validate("chat-a", "session-1").ok, false);
|
||
|
|
assert.equal(ownership.validate("chat-b", "session-2").ok, true);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("a host open that finishes after scope cleanup cannot restore ownership", () => {
|
||
|
|
const ownership = createSessionOwnershipRegistry();
|
||
|
|
const generation = ownership.captureGeneration("chat-a");
|
||
|
|
ownership.clearScope("chat-a");
|
||
|
|
|
||
|
|
assert.equal(ownership.register("chat-a", "session-1", generation), false);
|
||
|
|
assert.equal(ownership.validate("chat-a", "session-1").ok, false);
|
||
|
|
|
||
|
|
const nextGeneration = ownership.captureGeneration("chat-a");
|
||
|
|
assert.equal(ownership.register("chat-a", "session-2", nextGeneration), true);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("cleared scope generations are released without allowing a late host open", () => {
|
||
|
|
const ownership = createSessionOwnershipRegistry();
|
||
|
|
const staleGenerations = [];
|
||
|
|
|
||
|
|
for (let index = 0; index < 100; index += 1) {
|
||
|
|
const scopeId = `chat-${index}`;
|
||
|
|
staleGenerations.push([scopeId, ownership.captureGeneration(scopeId)]);
|
||
|
|
ownership.clearScope(scopeId);
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.equal(ownership.getTrackedGenerationCountForTests(), 0);
|
||
|
|
for (const [scopeId, generation] of staleGenerations) {
|
||
|
|
assert.equal(ownership.register(scopeId, `late-${scopeId}`, generation), false);
|
||
|
|
}
|
||
|
|
});
|