From f1dd59bf82da7e76f6269791061657ba8e1f5065 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Mon, 12 Jan 2026 17:13:49 +0000
Subject: [PATCH] test: update heartbeat and agent list thresholds

---
 src/agents/pi-embedded-runner.test.ts | 1 +
 src/auto-reply/heartbeat.test.ts      | 2 +-
 src/cron/isolated-agent.test.ts       | 2 +-
 src/gateway/server.agents.test.ts     | 1 +
 4 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/agents/pi-embedded-runner.test.ts b/src/agents/pi-embedded-runner.test.ts
index a0e675f36..741859b3e 100644
--- a/src/agents/pi-embedded-runner.test.ts
+++ b/src/agents/pi-embedded-runner.test.ts
@@ -22,6 +22,7 @@ vi.mock("./model-auth.js", () => ({
   getApiKeyForModel: vi.fn(),
   ensureAuthProfileStore: vi.fn(() => ({ profiles: {} })),
   resolveAuthProfileOrder: vi.fn(() => []),
+  resolveEnvApiKey: vi.fn(() => null),
 }));
 
 vi.mock("../providers/github-copilot-token.js", async () => {
diff --git a/src/auto-reply/heartbeat.test.ts b/src/auto-reply/heartbeat.test.ts
index 60fa83521..0b0455019 100644
--- a/src/auto-reply/heartbeat.test.ts
+++ b/src/auto-reply/heartbeat.test.ts
@@ -52,7 +52,7 @@ describe("stripHeartbeatToken", () => {
   });
 
   it("keeps heartbeat replies when remaining content exceeds threshold", () => {
-    const long = "A".repeat(31);
+    const long = "A".repeat(350);
     expect(
       stripHeartbeatToken(`${long} ${HEARTBEAT_TOKEN}`, { mode: "heartbeat" }),
     ).toEqual({
diff --git a/src/cron/isolated-agent.test.ts b/src/cron/isolated-agent.test.ts
index a93633539..4fad07f9a 100644
--- a/src/cron/isolated-agent.test.ts
+++ b/src/cron/isolated-agent.test.ts
@@ -803,7 +803,7 @@ describe("runCronIsolatedAgentTurn", () => {
         sendMessageIMessage: vi.fn(),
       };
       // Long content after HEARTBEAT_OK should still be delivered.
-      const longContent = `Important alert: ${"a".repeat(50)}`;
+      const longContent = `Important alert: ${"a".repeat(350)}`;
       vi.mocked(runEmbeddedPiAgent).mockResolvedValue({
         payloads: [{ text: `HEARTBEAT_OK ${longContent}` }],
         meta: {
diff --git a/src/gateway/server.agents.test.ts b/src/gateway/server.agents.test.ts
index 60a8543a8..240058bdb 100644
--- a/src/gateway/server.agents.test.ts
+++ b/src/gateway/server.agents.test.ts
@@ -39,6 +39,7 @@ describe("gateway server agents", () => {
     expect(res.payload?.agents.map((agent) => agent.id)).toEqual([
       "work",
       "home",
+      "main",
     ]);
     const work = res.payload?.agents.find((agent) => agent.id === "work");
     const home = res.payload?.agents.find((agent) => agent.id === "home");