From 31437b9e3bd1c219d2f84be2099ab1fd876858bb Mon Sep 17 00:00:00 2001 From: Ayaan Zaidi Date: Sat, 18 Apr 2026 14:20:36 +0530 Subject: [PATCH] test(cron): cover message tool with no delivery --- .../run.message-tool-policy.test.ts | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/src/cron/isolated-agent/run.message-tool-policy.test.ts b/src/cron/isolated-agent/run.message-tool-policy.test.ts index 3bff3561411..8b133fd6e5c 100644 --- a/src/cron/isolated-agent/run.message-tool-policy.test.ts +++ b/src/cron/isolated-agent/run.message-tool-policy.test.ts @@ -47,6 +47,19 @@ describe("runCronIsolatedAgentTurn message tool policy", () => { expect(runEmbeddedPiAgentMock.mock.calls[0]?.[0]?.disableMessageTool).toBe(true); } + async function expectMessageToolEnabledForPlan(plan: { + requested: boolean; + mode: "none" | "announce"; + channel?: string; + to?: string; + }) { + mockRunCronFallbackPassthrough(); + resolveCronDeliveryPlanMock.mockReturnValue(plan); + await runCronIsolatedAgentTurn(makeParams()); + expect(runEmbeddedPiAgentMock).toHaveBeenCalledTimes(1); + expect(runEmbeddedPiAgentMock.mock.calls[0]?.[0]?.disableMessageTool).toBe(false); + } + beforeEach(() => { previousFastTestEnv = clearFastTestEnv(); resetRunCronIsolatedAgentTurnHarness(); @@ -63,8 +76,8 @@ describe("runCronIsolatedAgentTurn message tool policy", () => { restoreFastTestEnv(previousFastTestEnv); }); - it('disables the message tool when delivery.mode is "none"', async () => { - await expectMessageToolDisabledForPlan({ + it('keeps the message tool enabled when delivery.mode is "none"', async () => { + await expectMessageToolEnabledForPlan({ requested: false, mode: "none", });