From 1945bd502e7cade3d772c24d23161f39c1d84314 Mon Sep 17 00:00:00 2001
From: mkorwel <matt.korwel@gmail.com>
Date: Fri, 17 Apr 2026 20:42:17 +0000
Subject: [PATCH] test: fix SessionContext tests

---
 .../cli/src/ui/contexts/SessionContext.test.tsx   | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/packages/cli/src/ui/contexts/SessionContext.test.tsx b/packages/cli/src/ui/contexts/SessionContext.test.tsx
index 46874d0917..46cf8421be 100644
--- a/packages/cli/src/ui/contexts/SessionContext.test.tsx
+++ b/packages/cli/src/ui/contexts/SessionContext.test.tsx
@@ -63,6 +63,9 @@ describe('SessionStatsContext', () => {
       <SessionStatsProvider sessionId="test-session-id">
         <TestHarness contextRef={contextRef} />
       </SessionStatsProvider>,
+      undefined,
+      undefined,
+      true,
     );
 
     const stats = contextRef.current?.stats;
@@ -82,6 +85,9 @@ describe('SessionStatsContext', () => {
       <SessionStatsProvider sessionId="test-session-id">
         <TestHarness contextRef={contextRef} />
       </SessionStatsProvider>,
+      undefined,
+      undefined,
+      true,
     );
 
     const newMetrics: SessionMetrics = {
@@ -165,6 +171,9 @@ describe('SessionStatsContext', () => {
       <SessionStatsProvider sessionId="test-session-id">
         <CountingTestHarness />
       </SessionStatsProvider>,
+      undefined,
+      undefined,
+      true,
     );
 
     expect(renderCount).toBe(1);
@@ -248,6 +257,9 @@ describe('SessionStatsContext', () => {
       <SessionStatsProvider sessionId="test-session-id">
         <TestHarness contextRef={contextRef} />
       </SessionStatsProvider>,
+      undefined,
+      undefined,
+      true,
     );
 
     const initialStartTime = contextRef.current?.stats.sessionStartTime;
@@ -276,6 +288,9 @@ describe('SessionStatsContext', () => {
       <ErrorBoundary onError={onError}>
         <TestHarness contextRef={{ current: undefined }} />
       </ErrorBoundary>,
+      undefined,
+      undefined,
+      true,
     );
 
     expect(onError).toHaveBeenCalledWith(