[Init] Initial commit - NetMesh terminal manager

application/state/sftp/transferCancelLatch.test.ts

@@ -0,0 +1,66 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+
+import {
+  clearTransferCancelledTree,
+  isTransferCancelledFlag,
+  isTransferOrRootCancelled,
+  markTransferCancelled,
+  markTransferCancelledTree,
+  resetTransferCancelLatchesForTests,
+  settleTransferCancelTree,
+} from "./transferCancelLatch";
+
+test("cancel flags are process-global for root and children", () => {
+  resetTransferCancelLatchesForTests();
+  markTransferCancelledTree("parent", ["c1", "c2"]);
+  assert.equal(isTransferCancelledFlag("parent"), true);
+  assert.equal(isTransferOrRootCancelled("parent", "c1"), true);
+  assert.equal(isTransferCancelledFlag("other"), false);
+  clearTransferCancelledTree("parent", ["c1", "c2"]);
+  assert.equal(isTransferCancelledFlag("parent"), false);
+  assert.equal(isTransferCancelledFlag("c1"), false);
+});
+
+test("settled single-file cancellations do not accumulate across a large batch", () => {
+  resetTransferCancelLatchesForTests();
+  const taskIds = Array.from({ length: 4_000 }, (_, index) => `single-${index}`);
+
+  for (const taskId of taskIds) {
+    markTransferCancelled(taskId);
+    assert.equal(isTransferCancelledFlag(taskId), true);
+    settleTransferCancelTree(taskId);
+  }
+
+  for (const taskId of taskIds) {
+    assert.equal(isTransferCancelledFlag(taskId), false);
+  }
+});
+
+test("settling a directory root releases every child recorded by tree cancellation", () => {
+  resetTransferCancelLatchesForTests();
+  const childIds = Array.from({ length: 4_000 }, (_, index) => `directory-child-${index}`);
+  markTransferCancelledTree("directory-root", childIds);
+
+  settleTransferCancelTree("directory-root");
+
+  assert.equal(isTransferCancelledFlag("directory-root"), false);
+  for (const childId of childIds) {
+    assert.equal(isTransferCancelledFlag(childId), false);
+  }
+});
+
+test("same-id resume clears the old tree without masking a later cancellation", () => {
+  resetTransferCancelLatchesForTests();
+  markTransferCancelledTree("same-root", ["old-child"]);
+  clearTransferCancelledTree("same-root");
+
+  markTransferCancelledTree("same-root", ["new-child"]);
+
+  assert.equal(isTransferCancelledFlag("old-child"), false);
+  assert.equal(isTransferCancelledFlag("same-root"), true);
+  assert.equal(isTransferCancelledFlag("new-child"), true);
+  settleTransferCancelTree("same-root");
+  assert.equal(isTransferCancelledFlag("same-root"), false);
+  assert.equal(isTransferCancelledFlag("new-child"), false);
+});