MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-04-19 09:41:17 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat(cli): Move key restore logic to core (#13013)

Browse Source
This commit is contained in:
Coco Sheng
2025-12-04 10:56:16 -05:00
committed by GitHub
parent 0a2971f9d3
commit b27cf0b0a8
14 changed files with 404 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
packages/cli/src/ui/commands/restoreCommand.test.ts
View File

@@ -155,10 +155,10 @@ describe('restoreCommand', () => {
it('should restore a tool call and project state', async () => {
const toolCallData = {
history: [{ type: 'user', text: 'do a thing' }],
history: [{ type: 'user', text: 'do a thing', id: 123 }],
clientHistory: [{ role: 'user', parts: [{ text: 'do a thing' }] }],
commitHash: 'abcdef123',
toolCall: { name: 'run_shell_command', args: 'ls' },
toolCall: { name: 'run_shell_command', args: { command: 'ls' } },
};
await fs.writeFile(
path.join(checkpointsDir, 'my-checkpoint.json'),
@@ -169,7 +169,7 @@ describe('restoreCommand', () => {
expect(await command?.action?.(mockContext, 'my-checkpoint')).toEqual({
type: 'tool',
toolName: 'run_shell_command',
toolArgs: 'ls',
toolArgs: { command: 'ls' },
});
expect(mockContext.ui.loadHistory).toHaveBeenCalledWith(
toolCallData.history,
@@ -189,7 +189,7 @@ describe('restoreCommand', () => {
it('should restore even if only toolCall is present', async () => {
const toolCallData = {
toolCall: { name: 'run_shell_command', args: 'ls' },
toolCall: { name: 'run_shell_command', args: { command: 'ls' } },
};
await fs.writeFile(
path.join(checkpointsDir, 'my-checkpoint.json'),
@@ -201,7 +201,7 @@ describe('restoreCommand', () => {
expect(await command?.action?.(mockContext, 'my-checkpoint')).toEqual({
type: 'tool',
toolName: 'run_shell_command',
toolArgs: 'ls',
toolArgs: { command: 'ls' },
});
expect(mockContext.ui.loadHistory).not.toHaveBeenCalled();
@@ -222,7 +222,7 @@ describe('restoreCommand', () => {
type: 'message',
messageType: 'error',
// A more specific error message would be ideal, but for now, we can assert the current behavior.
content: expect.stringContaining('Could not read restorable tool calls.'),
content: expect.stringContaining('Checkpoint file is invalid'),
});
});