From 73b0a7ecf160f7f9091b47c9e80d64a21f25cc3d Mon Sep 17 00:00:00 2001
From: Craig Feuerherdt <34226296+craigfeuerherdt@users.noreply.github.com>
Date: Wed, 2 Sep 2026 21:31:04 +1000
Subject: [PATCH] Enhance tests for Antigravity functionality

Added tests for Antigravity Google login and API key.
---
 request-logger/agents.test.ts | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/request-logger/agents.test.ts b/request-logger/agents.test.ts
index d121e60..8db7a15 100644
--- a/request-logger/agents.test.ts
+++ b/request-logger/agents.test.ts
@@ -83,6 +83,7 @@ describe("listAgents", () => {
       "pi",
       "omp",
       "gemini",
+      "antigravity",
       "junie",
       "cursor",
       "amp",
@@ -490,6 +491,18 @@ describe("resolveChoice — commands", () => {
     );
   });
 
+  it("uses the CLOUD_CODE_URL variable for an Antigravity Google login", () => {
+    expect(target("antigravity", "google-login").command).toBe(
+      "CLOUD_CODE_URL=http://localhost:8787 agy"
+    );
+  });
+
+  it("uses the GOOGLE_GEMINI_BASE_URL variable for an Antigravity API key", () => {
+    expect(target("antigravity", "api-key").command).toBe(
+      "GOOGLE_GEMINI_BASE_URL=http://localhost:8787 agy"
+    );
+  });
+
   it("gives Pi a bare command, because Pi has no base URL variable", () => {
     expect(target("pi", "anthropic").command).toBe("pi");
   });