From 84c3ee1f053a6e4ca1cf9c7d22e6a70cf10c04d6 Mon Sep 17 00:00:00 2001
From: Jack Wotherspoon <jackwoth@google.com>
Date: Mon, 9 Mar 2026 15:43:22 +0100
Subject: [PATCH] chore: fix test

---
 .../cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx  | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx b/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx
index d5cd9fd261..9397af0116 100644
--- a/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx
+++ b/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx
@@ -54,7 +54,12 @@ describe('LoginWithGoogleRestartDialog', () => {
     vi.useRealTimers();
     _resetRelaunchStateForTesting();
     originalSend = process.send;
-    process.send = vi.fn();
+    process.send = vi.fn((_message, callback) => {
+      if (typeof callback === 'function') {
+        callback(null);
+      }
+      return true;
+    });
   });
 
   afterEach(() => {