[Init] Initial commit - NetMesh terminal manager

electron/bridges/mcpServerBridge/sessionOwnership.test.cjs

@@ -0,0 +1,90 @@
+"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);
+  }
+});