@google/gemini-cli 0.1.22 → 0.3.0-nightly.20250821.653267a6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +32 -2
- package/dist/package.json +2 -2
- package/dist/src/config/config.js +2 -0
- package/dist/src/config/config.js.map +1 -1
- package/dist/src/config/extension.d.ts +0 -1
- package/dist/src/config/extension.js +3 -2
- package/dist/src/config/extension.js.map +1 -1
- package/dist/src/config/settings.d.ts +1 -2
- package/dist/src/config/settings.js +48 -47
- package/dist/src/config/settings.js.map +1 -1
- package/dist/src/config/settingsSchema.d.ts +33 -6
- package/dist/src/config/settingsSchema.js +33 -6
- package/dist/src/config/settingsSchema.js.map +1 -1
- package/dist/src/config/settingsSchema.test.js +1 -1
- package/dist/src/config/settingsSchema.test.js.map +1 -1
- package/dist/src/generated/git-commit.d.ts +2 -2
- package/dist/src/generated/git-commit.js +2 -2
- package/dist/src/generated/git-commit.js.map +1 -1
- package/dist/src/nonInteractiveCli.js +1 -2
- package/dist/src/nonInteractiveCli.js.map +1 -1
- package/dist/src/services/FileCommandLoader.js +4 -3
- package/dist/src/services/FileCommandLoader.js.map +1 -1
- package/dist/src/services/prompt-processors/shellProcessor.js +2 -10
- package/dist/src/services/prompt-processors/shellProcessor.js.map +1 -1
- package/dist/src/ui/App.js +45 -15
- package/dist/src/ui/App.js.map +1 -1
- package/dist/src/ui/commands/aboutCommand.js +2 -1
- package/dist/src/ui/commands/aboutCommand.js.map +1 -1
- package/dist/src/ui/commands/bugCommand.js +7 -1
- package/dist/src/ui/commands/bugCommand.js.map +1 -1
- package/dist/src/ui/commands/chatCommand.js +2 -1
- package/dist/src/ui/commands/chatCommand.js.map +1 -1
- package/dist/src/ui/commands/ideCommand.js +29 -11
- package/dist/src/ui/commands/ideCommand.js.map +1 -1
- package/dist/src/ui/commands/mcpCommand.js +6 -6
- package/dist/src/ui/commands/mcpCommand.js.map +1 -1
- package/dist/src/ui/commands/restoreCommand.js +2 -6
- package/dist/src/ui/commands/restoreCommand.js.map +1 -1
- package/dist/src/ui/commands/setupGithubCommand.d.ts +1 -0
- package/dist/src/ui/commands/setupGithubCommand.js +39 -1
- package/dist/src/ui/commands/setupGithubCommand.js.map +1 -1
- package/dist/src/ui/commands/setupGithubCommand.test.js +99 -1
- package/dist/src/ui/commands/setupGithubCommand.test.js.map +1 -1
- package/dist/src/ui/commands/toolsCommand.js +1 -1
- package/dist/src/ui/commands/toolsCommand.js.map +1 -1
- package/dist/src/ui/components/Help.js +1 -1
- package/dist/src/ui/components/Help.js.map +1 -1
- package/dist/src/ui/components/InputPrompt.js +6 -1
- package/dist/src/ui/components/InputPrompt.js.map +1 -1
- package/dist/src/ui/components/SettingsDialog.js +332 -62
- package/dist/src/ui/components/SettingsDialog.js.map +1 -1
- package/dist/src/ui/components/SettingsDialog.test.js +107 -94
- package/dist/src/ui/components/SettingsDialog.test.js.map +1 -1
- package/dist/src/ui/components/StatsDisplay.js +8 -7
- package/dist/src/ui/components/StatsDisplay.js.map +1 -1
- package/dist/src/ui/components/shared/text-buffer.js +6 -0
- package/dist/src/ui/components/shared/text-buffer.js.map +1 -1
- package/dist/src/ui/contexts/KeypressContext.js +24 -25
- package/dist/src/ui/contexts/KeypressContext.js.map +1 -1
- package/dist/src/ui/contexts/KeypressContext.test.js +36 -37
- package/dist/src/ui/contexts/KeypressContext.test.js.map +1 -1
- package/dist/src/ui/contexts/SessionContext.d.ts +2 -0
- package/dist/src/ui/contexts/SessionContext.js.map +1 -1
- package/dist/src/ui/hooks/atCommandProcessor.js +1 -3
- package/dist/src/ui/hooks/atCommandProcessor.js.map +1 -1
- package/dist/src/ui/hooks/atCommandProcessor.test.js +29 -7
- package/dist/src/ui/hooks/atCommandProcessor.test.js.map +1 -1
- package/dist/src/ui/hooks/shellCommandProcessor.js +8 -13
- package/dist/src/ui/hooks/shellCommandProcessor.js.map +1 -1
- package/dist/src/ui/hooks/shellCommandProcessor.test.js +11 -15
- package/dist/src/ui/hooks/shellCommandProcessor.test.js.map +1 -1
- package/dist/src/ui/hooks/slashCommandProcessor.js +17 -5
- package/dist/src/ui/hooks/slashCommandProcessor.js.map +1 -1
- package/dist/src/ui/hooks/useGeminiStream.js +16 -7
- package/dist/src/ui/hooks/useGeminiStream.js.map +1 -1
- package/dist/src/ui/hooks/useLogger.d.ts +2 -2
- package/dist/src/ui/hooks/useLogger.js +3 -3
- package/dist/src/ui/hooks/useLogger.js.map +1 -1
- package/dist/src/ui/hooks/useMessageQueue.d.ts +22 -0
- package/dist/src/ui/hooks/useMessageQueue.js +49 -0
- package/dist/src/ui/hooks/useMessageQueue.js.map +1 -0
- package/dist/src/ui/hooks/useMessageQueue.test.d.ts +6 -0
- package/dist/src/ui/hooks/useMessageQueue.test.js +158 -0
- package/dist/src/ui/hooks/useMessageQueue.test.js.map +1 -0
- package/dist/src/ui/hooks/usePrivacySettings.js +6 -2
- package/dist/src/ui/hooks/usePrivacySettings.js.map +1 -1
- package/dist/src/ui/hooks/usePrivacySettings.test.d.ts +6 -0
- package/dist/src/ui/hooks/usePrivacySettings.test.js +154 -0
- package/dist/src/ui/hooks/usePrivacySettings.test.js.map +1 -0
- package/dist/src/ui/hooks/useShellHistory.d.ts +2 -1
- package/dist/src/ui/hooks/useShellHistory.js +7 -8
- package/dist/src/ui/hooks/useShellHistory.js.map +1 -1
- package/dist/src/ui/hooks/useShellHistory.test.js +26 -1
- package/dist/src/ui/hooks/useShellHistory.test.js.map +1 -1
- package/dist/src/ui/hooks/useToolScheduler.test.js +5 -7
- package/dist/src/ui/hooks/useToolScheduler.test.js.map +1 -1
- package/dist/src/ui/utils/commandUtils.js +20 -8
- package/dist/src/ui/utils/commandUtils.js.map +1 -1
- package/dist/src/ui/utils/commandUtils.test.js +9 -12
- package/dist/src/ui/utils/commandUtils.test.js.map +1 -1
- package/dist/src/ui/utils/computeStats.js +3 -1
- package/dist/src/ui/utils/computeStats.js.map +1 -1
- package/dist/src/utils/cleanup.js +3 -2
- package/dist/src/utils/cleanup.js.map +1 -1
- package/dist/src/utils/settingsUtils.d.ts +8 -0
- package/dist/src/utils/settingsUtils.js +16 -7
- package/dist/src/utils/settingsUtils.js.map +1 -1
- package/dist/src/utils/settingsUtils.test.js +6 -6
- package/dist/src/utils/settingsUtils.test.js.map +1 -1
- package/dist/src/zed-integration/fileSystemService.d.ts +19 -0
- package/dist/src/zed-integration/fileSystemService.js +43 -0
- package/dist/src/zed-integration/fileSystemService.js.map +1 -0
- package/dist/src/zed-integration/zedIntegration.js +17 -4
- package/dist/src/zed-integration/zedIntegration.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +3 -3
|
@@ -0,0 +1,158 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
7
|
+
import { renderHook, act } from '@testing-library/react';
|
|
8
|
+
import { useMessageQueue } from './useMessageQueue.js';
|
|
9
|
+
import { StreamingState } from '../types.js';
|
|
10
|
+
describe('useMessageQueue', () => {
|
|
11
|
+
let mockSubmitQuery;
|
|
12
|
+
beforeEach(() => {
|
|
13
|
+
mockSubmitQuery = vi.fn();
|
|
14
|
+
vi.useFakeTimers();
|
|
15
|
+
});
|
|
16
|
+
afterEach(() => {
|
|
17
|
+
vi.useRealTimers();
|
|
18
|
+
vi.clearAllMocks();
|
|
19
|
+
});
|
|
20
|
+
it('should initialize with empty queue', () => {
|
|
21
|
+
const { result } = renderHook(() => useMessageQueue({
|
|
22
|
+
streamingState: StreamingState.Idle,
|
|
23
|
+
submitQuery: mockSubmitQuery,
|
|
24
|
+
}));
|
|
25
|
+
expect(result.current.messageQueue).toEqual([]);
|
|
26
|
+
expect(result.current.getQueuedMessagesText()).toBe('');
|
|
27
|
+
});
|
|
28
|
+
it('should add messages to queue', () => {
|
|
29
|
+
const { result } = renderHook(() => useMessageQueue({
|
|
30
|
+
streamingState: StreamingState.Responding,
|
|
31
|
+
submitQuery: mockSubmitQuery,
|
|
32
|
+
}));
|
|
33
|
+
act(() => {
|
|
34
|
+
result.current.addMessage('Test message 1');
|
|
35
|
+
result.current.addMessage('Test message 2');
|
|
36
|
+
});
|
|
37
|
+
expect(result.current.messageQueue).toEqual([
|
|
38
|
+
'Test message 1',
|
|
39
|
+
'Test message 2',
|
|
40
|
+
]);
|
|
41
|
+
});
|
|
42
|
+
it('should filter out empty messages', () => {
|
|
43
|
+
const { result } = renderHook(() => useMessageQueue({
|
|
44
|
+
streamingState: StreamingState.Responding,
|
|
45
|
+
submitQuery: mockSubmitQuery,
|
|
46
|
+
}));
|
|
47
|
+
act(() => {
|
|
48
|
+
result.current.addMessage('Valid message');
|
|
49
|
+
result.current.addMessage(' '); // Only whitespace
|
|
50
|
+
result.current.addMessage(''); // Empty
|
|
51
|
+
result.current.addMessage('Another valid message');
|
|
52
|
+
});
|
|
53
|
+
expect(result.current.messageQueue).toEqual([
|
|
54
|
+
'Valid message',
|
|
55
|
+
'Another valid message',
|
|
56
|
+
]);
|
|
57
|
+
});
|
|
58
|
+
it('should clear queue', () => {
|
|
59
|
+
const { result } = renderHook(() => useMessageQueue({
|
|
60
|
+
streamingState: StreamingState.Responding,
|
|
61
|
+
submitQuery: mockSubmitQuery,
|
|
62
|
+
}));
|
|
63
|
+
act(() => {
|
|
64
|
+
result.current.addMessage('Test message');
|
|
65
|
+
});
|
|
66
|
+
expect(result.current.messageQueue).toEqual(['Test message']);
|
|
67
|
+
act(() => {
|
|
68
|
+
result.current.clearQueue();
|
|
69
|
+
});
|
|
70
|
+
expect(result.current.messageQueue).toEqual([]);
|
|
71
|
+
});
|
|
72
|
+
it('should return queued messages as text with double newlines', () => {
|
|
73
|
+
const { result } = renderHook(() => useMessageQueue({
|
|
74
|
+
streamingState: StreamingState.Responding,
|
|
75
|
+
submitQuery: mockSubmitQuery,
|
|
76
|
+
}));
|
|
77
|
+
act(() => {
|
|
78
|
+
result.current.addMessage('Message 1');
|
|
79
|
+
result.current.addMessage('Message 2');
|
|
80
|
+
result.current.addMessage('Message 3');
|
|
81
|
+
});
|
|
82
|
+
expect(result.current.getQueuedMessagesText()).toBe('Message 1\n\nMessage 2\n\nMessage 3');
|
|
83
|
+
});
|
|
84
|
+
it('should auto-submit queued messages when transitioning to Idle', () => {
|
|
85
|
+
const { result, rerender } = renderHook(({ streamingState }) => useMessageQueue({
|
|
86
|
+
streamingState,
|
|
87
|
+
submitQuery: mockSubmitQuery,
|
|
88
|
+
}), {
|
|
89
|
+
initialProps: { streamingState: StreamingState.Responding },
|
|
90
|
+
});
|
|
91
|
+
// Add some messages
|
|
92
|
+
act(() => {
|
|
93
|
+
result.current.addMessage('Message 1');
|
|
94
|
+
result.current.addMessage('Message 2');
|
|
95
|
+
});
|
|
96
|
+
expect(result.current.messageQueue).toEqual(['Message 1', 'Message 2']);
|
|
97
|
+
// Transition to Idle
|
|
98
|
+
rerender({ streamingState: StreamingState.Idle });
|
|
99
|
+
expect(mockSubmitQuery).toHaveBeenCalledWith('Message 1\n\nMessage 2');
|
|
100
|
+
expect(result.current.messageQueue).toEqual([]);
|
|
101
|
+
});
|
|
102
|
+
it('should not auto-submit when queue is empty', () => {
|
|
103
|
+
const { rerender } = renderHook(({ streamingState }) => useMessageQueue({
|
|
104
|
+
streamingState,
|
|
105
|
+
submitQuery: mockSubmitQuery,
|
|
106
|
+
}), {
|
|
107
|
+
initialProps: { streamingState: StreamingState.Responding },
|
|
108
|
+
});
|
|
109
|
+
// Transition to Idle with empty queue
|
|
110
|
+
rerender({ streamingState: StreamingState.Idle });
|
|
111
|
+
expect(mockSubmitQuery).not.toHaveBeenCalled();
|
|
112
|
+
});
|
|
113
|
+
it('should not auto-submit when not transitioning to Idle', () => {
|
|
114
|
+
const { result, rerender } = renderHook(({ streamingState }) => useMessageQueue({
|
|
115
|
+
streamingState,
|
|
116
|
+
submitQuery: mockSubmitQuery,
|
|
117
|
+
}), {
|
|
118
|
+
initialProps: { streamingState: StreamingState.Responding },
|
|
119
|
+
});
|
|
120
|
+
// Add messages
|
|
121
|
+
act(() => {
|
|
122
|
+
result.current.addMessage('Message 1');
|
|
123
|
+
});
|
|
124
|
+
// Transition to WaitingForConfirmation (not Idle)
|
|
125
|
+
rerender({ streamingState: StreamingState.WaitingForConfirmation });
|
|
126
|
+
expect(mockSubmitQuery).not.toHaveBeenCalled();
|
|
127
|
+
expect(result.current.messageQueue).toEqual(['Message 1']);
|
|
128
|
+
});
|
|
129
|
+
it('should handle multiple state transitions correctly', () => {
|
|
130
|
+
const { result, rerender } = renderHook(({ streamingState }) => useMessageQueue({
|
|
131
|
+
streamingState,
|
|
132
|
+
submitQuery: mockSubmitQuery,
|
|
133
|
+
}), {
|
|
134
|
+
initialProps: { streamingState: StreamingState.Idle },
|
|
135
|
+
});
|
|
136
|
+
// Start responding
|
|
137
|
+
rerender({ streamingState: StreamingState.Responding });
|
|
138
|
+
// Add messages while responding
|
|
139
|
+
act(() => {
|
|
140
|
+
result.current.addMessage('First batch');
|
|
141
|
+
});
|
|
142
|
+
// Go back to idle - should submit
|
|
143
|
+
rerender({ streamingState: StreamingState.Idle });
|
|
144
|
+
expect(mockSubmitQuery).toHaveBeenCalledWith('First batch');
|
|
145
|
+
expect(result.current.messageQueue).toEqual([]);
|
|
146
|
+
// Start responding again
|
|
147
|
+
rerender({ streamingState: StreamingState.Responding });
|
|
148
|
+
// Add more messages
|
|
149
|
+
act(() => {
|
|
150
|
+
result.current.addMessage('Second batch');
|
|
151
|
+
});
|
|
152
|
+
// Go back to idle - should submit again
|
|
153
|
+
rerender({ streamingState: StreamingState.Idle });
|
|
154
|
+
expect(mockSubmitQuery).toHaveBeenCalledWith('Second batch');
|
|
155
|
+
expect(mockSubmitQuery).toHaveBeenCalledTimes(2);
|
|
156
|
+
});
|
|
157
|
+
});
|
|
158
|
+
//# sourceMappingURL=useMessageQueue.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useMessageQueue.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useMessageQueue.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,eAAyC,CAAC;IAE9C,UAAU,CAAC,GAAG,EAAE;QACd,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1B,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,eAAe,CAAC;YACd,cAAc,EAAE,cAAc,CAAC,IAAI;YACnC,WAAW,EAAE,eAAe;SAC7B,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,eAAe,CAAC;YACd,cAAc,EAAE,cAAc,CAAC,UAAU;YACzC,WAAW,EAAE,eAAe;SAC7B,CAAC,CACH,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;YAC1C,gBAAgB;YAChB,gBAAgB;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,eAAe,CAAC;YACd,cAAc,EAAE,cAAc,CAAC,UAAU;YACzC,WAAW,EAAE,eAAe;SAC7B,CAAC,CACH,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;YAC3C,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,kBAAkB;YACpD,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ;YACvC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;YAC1C,eAAe;YACf,uBAAuB;SACxB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,eAAe,CAAC;YACd,cAAc,EAAE,cAAc,CAAC,UAAU;YACzC,WAAW,EAAE,eAAe;SAC7B,CAAC,CACH,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC;QAE9D,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CACjC,eAAe,CAAC;YACd,cAAc,EAAE,cAAc,CAAC,UAAU;YACzC,WAAW,EAAE,eAAe;SAC7B,CAAC,CACH,CAAC;QAEF,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC,IAAI,CACjD,qCAAqC,CACtC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CACrC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CACrB,eAAe,CAAC;YACd,cAAc;YACd,WAAW,EAAE,eAAe;SAC7B,CAAC,EACJ;YACE,YAAY,EAAE,EAAE,cAAc,EAAE,cAAc,CAAC,UAAU,EAAE;SAC5D,CACF,CAAC;QAEF,oBAAoB;QACpB,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;QAExE,qBAAqB;QACrB,QAAQ,CAAC,EAAE,cAAc,EAAE,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;QAElD,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAC7B,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CACrB,eAAe,CAAC;YACd,cAAc;YACd,WAAW,EAAE,eAAe;SAC7B,CAAC,EACJ;YACE,YAAY,EAAE,EAAE,cAAc,EAAE,cAAc,CAAC,UAAU,EAAE;SAC5D,CACF,CAAC;QAEF,sCAAsC;QACtC,QAAQ,CAAC,EAAE,cAAc,EAAE,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;QAElD,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CACrC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CACrB,eAAe,CAAC;YACd,cAAc;YACd,WAAW,EAAE,eAAe;SAC7B,CAAC,EACJ;YACE,YAAY,EAAE,EAAE,cAAc,EAAE,cAAc,CAAC,UAAU,EAAE;SAC5D,CACF,CAAC;QAEF,eAAe;QACf,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,kDAAkD;QAClD,QAAQ,CAAC,EAAE,cAAc,EAAE,cAAc,CAAC,sBAAsB,EAAE,CAAC,CAAC;QAEpE,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CACrC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CACrB,eAAe,CAAC;YACd,cAAc;YACd,WAAW,EAAE,eAAe;SAC7B,CAAC,EACJ;YACE,YAAY,EAAE,EAAE,cAAc,EAAE,cAAc,CAAC,IAAI,EAAE;SACtD,CACF,CAAC;QAEF,mBAAmB;QACnB,QAAQ,CAAC,EAAE,cAAc,EAAE,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC;QAExD,gCAAgC;QAChC,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,kCAAkC;QAClC,QAAQ,CAAC,EAAE,cAAc,EAAE,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;QAElD,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEhD,yBAAyB;QACzB,QAAQ,CAAC,EAAE,cAAc,EAAE,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC;QAExD,oBAAoB;QACpB,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,wCAAwC;QACxC,QAAQ,CAAC,EAAE,cAAc,EAAE,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;QAElD,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QAC7D,MAAM,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
6
|
import { useState, useEffect, useCallback } from 'react';
|
|
7
|
-
import { CodeAssistServer, UserTierId } from '@google/gemini-cli-core';
|
|
7
|
+
import { CodeAssistServer, UserTierId, LoggingContentGenerator, } from '@google/gemini-cli-core';
|
|
8
8
|
export const usePrivacySettings = (config) => {
|
|
9
9
|
const [privacyState, setPrivacyState] = useState({
|
|
10
10
|
isLoading: true,
|
|
@@ -65,7 +65,11 @@ export const usePrivacySettings = (config) => {
|
|
|
65
65
|
};
|
|
66
66
|
};
|
|
67
67
|
function getCodeAssistServer(config) {
|
|
68
|
-
|
|
68
|
+
let server = config.getGeminiClient().getContentGenerator();
|
|
69
|
+
// Unwrap LoggingContentGenerator if present
|
|
70
|
+
if (server instanceof LoggingContentGenerator) {
|
|
71
|
+
server = server.getWrapped();
|
|
72
|
+
}
|
|
69
73
|
// Neither of these cases should ever happen.
|
|
70
74
|
if (!(server instanceof CodeAssistServer)) {
|
|
71
75
|
throw new Error('Oauth not being used');
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"usePrivacySettings.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePrivacySettings.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,
|
|
1
|
+
{"version":3,"file":"usePrivacySettings.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePrivacySettings.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,EAEL,gBAAgB,EAChB,UAAU,EACV,uBAAuB,GACxB,MAAM,yBAAyB,CAAC;AASjC,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,MAAc,EAAE,EAAE;IACnD,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAe;QAC7D,SAAS,EAAE,IAAI;KAChB,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,iBAAiB,GAAG,KAAK,IAAI,EAAE;YACnC,eAAe,CAAC;gBACd,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;gBAC3C,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;gBACnC,IAAI,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;oBAC7B,0DAA0D;oBAC1D,uDAAuD;oBACvD,eAAe,CAAC;wBACd,SAAS,EAAE,KAAK;wBAChB,UAAU,EAAE,KAAK;qBAClB,CAAC,CAAC;oBACH,OAAO;gBACT,CAAC;gBAED,MAAM,KAAK,GAAG,MAAM,4BAA4B,CAAC,MAAM,CAAC,CAAC;gBACzD,eAAe,CAAC;oBACd,SAAS,EAAE,KAAK;oBAChB,UAAU,EAAE,IAAI;oBAChB,mBAAmB,EAAE,KAAK;iBAC3B,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,eAAe,CAAC;oBACd,SAAS,EAAE,KAAK;oBAChB,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;iBAClD,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC;QACF,iBAAiB,EAAE,CAAC;IACtB,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAEb,MAAM,yBAAyB,GAAG,WAAW,CAC3C,KAAK,EAAE,KAAc,EAAE,EAAE;QACvB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAC3C,MAAM,YAAY,GAAG,MAAM,4BAA4B,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YACvE,eAAe,CAAC;gBACd,SAAS,EAAE,KAAK;gBAChB,UAAU,EAAE,IAAI;gBAChB,mBAAmB,EAAE,YAAY;aAClC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,eAAe,CAAC;gBACd,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;aAClD,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EACD,CAAC,MAAM,CAAC,CACT,CAAC;IAEF,OAAO;QACL,YAAY;QACZ,yBAAyB;KAC1B,CAAC;AACJ,CAAC,CAAC;AAEF,SAAS,mBAAmB,CAAC,MAAc;IACzC,IAAI,MAAM,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC,mBAAmB,EAAE,CAAC;IAE5D,4CAA4C;IAC5C,IAAI,MAAM,YAAY,uBAAuB,EAAE,CAAC;QAC9C,MAAM,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;IAC/B,CAAC;IAED,6CAA6C;IAC7C,IAAI,CAAC,CAAC,MAAM,YAAY,gBAAgB,CAAC,EAAE,CAAC;QAC1C,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;SAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,OAAO,CAAC,MAAwB;IAC7C,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC;QAC1C,uBAAuB,EAAE,MAAM,CAAC,SAAS;QACzC,QAAQ,EAAE;YACR,OAAO,EAAE,iBAAiB;YAC1B,QAAQ,EAAE,sBAAsB;YAChC,UAAU,EAAE,QAAQ;YACpB,WAAW,EAAE,MAAM,CAAC,SAAS;SAC9B;KACF,CAAC,CAAC;IACH,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;AAChC,CAAC;AAED,KAAK,UAAU,4BAA4B,CACzC,MAAwB;IAExB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,8BAA8B,EAAE,CAAC;QAC3D,OAAO,IAAI,CAAC,2BAA2B,CAAC;IAC1C,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;YAC9D,MAAM,WAAW,GAAG,KAInB,CAAC;YACF,IAAI,WAAW,CAAC,QAAQ,EAAE,MAAM,KAAK,GAAG,EAAE,CAAC;gBACzC,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,4BAA4B,CACzC,MAAwB,EACxB,KAAc;IAEd,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,8BAA8B,CAAC;QACvD,uBAAuB,EAAE,MAAM,CAAC,SAAS;QACzC,2BAA2B,EAAE,KAAK;KACnC,CAAC,CAAC;IACH,OAAO,IAAI,CAAC,2BAA2B,CAAC;AAC1C,CAAC"}
|
|
@@ -0,0 +1,154 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @license
|
|
3
|
+
* Copyright 2025 Google LLC
|
|
4
|
+
* SPDX-License-Identifier: Apache-2.0
|
|
5
|
+
*/
|
|
6
|
+
import { describe, it, expect, beforeEach, vi } from 'vitest';
|
|
7
|
+
import { renderHook, waitFor } from '@testing-library/react';
|
|
8
|
+
import { CodeAssistServer, LoggingContentGenerator, UserTierId, } from '@google/gemini-cli-core';
|
|
9
|
+
import { usePrivacySettings } from './usePrivacySettings.js';
|
|
10
|
+
// Mock the dependencies
|
|
11
|
+
vi.mock('@google/gemini-cli-core', () => {
|
|
12
|
+
// Mock classes for instanceof checks
|
|
13
|
+
class MockCodeAssistServer {
|
|
14
|
+
projectId = 'test-project-id';
|
|
15
|
+
loadCodeAssist = vi.fn();
|
|
16
|
+
getCodeAssistGlobalUserSetting = vi.fn();
|
|
17
|
+
setCodeAssistGlobalUserSetting = vi.fn();
|
|
18
|
+
constructor(_client, _projectId, _httpOptions, _sessionId, _userTier) { }
|
|
19
|
+
}
|
|
20
|
+
class MockLoggingContentGenerator {
|
|
21
|
+
getWrapped = vi.fn();
|
|
22
|
+
constructor(_wrapped, _config) { }
|
|
23
|
+
}
|
|
24
|
+
return {
|
|
25
|
+
Config: vi.fn(),
|
|
26
|
+
CodeAssistServer: MockCodeAssistServer,
|
|
27
|
+
LoggingContentGenerator: MockLoggingContentGenerator,
|
|
28
|
+
GeminiClient: vi.fn(),
|
|
29
|
+
UserTierId: {
|
|
30
|
+
FREE: 'free-tier',
|
|
31
|
+
LEGACY: 'legacy-tier',
|
|
32
|
+
STANDARD: 'standard-tier',
|
|
33
|
+
},
|
|
34
|
+
};
|
|
35
|
+
});
|
|
36
|
+
describe('usePrivacySettings', () => {
|
|
37
|
+
let mockConfig;
|
|
38
|
+
let mockClient;
|
|
39
|
+
let mockCodeAssistServer;
|
|
40
|
+
let mockLoggingContentGenerator;
|
|
41
|
+
beforeEach(() => {
|
|
42
|
+
vi.clearAllMocks();
|
|
43
|
+
// Create mock CodeAssistServer instance
|
|
44
|
+
mockCodeAssistServer = new CodeAssistServer(null, 'test-project-id');
|
|
45
|
+
mockCodeAssistServer.loadCodeAssist.mockResolvedValue({
|
|
46
|
+
currentTier: { id: UserTierId.FREE },
|
|
47
|
+
});
|
|
48
|
+
mockCodeAssistServer.getCodeAssistGlobalUserSetting.mockResolvedValue({
|
|
49
|
+
freeTierDataCollectionOptin: true,
|
|
50
|
+
});
|
|
51
|
+
mockCodeAssistServer.setCodeAssistGlobalUserSetting.mockResolvedValue({
|
|
52
|
+
freeTierDataCollectionOptin: false,
|
|
53
|
+
});
|
|
54
|
+
// Create mock LoggingContentGenerator that wraps the CodeAssistServer
|
|
55
|
+
mockLoggingContentGenerator = new LoggingContentGenerator(mockCodeAssistServer, null);
|
|
56
|
+
mockLoggingContentGenerator.getWrapped.mockReturnValue(mockCodeAssistServer);
|
|
57
|
+
// Create mock GeminiClient
|
|
58
|
+
mockClient = {
|
|
59
|
+
getContentGenerator: vi.fn().mockReturnValue(mockLoggingContentGenerator),
|
|
60
|
+
};
|
|
61
|
+
// Create mock Config
|
|
62
|
+
mockConfig = {
|
|
63
|
+
getGeminiClient: vi.fn().mockReturnValue(mockClient),
|
|
64
|
+
};
|
|
65
|
+
});
|
|
66
|
+
it('should handle LoggingContentGenerator wrapper correctly and not throw "Oauth not being used" error', async () => {
|
|
67
|
+
const { result } = renderHook(() => usePrivacySettings(mockConfig));
|
|
68
|
+
// Initial state should be loading
|
|
69
|
+
expect(result.current.privacyState.isLoading).toBe(true);
|
|
70
|
+
expect(result.current.privacyState.error).toBeUndefined();
|
|
71
|
+
// Wait for the hook to complete
|
|
72
|
+
await waitFor(() => {
|
|
73
|
+
expect(result.current.privacyState.isLoading).toBe(false);
|
|
74
|
+
});
|
|
75
|
+
// Should not have the "Oauth not being used" error
|
|
76
|
+
expect(result.current.privacyState.error).toBeUndefined();
|
|
77
|
+
expect(result.current.privacyState.isFreeTier).toBe(true);
|
|
78
|
+
expect(result.current.privacyState.dataCollectionOptIn).toBe(true);
|
|
79
|
+
// Verify that getWrapped was called to unwrap the LoggingContentGenerator
|
|
80
|
+
expect(mockLoggingContentGenerator.getWrapped).toHaveBeenCalled();
|
|
81
|
+
});
|
|
82
|
+
it('should work with direct CodeAssistServer (no wrapper)', async () => {
|
|
83
|
+
// Test case where the content generator is directly a CodeAssistServer
|
|
84
|
+
const directServer = new CodeAssistServer(null, 'test-project-id');
|
|
85
|
+
directServer.loadCodeAssist.mockResolvedValue({
|
|
86
|
+
currentTier: { id: UserTierId.FREE },
|
|
87
|
+
});
|
|
88
|
+
directServer.getCodeAssistGlobalUserSetting.mockResolvedValue({
|
|
89
|
+
freeTierDataCollectionOptin: true,
|
|
90
|
+
});
|
|
91
|
+
mockClient.getContentGenerator = vi.fn().mockReturnValue(directServer);
|
|
92
|
+
const { result } = renderHook(() => usePrivacySettings(mockConfig));
|
|
93
|
+
await waitFor(() => {
|
|
94
|
+
expect(result.current.privacyState.isLoading).toBe(false);
|
|
95
|
+
});
|
|
96
|
+
expect(result.current.privacyState.error).toBeUndefined();
|
|
97
|
+
expect(result.current.privacyState.isFreeTier).toBe(true);
|
|
98
|
+
expect(result.current.privacyState.dataCollectionOptIn).toBe(true);
|
|
99
|
+
});
|
|
100
|
+
it('should handle paid tier users correctly', async () => {
|
|
101
|
+
// Mock paid tier response
|
|
102
|
+
mockCodeAssistServer.loadCodeAssist.mockResolvedValue({
|
|
103
|
+
currentTier: { id: UserTierId.STANDARD },
|
|
104
|
+
});
|
|
105
|
+
const { result } = renderHook(() => usePrivacySettings(mockConfig));
|
|
106
|
+
await waitFor(() => {
|
|
107
|
+
expect(result.current.privacyState.isLoading).toBe(false);
|
|
108
|
+
});
|
|
109
|
+
expect(result.current.privacyState.error).toBeUndefined();
|
|
110
|
+
expect(result.current.privacyState.isFreeTier).toBe(false);
|
|
111
|
+
expect(result.current.privacyState.dataCollectionOptIn).toBeUndefined();
|
|
112
|
+
});
|
|
113
|
+
it('should throw error when content generator is not a CodeAssistServer', async () => {
|
|
114
|
+
// Mock a non-CodeAssistServer content generator
|
|
115
|
+
const mockOtherGenerator = { someOtherMethod: vi.fn() };
|
|
116
|
+
mockLoggingContentGenerator.getWrapped.mockReturnValue(mockOtherGenerator);
|
|
117
|
+
const { result } = renderHook(() => usePrivacySettings(mockConfig));
|
|
118
|
+
await waitFor(() => {
|
|
119
|
+
expect(result.current.privacyState.isLoading).toBe(false);
|
|
120
|
+
});
|
|
121
|
+
expect(result.current.privacyState.error).toBe('Oauth not being used');
|
|
122
|
+
});
|
|
123
|
+
it('should throw error when CodeAssistServer has no projectId', async () => {
|
|
124
|
+
// Mock CodeAssistServer without projectId
|
|
125
|
+
const mockServerNoProject = {
|
|
126
|
+
...mockCodeAssistServer,
|
|
127
|
+
projectId: undefined,
|
|
128
|
+
};
|
|
129
|
+
mockLoggingContentGenerator.getWrapped.mockReturnValue(mockServerNoProject);
|
|
130
|
+
const { result } = renderHook(() => usePrivacySettings(mockConfig));
|
|
131
|
+
await waitFor(() => {
|
|
132
|
+
expect(result.current.privacyState.isLoading).toBe(false);
|
|
133
|
+
});
|
|
134
|
+
expect(result.current.privacyState.error).toBe('Oauth not being used');
|
|
135
|
+
});
|
|
136
|
+
it('should update data collection opt-in setting', async () => {
|
|
137
|
+
const { result } = renderHook(() => usePrivacySettings(mockConfig));
|
|
138
|
+
// Wait for initial load
|
|
139
|
+
await waitFor(() => {
|
|
140
|
+
expect(result.current.privacyState.isLoading).toBe(false);
|
|
141
|
+
});
|
|
142
|
+
// Update the setting
|
|
143
|
+
await result.current.updateDataCollectionOptIn(false);
|
|
144
|
+
// Wait for update to complete
|
|
145
|
+
await waitFor(() => {
|
|
146
|
+
expect(result.current.privacyState.dataCollectionOptIn).toBe(false);
|
|
147
|
+
});
|
|
148
|
+
expect(mockCodeAssistServer.setCodeAssistGlobalUserSetting).toHaveBeenCalledWith({
|
|
149
|
+
cloudaicompanionProject: 'test-project-id',
|
|
150
|
+
freeTierDataCollectionOptin: false,
|
|
151
|
+
});
|
|
152
|
+
});
|
|
153
|
+
});
|
|
154
|
+
//# sourceMappingURL=usePrivacySettings.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usePrivacySettings.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePrivacySettings.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAEL,gBAAgB,EAChB,uBAAuB,EACvB,UAAU,GAGX,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,wBAAwB;AACxB,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACtC,qCAAqC;IACrC,MAAM,oBAAoB;QACxB,SAAS,GAAG,iBAAiB,CAAC;QAC9B,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,8BAA8B,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,8BAA8B,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEzC,YACE,OAAsB,EACtB,UAAmB,EACnB,YAAsC,EACtC,UAAmB,EACnB,SAAsB,IACrB,CAAC;KACL;IAED,MAAM,2BAA2B;QAC/B,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAErB,YACE,QAA2B,EAC3B,OAAiC,IAChC,CAAC;KACL;IAED,OAAO;QACL,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,gBAAgB,EAAE,oBAAoB;QACtC,uBAAuB,EAAE,2BAA2B;QACpD,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,UAAU,EAAE;YACV,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE,eAAe;SAC1B;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,UAAkB,CAAC;IACvB,IAAI,UAAwB,CAAC;IAC7B,IAAI,oBAAsC,CAAC;IAC3C,IAAI,2BAAoD,CAAC;IAEzD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,wCAAwC;QACxC,oBAAoB,GAAG,IAAI,gBAAgB,CACzC,IAA+B,EAC/B,iBAAiB,CACa,CAAC;QAE/B,oBAAoB,CAAC,cACtB,CAAC,iBAAiB,CAAC;YAClB,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,IAAI,EAAE;SACrC,CAAC,CAAC;QAED,oBAAoB,CAAC,8BAGtB,CAAC,iBAAiB,CAAC;YAClB,2BAA2B,EAAE,IAAI;SAClC,CAAC,CAAC;QAED,oBAAoB,CAAC,8BAGtB,CAAC,iBAAiB,CAAC;YAClB,2BAA2B,EAAE,KAAK;SACnC,CAAC,CAAC;QAEH,sEAAsE;QACtE,2BAA2B,GAAG,IAAI,uBAAuB,CACvD,oBAAoB,EACpB,IAAyB,CACY,CAAC;QAEtC,2BAA2B,CAAC,UAC7B,CAAC,eAAe,CAAC,oBAAoB,CAAC,CAAC;QAExC,2BAA2B;QAC3B,UAAU,GAAG;YACX,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,2BAA2B,CAAC;SAC/C,CAAC;QAE7B,qBAAqB;QACrB,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC;SAChC,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oGAAoG,EAAE,KAAK,IAAI,EAAE;QAClH,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,kCAAkC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAE1D,gCAAgC;QAChC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,mDAAmD;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEnE,0EAA0E;QAC1E,MAAM,CAAC,2BAA2B,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,uEAAuE;QACvE,MAAM,YAAY,GAAG,IAAI,gBAAgB,CACvC,IAA+B,EAC/B,iBAAiB,CACa,CAAC;QAChC,YAAY,CAAC,cAA2C,CAAC,iBAAiB,CACzE;YACE,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,IAAI,EAAE;SACrC,CACF,CAAC;QAEA,YAAY,CAAC,8BACd,CAAC,iBAAiB,CAAC;YAClB,2BAA2B,EAAE,IAAI;SAClC,CAAC,CAAC;QAEH,UAAU,CAAC,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;QAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,0BAA0B;QAExB,oBAAoB,CAAC,cACtB,CAAC,iBAAiB,CAAC;YAClB,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,QAAQ,EAAE;SACzC,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,gDAAgD;QAChD,MAAM,kBAAkB,GAAG,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;QAEtD,2BAA2B,CAAC,UAC7B,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEtC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,0CAA0C;QAC1C,MAAM,mBAAmB,GAAG;YAC1B,GAAG,oBAAoB;YACvB,SAAS,EAAE,SAAS;SACrB,CAAC;QAEA,2BAA2B,CAAC,UAC7B,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAEvC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,wBAAwB;QACxB,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAEtD,8BAA8B;QAC9B,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,MAAM,CACJ,oBAAoB,CAAC,8BAA8B,CACpD,CAAC,oBAAoB,CAAC;YACrB,uBAAuB,EAAE,iBAAiB;YAC1C,2BAA2B,EAAE,KAAK;SACnC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -3,6 +3,7 @@
|
|
|
3
3
|
* Copyright 2025 Google LLC
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
+
import { Storage } from '@google/gemini-cli-core';
|
|
6
7
|
export interface UseShellHistoryReturn {
|
|
7
8
|
history: string[];
|
|
8
9
|
addCommandToHistory: (command: string) => void;
|
|
@@ -10,4 +11,4 @@ export interface UseShellHistoryReturn {
|
|
|
10
11
|
getNextCommand: () => string | null;
|
|
11
12
|
resetHistoryPosition: () => void;
|
|
12
13
|
}
|
|
13
|
-
export declare function useShellHistory(projectRoot: string): UseShellHistoryReturn;
|
|
14
|
+
export declare function useShellHistory(projectRoot: string, storage?: Storage): UseShellHistoryReturn;
|
|
@@ -6,12 +6,11 @@
|
|
|
6
6
|
import { useState, useEffect, useCallback } from 'react';
|
|
7
7
|
import * as fs from 'fs/promises';
|
|
8
8
|
import * as path from 'path';
|
|
9
|
-
import { isNodeError,
|
|
10
|
-
const HISTORY_FILE = 'shell_history';
|
|
9
|
+
import { isNodeError, Storage } from '@google/gemini-cli-core';
|
|
11
10
|
const MAX_HISTORY_LENGTH = 100;
|
|
12
|
-
async function getHistoryFilePath(projectRoot) {
|
|
13
|
-
const
|
|
14
|
-
return
|
|
11
|
+
async function getHistoryFilePath(projectRoot, configStorage) {
|
|
12
|
+
const storage = configStorage ?? new Storage(projectRoot);
|
|
13
|
+
return storage.getHistoryFilePath();
|
|
15
14
|
}
|
|
16
15
|
// Handle multiline commands
|
|
17
16
|
async function readHistoryFile(filePath) {
|
|
@@ -54,19 +53,19 @@ async function writeHistoryFile(filePath, history) {
|
|
|
54
53
|
console.error('Error writing shell history:', error);
|
|
55
54
|
}
|
|
56
55
|
}
|
|
57
|
-
export function useShellHistory(projectRoot) {
|
|
56
|
+
export function useShellHistory(projectRoot, storage) {
|
|
58
57
|
const [history, setHistory] = useState([]);
|
|
59
58
|
const [historyIndex, setHistoryIndex] = useState(-1);
|
|
60
59
|
const [historyFilePath, setHistoryFilePath] = useState(null);
|
|
61
60
|
useEffect(() => {
|
|
62
61
|
async function loadHistory() {
|
|
63
|
-
const filePath = await getHistoryFilePath(projectRoot);
|
|
62
|
+
const filePath = await getHistoryFilePath(projectRoot, storage);
|
|
64
63
|
setHistoryFilePath(filePath);
|
|
65
64
|
const loadedHistory = await readHistoryFile(filePath);
|
|
66
65
|
setHistory(loadedHistory.reverse()); // Newest first
|
|
67
66
|
}
|
|
68
67
|
loadHistory();
|
|
69
|
-
}, [projectRoot]);
|
|
68
|
+
}, [projectRoot, storage]);
|
|
70
69
|
const addCommandToHistory = useCallback((command) => {
|
|
71
70
|
if (!command.trim() || !historyFilePath) {
|
|
72
71
|
return;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useShellHistory.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useShellHistory.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,
|
|
1
|
+
{"version":3,"file":"useShellHistory.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useShellHistory.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAE/D,MAAM,kBAAkB,GAAG,GAAG,CAAC;AAU/B,KAAK,UAAU,kBAAkB,CAC/B,WAAmB,EACnB,aAAuB;IAEvB,MAAM,OAAO,GAAG,aAAa,IAAI,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC;IAC1D,OAAO,OAAO,CAAC,kBAAkB,EAAE,CAAC;AACtC,CAAC;AAED,4BAA4B;AAC5B,KAAK,UAAU,eAAe,CAAC,QAAgB;IAC7C,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,IAAI,GAAG,GAAG,EAAE,CAAC;QAEb,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE;gBAAE,SAAS;YAC1B,MAAM,IAAI,GAAG,GAAG,CAAC;YAEjB,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,6BAA6B;gBAC7B,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC;YACtC,CAAC;iBAAM,CAAC;gBACN,IAAI,GAAG;oBAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC1B,GAAG,GAAG,IAAI,CAAC;YACb,CAAC;QACH,CAAC;QAED,IAAI,GAAG;YAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1B,OAAO,MAAM,CAAC;IAChB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,WAAW,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;YAAE,OAAO,EAAE,CAAC;QACzD,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,GAAG,CAAC,CAAC;QAC7C,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,gBAAgB,CAC7B,QAAgB,EAChB,OAAiB;IAEjB,IAAI,CAAC;QACH,MAAM,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;IACvD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,WAAmB,EACnB,OAAiB;IAEjB,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,QAAQ,CAAW,EAAE,CAAC,CAAC;IACrD,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IAE5E,SAAS,CAAC,GAAG,EAAE;QACb,KAAK,UAAU,WAAW;YACxB,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YAChE,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAC7B,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;YACtD,UAAU,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,eAAe;QACtD,CAAC;QACD,WAAW,EAAE,CAAC;IAChB,CAAC,EAAE,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;IAE3B,MAAM,mBAAmB,GAAG,WAAW,CACrC,CAAC,OAAe,EAAE,EAAE;QAClB,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,EAAE,CAAC;YACxC,OAAO;QACT,CAAC;QACD,MAAM,UAAU,GAAG,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,OAAO,CAAC,CAAC;aAClE,KAAK,CAAC,CAAC,EAAE,kBAAkB,CAAC;aAC5B,MAAM,CAAC,OAAO,CAAC,CAAC;QACnB,UAAU,CAAC,UAAU,CAAC,CAAC;QACvB,gDAAgD;QAChD,gBAAgB,CAAC,eAAe,EAAE,CAAC,GAAG,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7D,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC,EACD,CAAC,OAAO,EAAE,eAAe,CAAC,CAC3B,CAAC;IAEF,MAAM,kBAAkB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC1C,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,GAAG,CAAC,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAChE,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC1B,OAAO,OAAO,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC;IACnC,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC;IAE5B,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;QACtC,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,QAAQ,GAAG,YAAY,GAAG,CAAC,CAAC;QAClC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC1B,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YACjB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,OAAO,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC;IACnC,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC;IAE5B,MAAM,oBAAoB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC5C,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACL,OAAO;QACP,mBAAmB;QACnB,kBAAkB;QAClB,cAAc;QACd,oBAAoB;KACrB,CAAC;AACJ,CAAC"}
|
|
@@ -9,9 +9,34 @@ import * as fs from 'fs/promises';
|
|
|
9
9
|
import * as path from 'path';
|
|
10
10
|
import * as os from 'os';
|
|
11
11
|
import * as crypto from 'crypto';
|
|
12
|
-
vi.mock('fs/promises')
|
|
12
|
+
vi.mock('fs/promises', () => ({
|
|
13
|
+
readFile: vi.fn(),
|
|
14
|
+
writeFile: vi.fn(),
|
|
15
|
+
mkdir: vi.fn(),
|
|
16
|
+
}));
|
|
13
17
|
vi.mock('os');
|
|
14
18
|
vi.mock('crypto');
|
|
19
|
+
vi.mock('fs', async (importOriginal) => {
|
|
20
|
+
const actualFs = await importOriginal();
|
|
21
|
+
return {
|
|
22
|
+
...actualFs,
|
|
23
|
+
mkdirSync: vi.fn(),
|
|
24
|
+
};
|
|
25
|
+
});
|
|
26
|
+
vi.mock('@google/gemini-cli-core', () => {
|
|
27
|
+
class Storage {
|
|
28
|
+
getProjectTempDir() {
|
|
29
|
+
return path.join('/test/home/', '.gemini', 'tmp', 'mocked_hash');
|
|
30
|
+
}
|
|
31
|
+
getHistoryFilePath() {
|
|
32
|
+
return path.join('/test/home/', '.gemini', 'tmp', 'mocked_hash', 'shell_history');
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
return {
|
|
36
|
+
isNodeError: (err) => typeof err === 'object' && err !== null && 'code' in err,
|
|
37
|
+
Storage,
|
|
38
|
+
};
|
|
39
|
+
});
|
|
15
40
|
const MOCKED_PROJECT_ROOT = '/test/project';
|
|
16
41
|
const MOCKED_HOME_DIR = '/test/home';
|
|
17
42
|
const MOCKED_PROJECT_HASH = 'mocked_hash';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useShellHistory.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useShellHistory.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AAEjC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"useShellHistory.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/useShellHistory.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AAEjC,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;IAClB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;CACf,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAClB,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IACrC,MAAM,QAAQ,GAAG,MAAM,cAAc,EAAuB,CAAC;IAC7D,OAAO;QACL,GAAG,QAAQ;QACX,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;KACnB,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACtC,MAAM,OAAO;QACX,iBAAiB;YACf,OAAO,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,CAAC,CAAC;QACnE,CAAC;QACD,kBAAkB;YAChB,OAAO,IAAI,CAAC,IAAI,CACd,aAAa,EACb,SAAS,EACT,KAAK,EACL,aAAa,EACb,eAAe,CAChB,CAAC;QACJ,CAAC;KACF;IACD,OAAO;QACL,WAAW,EAAE,CAAC,GAAY,EAAgC,EAAE,CAC1D,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,IAAI,MAAM,IAAI,GAAG;QAC1D,OAAO;KACR,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAG,eAAe,CAAC;AAC5C,MAAM,eAAe,GAAG,YAAY,CAAC;AACrC,MAAM,mBAAmB,GAAG,aAAa,CAAC;AAE1C,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAClC,eAAe,EACf,SAAS,EACT,KAAK,EACL,mBAAmB,CACpB,CAAC;AACF,MAAM,mBAAmB,GAAG,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,eAAe,CAAC,CAAC;AAE3E,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC/B,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC/B,MAAM,YAAY,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACxC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAChD,QAAQ,CAAC,KAAK,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC5C,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QAElD,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAChC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC;SACrD,CAAC;QACF,YAAY,CAAC,UAAU,CAAC,eAAe,CAAC,QAAiB,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAE1E,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAC5C,mBAAmB,EACnB,OAAO,CACR,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,GAAkB,IAAI,CAAC;QAClC,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,mDAAmD;QACnD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAA0B,CAAC;QACnE,KAAK,CAAC,IAAI,GAAG,QAAQ,CAAC;QACtB,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAE1E,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,GAAkB,IAAI,CAAC;QAClC,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAE1E,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,kBAAkB,EAAE;gBAC9D,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YACH,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAC7C,mBAAmB,EACnB,aAAa,CACd,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,OAAO,GAAkB,IAAI,CAAC;QAClC,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;QACxD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAE1E,0DAA0D;QAC1D,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElE,IAAI,OAAO,GAAkB,IAAI,CAAC;QAElC,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7B,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7B,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7B,oCAAoC;QACpC,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7B,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAC5C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7B,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAC5C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE7B,wEAAwE;QACxE,GAAG,CAAC,GAAG,EAAE;YACP,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAC5C,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAC1E,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC1E,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAE5D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAC1E,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,qEAAqE;QACrE,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAEnE,wCAAwC;QACxC,mDAAmD;QACnD,oFAAoF;QACpF,4EAA4E;QAC5E,MAAM,cAAc,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CAAC;QACrE,MAAM,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhD,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,qBAAqB;QACjE,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;QACxD,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAE1E,0CAA0C;QAC1C,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElE,GAAG,CAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,mDAAmD;QACnD,uDAAuD;QACvD,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAEnE,MAAM,cAAc,GAAG,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAW,CAAC;QACrE,MAAM,YAAY,GAAG,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEhD,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -26,6 +26,11 @@ const mockConfig = {
|
|
|
26
26
|
getApprovalMode: vi.fn(() => ApprovalMode.DEFAULT),
|
|
27
27
|
getUsageStatisticsEnabled: () => true,
|
|
28
28
|
getDebugMode: () => false,
|
|
29
|
+
getSessionId: () => 'test-session-id',
|
|
30
|
+
getContentGeneratorConfig: () => ({
|
|
31
|
+
model: 'test-model',
|
|
32
|
+
authType: 'oauth-personal',
|
|
33
|
+
}),
|
|
29
34
|
};
|
|
30
35
|
class MockToolInvocation extends BaseToolInvocation {
|
|
31
36
|
tool;
|
|
@@ -95,7 +100,6 @@ describe('useReactToolScheduler in YOLO Mode', () => {
|
|
|
95
100
|
mockToolRequiresConfirmation.execute.mockResolvedValue({
|
|
96
101
|
llmContent: expectedOutput,
|
|
97
102
|
returnDisplay: 'YOLO Formatted tool output',
|
|
98
|
-
summary: 'YOLO summary',
|
|
99
103
|
});
|
|
100
104
|
const { result } = renderSchedulerInYoloMode();
|
|
101
105
|
const schedule = result.current[1];
|
|
@@ -211,7 +215,6 @@ describe('useReactToolScheduler', () => {
|
|
|
211
215
|
mockTool.execute.mockResolvedValue({
|
|
212
216
|
llmContent: 'Tool output',
|
|
213
217
|
returnDisplay: 'Formatted tool output',
|
|
214
|
-
summary: 'Formatted summary',
|
|
215
218
|
});
|
|
216
219
|
mockTool.shouldConfirmExecute.mockResolvedValue(null);
|
|
217
220
|
const { result } = renderScheduler();
|
|
@@ -355,7 +358,6 @@ describe('useReactToolScheduler', () => {
|
|
|
355
358
|
mockToolRequiresConfirmation.execute.mockResolvedValue({
|
|
356
359
|
llmContent: expectedOutput,
|
|
357
360
|
returnDisplay: 'Confirmed display',
|
|
358
|
-
summary: 'Confirmed summary',
|
|
359
361
|
});
|
|
360
362
|
const { result } = renderScheduler();
|
|
361
363
|
const schedule = result.current[1];
|
|
@@ -491,7 +493,6 @@ describe('useReactToolScheduler', () => {
|
|
|
491
493
|
resolveExecutePromise({
|
|
492
494
|
llmContent: 'Final output',
|
|
493
495
|
returnDisplay: 'Final display',
|
|
494
|
-
summary: 'Final summary',
|
|
495
496
|
});
|
|
496
497
|
});
|
|
497
498
|
await act(async () => {
|
|
@@ -523,14 +524,12 @@ describe('useReactToolScheduler', () => {
|
|
|
523
524
|
tool1.execute.mockResolvedValue({
|
|
524
525
|
llmContent: 'Output 1',
|
|
525
526
|
returnDisplay: 'Display 1',
|
|
526
|
-
summary: 'Summary 1',
|
|
527
527
|
});
|
|
528
528
|
tool1.shouldConfirmExecute.mockResolvedValue(null);
|
|
529
529
|
const tool2 = new MockTool('tool2', 'Tool 2');
|
|
530
530
|
tool2.execute.mockResolvedValue({
|
|
531
531
|
llmContent: 'Output 2',
|
|
532
532
|
returnDisplay: 'Display 2',
|
|
533
|
-
summary: 'Summary 2',
|
|
534
533
|
});
|
|
535
534
|
tool2.shouldConfirmExecute.mockResolvedValue(null);
|
|
536
535
|
mockToolRegistry.getTool.mockImplementation((name) => {
|
|
@@ -601,7 +600,6 @@ describe('useReactToolScheduler', () => {
|
|
|
601
600
|
const longExecutePromise = new Promise((resolve) => setTimeout(() => resolve({
|
|
602
601
|
llmContent: 'done',
|
|
603
602
|
returnDisplay: 'done display',
|
|
604
|
-
summary: 'done summary',
|
|
605
603
|
}), 50));
|
|
606
604
|
mockTool.execute.mockReturnValue(longExecutePromise);
|
|
607
605
|
mockTool.shouldConfirmExecute.mockResolvedValue(null);
|