From fff88e794db7ab1fe6e6edc2bcf423e159c8a497 Mon Sep 17 00:00:00 2001
From: Jack Wotherspoon <jackwoth@google.com>
Date: Mon, 9 Mar 2026 14:49:21 +0100
Subject: [PATCH] chore: fix test

---
 .../LoginWithGoogleRestartDialog.test.tsx     | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx b/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx
index 77310e3069..d5cd9fd261 100644
--- a/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx
+++ b/packages/cli/src/ui/auth/LoginWithGoogleRestartDialog.test.tsx
@@ -5,7 +5,15 @@
  */
 
 import { render } from '../../test-utils/render.js';
-import { describe, it, expect, vi, beforeEach, type Mock } from 'vitest';
+import {
+  describe,
+  it,
+  expect,
+  vi,
+  beforeEach,
+  afterEach,
+  type Mock,
+} from 'vitest';
 import { LoginWithGoogleRestartDialog } from './LoginWithGoogleRestartDialog.js';
 import { useKeypress } from '../hooks/useKeypress.js';
 import { runExitCleanup } from '../../utils/cleanup.js';
@@ -35,13 +43,22 @@ describe('LoginWithGoogleRestartDialog', () => {
 
   const mockConfig = {
     getRemoteAdminSettings: vi.fn(),
+    getSessionId: vi.fn().mockReturnValue('mock-session-id'),
   } as unknown as Config;
 
+  let originalSend: typeof process.send;
+
   beforeEach(() => {
     vi.clearAllMocks();
     exitSpy.mockClear();
     vi.useRealTimers();
     _resetRelaunchStateForTesting();
+    originalSend = process.send;
+    process.send = vi.fn();
+  });
+
+  afterEach(() => {
+    process.send = originalSend;
   });
 
   it('renders correctly', async () => {