67 lines
2.4 KiB
TypeScript
67 lines
2.4 KiB
TypeScript
|
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);
|
||
|
|
});
|