From 0ac14b0d37d87659f3c0b129708706efab432ca4 Mon Sep 17 00:00:00 2001
From: km-anthropic <km-anthropic@users.noreply.github.com>
Date: Wed, 13 Aug 2025 10:17:25 -0700
Subject: [PATCH] Fix agent test to handle dynamic branch names from
 environment

---
 test/modes/agent.test.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/test/modes/agent.test.ts b/test/modes/agent.test.ts
index 679ed4a..4dcf2b0 100644
--- a/test/modes/agent.test.ts
+++ b/test/modes/agent.test.ts
@@ -120,12 +120,13 @@ describe("Agent Mode", () => {
     expect(callArgs[1]).toContain("--mcp-config");
     expect(callArgs[1]).toContain("--model claude-sonnet-4 --max-turns 10");
 
-    // Verify return structure
+    // Verify return structure - branch will be whatever is in env or "main" as fallback
+    const expectedBranch = process.env.GITHUB_HEAD_REF || process.env.GITHUB_REF_NAME || "main";
     expect(result).toEqual({
       commentId: undefined,
       branchInfo: {
         baseBranch: "main",
-        currentBranch: "main",
+        currentBranch: expectedBranch,
         claudeBranch: undefined,
       },
       mcpConfig: expect.any(String),