NetMesh/application/state/vaultManagedImportLock.test.ts

import assert from "node:assert/strict";
import test from "node:test";

import {
  isActiveVaultLockHandle,
  isVaultImportLockHeld,
  withVaultImportLock,
  withVaultImportLockIfNeeded,
} from "./vaultManagedImportLock.ts";

test("Vault imports are serialized without Web Locks", async () => {
  const events: string[] = [];
  let releaseFirst!: () => void;
  const firstGate = new Promise<void>((resolve) => {
    releaseFirst = resolve;
  });
  const first = withVaultImportLock("shared", async () => {
    events.push("first:start");
    await firstGate;
    events.push("first:end");
  }, null);
  const second = withVaultImportLock("shared", async () => {
    events.push("second:start");
    events.push("second:end");
  }, null);

  await Promise.resolve();
  assert.deepEqual(events, ["first:start"]);
  releaseFirst();
  await Promise.all([first, second]);
  assert.deepEqual(events, [
    "first:start",
    "first:end",
    "second:start",
    "second:end",
  ]);
});

test("Vault import lock serializes concurrent work while nested helpers use the handle", async () => {
  const events: string[] = [];
  let releaseOuter!: () => void;
  const outerGate = new Promise<void>((resolve) => {
    releaseOuter = resolve;
  });

  const outer = withVaultImportLock("owned", async (lock) => {
    events.push("outer:start");
    assert.equal(isVaultImportLockHeld("owned"), true);
    assert.equal(isActiveVaultLockHandle("owned", lock), true);

    await withVaultImportLockIfNeeded("owned", async () => {
      events.push("nested");
    }, lock, null);

    await outerGate;
    events.push("outer:end");
  }, null);

  await Promise.resolve();
  assert.deepEqual(events, ["outer:start", "nested"]);

  // Started without the active handle: must queue behind the outer owner.
  const concurrent = withVaultImportLockIfNeeded("owned", async () => {
    events.push("concurrent-if-needed");
  }, undefined, null);

  await Promise.resolve();
  assert.deepEqual(events, ["outer:start", "nested"]);
  releaseOuter();
  await Promise.all([outer, concurrent]);
  assert.equal(isVaultImportLockHeld("owned"), false);
  assert.deepEqual(events, [
    "outer:start",
    "nested",
    "outer:end",
    "concurrent-if-needed",
  ]);
});

test("Vault imports fail safely in a window without shared locking", async () => {
  const previousWindow = Object.getOwnPropertyDescriptor(globalThis, "window");
  Object.defineProperty(globalThis, "window", {
    configurable: true,
    value: {},
  });
  try {
    await assert.rejects(
      () => withVaultImportLock("shared", async () => undefined, null),
      /Cross-window Vault import locking is unavailable/,
    );
  } finally {
    if (previousWindow) Object.defineProperty(globalThis, "window", previousWindow);
    else delete (globalThis as { window?: unknown }).window;
  }
});