@google/gemini-cli 0.1.9-nightly.250708.137ffec3 → 0.1.9-nightly.250708.a4097ae6

package/dist/src/ui/contexts/SessionContext.test.js

@@ -0,0 +1,96 @@
+import { jsx as _jsx } from "react/jsx-runtime";
+import { render } from 'ink-testing-library';
+import { renderHook } from '@testing-library/react';
+import { act } from 'react-dom/test-utils';
+import { SessionStatsProvider, useSessionStats, } from './SessionContext.js';
+import { describe, it, expect, vi } from 'vitest';
+import { uiTelemetryService } from '@google/gemini-cli-core';
+/**
+ * A test harness component that uses the hook and exposes the context value
+ * via a mutable ref. This allows us to interact with the context's functions
+ * and assert against its state directly in our tests.
+ */
+const TestHarness = ({ contextRef, }) => {
+    contextRef.current = useSessionStats();
+    return null;
+};
+describe('SessionStatsContext', () => {
+    it('should provide the correct initial state', () => {
+        const contextRef = { current: undefined };
+        render(_jsx(SessionStatsProvider, { children: _jsx(TestHarness, { contextRef: contextRef }) }));
+        const stats = contextRef.current?.stats;
+        expect(stats?.sessionStartTime).toBeInstanceOf(Date);
+        expect(stats?.metrics).toBeDefined();
+        expect(stats?.metrics.models).toEqual({});
+    });
+    it('should update metrics when the uiTelemetryService emits an update', () => {
+        const contextRef = { current: undefined };
+        render(_jsx(SessionStatsProvider, { children: _jsx(TestHarness, { contextRef: contextRef }) }));
+        const newMetrics = {
+            models: {
+                'gemini-pro': {
+                    api: {
+                        totalRequests: 1,
+                        totalErrors: 0,
+                        totalLatencyMs: 123,
+                    },
+                    tokens: {
+                        prompt: 100,
+                        candidates: 200,
+                        total: 300,
+                        cached: 50,
+                        thoughts: 20,
+                        tool: 10,
+                    },
+                },
+            },
+            tools: {
+                totalCalls: 1,
+                totalSuccess: 1,
+                totalFail: 0,
+                totalDurationMs: 456,
+                totalDecisions: {
+                    accept: 1,
+                    reject: 0,
+                    modify: 0,
+                },
+                byName: {
+                    'test-tool': {
+                        count: 1,
+                        success: 1,
+                        fail: 0,
+                        durationMs: 456,
+                        decisions: {
+                            accept: 1,
+                            reject: 0,
+                            modify: 0,
+                        },
+                    },
+                },
+            },
+        };
+        act(() => {
+            uiTelemetryService.emit('update', {
+                metrics: newMetrics,
+                lastPromptTokenCount: 100,
+            });
+        });
+        const stats = contextRef.current?.stats;
+        expect(stats?.metrics).toEqual(newMetrics);
+        expect(stats?.lastPromptTokenCount).toBe(100);
+    });
+    it('should throw an error when useSessionStats is used outside of a provider', () => {
+        // Suppress console.error for this test since we expect an error
+        const consoleSpy = vi.spyOn(console, 'error').mockImplementation(() => { });
+        try {
+            // Expect renderHook itself to throw when the hook is used outside a provider
+            expect(() => {
+                renderHook(() => useSessionStats());
+            }).toThrow('useSessionStats must be used within a SessionStatsProvider');
+        }
+        finally {
+            consoleSpy.mockRestore();
+        }
+    });
+});
+//# sourceMappingURL=SessionContext.test.js.map

package/dist/src/ui/contexts/SessionContext.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SessionContext.test.js","sourceRoot":"","sources":["../../../../src/ui/contexts/SessionContext.test.tsx"],"names":[],"mappings":";AAOA,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,GAAG,EAAE,MAAM,sBAAsB,CAAC;AAC3C,OAAO,EACL,oBAAoB,EACpB,eAAe,GAEhB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D;;;;GAIG;AACH,MAAM,WAAW,GAAG,CAAC,EACnB,UAAU,GAGX,EAAE,EAAE;IACH,UAAU,CAAC,OAAO,GAAG,eAAe,EAAE,CAAC;IACvC,OAAO,IAAI,CAAC;AACd,CAAC,CAAC;AAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,UAAU,GAEZ,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;QAE3B,MAAM,CACJ,KAAC,oBAAoB,cACnB,KAAC,WAAW,IAAC,UAAU,EAAE,UAAU,GAAI,GAClB,CACxB,CAAC;QAEF,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC;QAExC,MAAM,CAAC,KAAK,EAAE,gBAAgB,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,UAAU,GAEZ,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;QAE3B,MAAM,CACJ,KAAC,oBAAoB,cACnB,KAAC,WAAW,IAAC,UAAU,EAAE,UAAU,GAAI,GAClB,CACxB,CAAC;QAEF,MAAM,UAAU,GAAmB;YACjC,MAAM,EAAE;gBACN,YAAY,EAAE;oBACZ,GAAG,EAAE;wBACH,aAAa,EAAE,CAAC;wBAChB,WAAW,EAAE,CAAC;wBACd,cAAc,EAAE,GAAG;qBACpB;oBACD,MAAM,EAAE;wBACN,MAAM,EAAE,GAAG;wBACX,UAAU,EAAE,GAAG;wBACf,KAAK,EAAE,GAAG;wBACV,MAAM,EAAE,EAAE;wBACV,QAAQ,EAAE,EAAE;wBACZ,IAAI,EAAE,EAAE;qBACT;iBACF;aACF;YACD,KAAK,EAAE;gBACL,UAAU,EAAE,CAAC;gBACb,YAAY,EAAE,CAAC;gBACf,SAAS,EAAE,CAAC;gBACZ,eAAe,EAAE,GAAG;gBACpB,cAAc,EAAE;oBACd,MAAM,EAAE,CAAC;oBACT,MAAM,EAAE,CAAC;oBACT,MAAM,EAAE,CAAC;iBACV;gBACD,MAAM,EAAE;oBACN,WAAW,EAAE;wBACX,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE,CAAC;wBACV,IAAI,EAAE,CAAC;wBACP,UAAU,EAAE,GAAG;wBACf,SAAS,EAAE;4BACT,MAAM,EAAE,CAAC;4BACT,MAAM,EAAE,CAAC;4BACT,MAAM,EAAE,CAAC;yBACV;qBACF;iBACF;aACF;SACF,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,kBAAkB,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAChC,OAAO,EAAE,UAAU;gBACnB,oBAAoB,EAAE,GAAG;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC;QACxC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,EAAE,oBAAoB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;QAClF,gEAAgE;QAChE,MAAM,UAAU,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAE3E,IAAI,CAAC;YACH,6EAA6E;YAC7E,MAAM,CAAC,GAAG,EAAE;gBACV,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,EAAE,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC,OAAO,CAAC,4DAA4D,CAAC,CAAC;QAC3E,CAAC;gBAAS,CAAC;YACT,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/hooks/useGeminiStream.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};