feat: ship enterprise control and desktop governance

local-agent/master-task-timeout.test.mjs

@@ -0,0 +1,77 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+import { resolveMasterAgentTaskTimeoutMs, runWithTaskTimeout } from "./master-task-timeout.mjs";
+
+test("resolveMasterAgentTaskTimeoutMs prefers short timeout for project understanding sync", () => {
+  const timeoutMs = resolveMasterAgentTaskTimeoutMs(
+    {},
+    {
+      taskType: "conversation_reply",
+      projectId: "master-agent",
+      projectUnderstandingTargetProjectId: "project-1",
+    },
+  );
+
+  assert.equal(timeoutMs, 8 * 60 * 1000);
+});
+
+test("resolveMasterAgentTaskTimeoutMs uses thread execution timeout for child thread work", () => {
+  const timeoutMs = resolveMasterAgentTaskTimeoutMs(
+    {},
+    {
+      taskType: "conversation_reply",
+      projectId: "project-1",
+    },
+  );
+
+  assert.equal(timeoutMs, 30 * 60 * 1000);
+});
+
+test("resolveMasterAgentTaskTimeoutMs honors config overrides", () => {
+  const timeoutMs = resolveMasterAgentTaskTimeoutMs(
+    {
+      projectUnderstandingTaskTimeoutMs: 1234,
+      masterAgentReplyTimeoutMs: 2345,
+      threadExecutionTaskTimeoutMs: 3456,
+    },
+    {
+      taskType: "conversation_reply",
+      projectId: "master-agent",
+      projectUnderstandingTargetProjectId: "project-1",
+    },
+  );
+
+  assert.equal(timeoutMs, 1234);
+});
+
+test("runWithTaskTimeout resolves before timeout when operation finishes", async () => {
+  const result = await runWithTaskTimeout(
+    {
+      timeoutMs: 100,
+      label: "quick-task",
+    },
+    async () => "ok",
+  );
+
+  assert.equal(result, "ok");
+});
+
+test("runWithTaskTimeout rejects and invokes timeout cleanup when operation hangs", async () => {
+  let timeoutCleanupCalled = 0;
+
+  await assert.rejects(
+    runWithTaskTimeout(
+      {
+        timeoutMs: 20,
+        label: "hung-task",
+        onTimeout: async () => {
+          timeoutCleanupCalled += 1;
+        },
+      },
+      async () => await new Promise(() => {}),
+    ),
+    /hung-task exceeded timeout after 20ms/,
+  );
+
+  assert.equal(timeoutCleanupCalled, 1);
+});