From e735ff45ed6ecd4e3abe62bd757c261304fc8025 Mon Sep 17 00:00:00 2001
From: mkorwel <matt.korwel@gmail.com>
Date: Sat, 18 Apr 2026 22:27:45 +0000
Subject: [PATCH] test: fix useMemoryMonitor tests

---
 packages/cli/src/ui/hooks/useMemoryMonitor.test.tsx | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/packages/cli/src/ui/hooks/useMemoryMonitor.test.tsx b/packages/cli/src/ui/hooks/useMemoryMonitor.test.tsx
index cfaf2fb470..c2ac79f6a9 100644
--- a/packages/cli/src/ui/hooks/useMemoryMonitor.test.tsx
+++ b/packages/cli/src/ui/hooks/useMemoryMonitor.test.tsx
@@ -36,7 +36,7 @@ describe('useMemoryMonitor', () => {
     memoryUsageSpy.mockReturnValue({
       rss: MEMORY_WARNING_THRESHOLD / 2,
     } as NodeJS.MemoryUsage);
-    await render(<TestComponent />);
+    await render(<TestComponent />, undefined, undefined, true);
     vi.advanceTimersByTime(10000);
     expect(addItem).not.toHaveBeenCalled();
   });
@@ -45,7 +45,7 @@ describe('useMemoryMonitor', () => {
     memoryUsageSpy.mockReturnValue({
       rss: MEMORY_WARNING_THRESHOLD * 1.5,
     } as NodeJS.MemoryUsage);
-    await render(<TestComponent />);
+    await render(<TestComponent />, undefined, undefined, true);
     vi.advanceTimersByTime(MEMORY_CHECK_INTERVAL);
     expect(addItem).toHaveBeenCalledTimes(1);
     expect(addItem).toHaveBeenCalledWith(
@@ -61,7 +61,12 @@ describe('useMemoryMonitor', () => {
     memoryUsageSpy.mockReturnValue({
       rss: MEMORY_WARNING_THRESHOLD * 1.5,
     } as NodeJS.MemoryUsage);
-    const { rerender } = await render(<TestComponent />);
+    const { rerender } = await render(
+      <TestComponent />,
+      undefined,
+      undefined,
+      true,
+    );
     vi.advanceTimersByTime(MEMORY_CHECK_INTERVAL);
     expect(addItem).toHaveBeenCalledTimes(1);