arisa 5.2.7 → 5.2.17

package/test/task-store.test.js

@@ -121,6 +121,38 @@ test("recovers interrupted running tasks for retry after restart", async () => {
   );
 });
 
+test("persists auth blocks, claims only due probes, and clears the block after success", async () => {
+  await resetHome();
+  const store = new TaskStore();
+  await store.add({
+    id: "auth-poll",
+    kind: "poll_tool",
+    runAt: new Date(Date.now() - 1000).toISOString(),
+    payload: { toolName: "checker", args: { action: "poll" } },
+    recurrence: { type: "interval", everySeconds: 60 }
+  });
+
+  await store.claimDue();
+  const blocked = await store.blockAuth("auth-poll", "authentication expired", {
+    retryAfterSeconds: 3600,
+    probeArgs: { action: "auth-status" }
+  });
+  assert.equal(blocked.status, "blocked_auth");
+  assert.equal(blocked.authBlockedNew, true);
+  assert.deepEqual(blocked.authBlock.probeArgs, { action: "auth-status" });
+  assert.deepEqual(await store.claimDue(), []);
+
+  store.tasks.find((task) => task.id === "auth-poll").runAt = new Date(Date.now() - 1000).toISOString();
+  await store.save();
+  const [probe] = await store.claimDue();
+  assert.equal(probe.status, "running");
+  assert.ok(probe.authBlock);
+
+  const completed = await store.complete("auth-poll");
+  assert.equal(completed.status, "pending");
+  assert.equal(completed.authBlock, undefined);
+});
+
 test("completes one-off tasks and re-schedules recurring interval tasks", async () => {
   await resetHome();
   const store = new TaskStore();

package/test/telegram-prompt-controller.test.js

@@ -12,7 +12,8 @@ function createController(overrides = {}) {
     artifactStore: {},
     toolRegistry: {},
     agentManager: {
-      resetSession: (...args) => calls.reset.push(args)
+      resetSession: (...args) => calls.reset.push(args),
+      runTurn: async (_options, work) => work()
     },
     sessionSeeds: {
       clear: async (chatId) => calls.cleared.push(chatId)

package/test/telegram-task-dispatcher.test.js

@@ -7,6 +7,10 @@ function createHarness(overrides = {}) {
   const taskStore = {
     async fail(...args) { calls.push(["fail", ...args]); return { status: "failed" }; },
     async complete(...args) { calls.push(["complete", ...args]); return { status: "done" }; },
+    async blockAuth(taskId, error, resolution) {
+      calls.push(["blockAuth", taskId, error.message, resolution]);
+      return { status: "blocked_auth", authBlockedNew: true, runAt: "2026-09-02T05:00:00.000Z" };
+    },
     async retryOrFail(taskId, error, options) {
       calls.push(["retryOrFail", taskId, error.message, options]);
       return { status: options.retryable ? "pending" : "failed" };
@@ -21,7 +25,10 @@ function createHarness(overrides = {}) {
     artifactStore: { forChat() { throw new Error("unexpected artifact access"); } },
     toolRegistry: {},
     resourceNotes: { async get() { return ""; } },
-    agentManager: { async runTool(input) { calls.push(["runTool", input]); return { ok: true }; } },
+    agentManager: {
+      async runTurn(options, work) { calls.push(["runTurn", options]); return work(); },
+      async runTool(input) { calls.push(["runTool", input]); return { ok: true }; }
+    },
     logger: null,
     ...overrides.dependencies
   });
@@ -74,7 +81,9 @@ test("passes bounded execution deadlines to scheduled prompts", async () => {
 
   const enqueues = calls.filter(([name]) => name === "enqueue");
   assert.equal(enqueues[0][1].timeoutMs, 900);
+  assert.equal(enqueues[0][1].priority, "background");
   assert.equal(enqueues[1][1].timeoutMs, 300);
+  assert.equal(enqueues[1][1].priority, "interactive");
 });
 
 test("acknowledges an agent event before executing it", async () => {
@@ -101,15 +110,66 @@ test("runs poll tools headlessly and confirms their result", async () => {
   });
 
   assert.deepEqual(calls, [
+    ["runTurn", { priority: "background", label: "poll tool checker" }],
     ["runTool", { name: "checker", request: { args: { cursor: "4" } }, chatId: 123 }],
     ["complete", "poll-1"]
   ]);
 });
 
+test("pauses a poll once when its tool reports terminal authentication failure", async () => {
+  const resolution = {
+    type: "reauthentication_required",
+    retryAfterSeconds: 3600,
+    probeArgs: { action: "auth-status" }
+  };
+  const { calls, dispatcher } = createHarness({
+    dependencies: {
+      agentManager: {
+        async runTurn(options, work) { calls.push(["runTurn", options]); return work(); },
+        async runTool(input) {
+          calls.push(["runTool", input]);
+          return { ok: false, status: "blocked_auth", error: "authentication expired", resolution };
+        }
+      }
+    }
+  });
+
+  await dispatcher.runClaimedTask({
+    id: "poll-auth",
+    kind: "poll_tool",
+    payload: { chatId: 123, toolName: "checker", args: { action: "poll" } }
+  });
+
+  assert.deepEqual(calls.at(-2), ["blockAuth", "poll-auth", "authentication expired", resolution]);
+  assert.deepEqual(calls.at(-1), [
+    "send",
+    123,
+    "⚠️ Arisa automation paused for authentication\nTool: checker\nReason: authentication expired\nNext authentication check: 2026-09-02T05:00:00.000Z",
+    undefined
+  ]);
+});
+
+test("uses a lightweight auth probe before resuming blocked poll work", async () => {
+  const { calls, dispatcher } = createHarness();
+  await dispatcher.runClaimedTask({
+    id: "poll-resume",
+    kind: "poll_tool",
+    authBlock: { probeArgs: { action: "auth-status" }, retryAfterSeconds: 3600 },
+    payload: { chatId: 123, toolName: "checker", args: { action: "poll" } }
+  });
+
+  assert.deepEqual(calls.filter(([name]) => name === "runTool").map((call) => call[1].request.args), [
+    { action: "auth-status" },
+    { action: "poll" }
+  ]);
+  assert.deepEqual(calls.at(-1), ["complete", "poll-resume"]);
+});
+
 test("retries a known poll failure with backoff", async () => {
   const { calls, dispatcher } = createHarness({
     dependencies: {
       agentManager: {
+        async runTurn(options, work) { calls.push(["runTurn", options]); return work(); },
         async runTool(input) {
           calls.push(["runTool", input]);
           return { ok: false, status: "failed", error: "temporary checker failure" };
@@ -220,10 +280,11 @@ test("due-task dispatch retries one failure without blocking another task", asyn
 
   await dispatcher.dispatchDueTasks();
 
-  assert.deepEqual(calls, [
+  assert.deepEqual(calls.slice(0, 3), [
     ["claimDue", 10],
-    ["runTool", { name: "checker", request: { args: {} }, chatId: 123 }],
-    ["complete", "good"],
-    ["retryOrFail", "bad", "queue unavailable", { retryable: true }]
+    ["runTurn", { priority: "background", label: "poll tool checker" }],
+    ["runTool", { name: "checker", request: { args: {} }, chatId: 123 }]
   ]);
+  assert.ok(calls.some((call) => JSON.stringify(call) === JSON.stringify(["complete", "good"])));
+  assert.ok(calls.some((call) => JSON.stringify(call) === JSON.stringify(["retryOrFail", "bad", "queue unavailable", { retryable: true }])));
 });

package/test/tool-registry-run.test.js

@@ -8,7 +8,7 @@ const homeDir = await mkdtemp(path.join(os.tmpdir(), "arisa-tool-registry-home-"
 process.env.HOME = homeDir;
 process.env.USERPROFILE = homeDir;
 
-const { ToolRegistry, createToolOutputParser, isolatedToolProcessInvocation } = await import("../src/core/tools/tool-registry.js");
+const { ToolRegistry, createToolOutputParser, isolatedToolProcessInvocation, readyDaemonAdmission } = await import("../src/core/tools/tool-registry.js");
 const { createToolOutputParser: directToolOutputParser } = await import("../src/core/tools/tool-process-output.js");
 const { isolatedToolProcessInvocation: directToolProcessInvocation } = await import("../src/core/tools/tool-process-runner.js");
 const {
@@ -242,6 +242,15 @@ test("reports dependency status in help and blocks a tool with a missing depende
   );
 });
 
+test("only a live ready declared daemon bypasses worker RSS spawn admission", () => {
+  const tool = { daemon: { scope: "chat", health: "internal" } };
+  assert.deepEqual(readyDaemonAdmission(tool, { alive: true, state: "ready", restart: { requested: false } }), { ignoreWorkerRss: true });
+  assert.deepEqual(readyDaemonAdmission(tool, { alive: false, state: "ready", restart: { requested: false } }), {});
+  assert.deepEqual(readyDaemonAdmission(tool, { alive: true, state: "degraded", restart: { requested: false } }), {});
+  assert.deepEqual(readyDaemonAdmission(tool, { alive: true, state: "ready", restart: { requested: true } }), {});
+  assert.deepEqual(readyDaemonAdmission({}, { alive: true, state: "ready" }), {});
+});
+
 test("wraps declared tool runs in the shared execution governor", async () => {
   await resetHome();
   await createFakeTool("heavy-tool", {

package/test/weighted-resource-governor.test.js

@@ -133,6 +133,34 @@ test("rejects declared heavy tools before spawn when memory pressure is unsafe",
   assert.equal(governor.snapshot().resources.browser.activeWeight, 0);
 });
 
+test("ready daemon jobs bypass only worker RSS spawn admission", async () => {
+  const governor = new WeightedResourceGovernor({
+    policy: { maxWorkerRssMb: 384, maxSwapUsedPercent: 95 },
+    memoryPressure: async () => ({
+      availableBytes: 512 * 1024 * 1024,
+      totalBytes: 4 * 1024 * 1024 * 1024,
+      workerRssBytes: 450 * 1024 * 1024,
+      swapTotalBytes: 100,
+      swapUsedPercent: 50
+    })
+  });
+  const lease = await governor.acquire({ resourceClass: "browser", weight: 1 }, "ready-daemon", { ignoreWorkerRss: true });
+  assert.equal(lease.memoryLimitMb, 384);
+  lease.release({ success: true });
+
+  governor.memoryPressure = async () => ({
+    availableBytes: 512 * 1024 * 1024,
+    totalBytes: 4 * 1024 * 1024 * 1024,
+    workerRssBytes: 450 * 1024 * 1024,
+    swapTotalBytes: 100,
+    swapUsedPercent: 96
+  });
+  await assert.rejects(
+    () => governor.acquire({ resourceClass: "browser", weight: 1 }, "ready-daemon", { ignoreWorkerRss: true }),
+    /swap use/
+  );
+});
+
 test("larger weights consume shared capacity and worker RSS peaks are retained", async () => {
   let rss = 120 * 1024 * 1024;
   const governor = new WeightedResourceGovernor({

package/test/worker-tool-fanout.test.js

@@ -0,0 +1,79 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+import { WorkerToolFanoutController } from "../src/core/agent/worker-tool-fanout.js";
+
+function deferred() {
+  let resolve;
+  const promise = new Promise((done) => { resolve = done; });
+  return { promise, resolve };
+}
+
+function breaker(percent = 20) {
+  return {
+    sample: () => ({ heapUsed: percent, heapLimit: 100, percent }),
+    admit: async () => ({ heapUsed: percent, heapLimit: 100, percent })
+  };
+}
+
+test("admits low-pressure tool calls in pairs", async () => {
+  const controller = new WorkerToolFanoutController({ heapCircuitBreaker: breaker() });
+  const gates = Array.from({ length: 4 }, deferred);
+  let active = 0;
+  let peak = 0;
+  const runs = gates.map((gate) => controller.run(async () => {
+    active += 1;
+    peak = Math.max(peak, active);
+    await gate.promise;
+    active -= 1;
+  }));
+
+  await new Promise((resolve) => setImmediate(resolve));
+  assert.equal(active, 2);
+  gates[0].resolve();
+  gates[1].resolve();
+  await new Promise((resolve) => setImmediate(resolve));
+  assert.equal(active, 2);
+  gates[2].resolve();
+  gates[3].resolve();
+  await Promise.all(runs);
+
+  assert.equal(peak, 2);
+  assert.equal(controller.getDiagnostic().peakQueued, 3);
+});
+
+test("serializes tool calls when the worker heap is elevated", async () => {
+  const controller = new WorkerToolFanoutController({ heapCircuitBreaker: breaker(70) });
+  const gates = Array.from({ length: 3 }, deferred);
+  let active = 0;
+  let peak = 0;
+  const runs = gates.map((gate) => controller.run(async () => {
+    active += 1;
+    peak = Math.max(peak, active);
+    await gate.promise;
+    active -= 1;
+  }));
+
+  await new Promise((resolve) => setImmediate(resolve));
+  assert.equal(active, 1);
+  for (const gate of gates) {
+    gate.resolve();
+    await new Promise((resolve) => setImmediate(resolve));
+  }
+  await Promise.all(runs);
+
+  assert.equal(peak, 1);
+  assert.ok(controller.getDiagnostic().pressureSerializations > 0);
+});
+
+test("rejects a queued tool call when heap admission remains critical", async () => {
+  const expected = Object.assign(new Error("critical"), { code: "WORKER_HEAP_PRESSURE" });
+  const controller = new WorkerToolFanoutController({
+    heapCircuitBreaker: {
+      sample: () => ({ heapUsed: 90, heapLimit: 100, percent: 90 }),
+      admit: async () => { throw expected; }
+    }
+  });
+
+  await assert.rejects(controller.run(async () => "unreachable"), expected);
+  assert.equal(controller.getDiagnostic().rejectedAdmissions, 1);
+});