vrr/qwen-code
2
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2026-08-14 19:24:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 27

test: cover rewind selector fallback states (#4905)

Browse source
This commit is contained in:
han 2026-06-10 23:14:10 +08:00 • committed by GitHub
parent debd87469f
commit f479f7a660
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 84 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

84
packages/cli/src/ui/components/RewindSelector.test.tsx
View file

@ -154,4 +154,88 @@ describe('RewindSelector', () => {
expect(lastFrame()).toContain('› #2 second prompt');
expect(lastFrame()).not.toContain('Restore conversation only');
});
it('falls back to conversation-only options when diff stats fail', async () => {
vi.spyOn(fileHistoryService, 'getDiffStats').mockRejectedValue(
new Error('diff unavailable'),
);
const { lastFrame } = render(
<RewindSelector
history={[
userTurn(1, 'first prompt', 'prompt-1'),
userTurn(2, 'second prompt', 'prompt-2'),
]}
onRewind={vi.fn()}
onCancel={vi.fn()}
fileCheckpointingEnabled={true}
fileHistoryService={fileHistoryService}
/>,
);
pressKey({ name: 'return' });
await flush();
expect(fileHistoryService.getDiffStats).toHaveBeenCalledWith('prompt-2');
expect(lastFrame()).toContain('Restore conversation only');
expect(lastFrame()).toContain('Never mind');
expect(lastFrame()).toContain('File restore is unavailable for this turn');
expect(lastFrame()).not.toContain('Restore code and conversation');
expect(lastFrame()).not.toContain('Restore code only');
});
it('does not request diff stats when the selected turn has no prompt id', async () => {
vi.spyOn(fileHistoryService, 'getDiffStats').mockResolvedValue({
filesChanged: ['src/foo.ts'],
insertions: 2,
deletions: 0,
});
const { lastFrame } = render(
<RewindSelector
history={[userTurn(1, 'first prompt'), userTurn(2, 'second prompt')]}
onRewind={vi.fn()}
onCancel={vi.fn()}
fileCheckpointingEnabled={true}
fileHistoryService={fileHistoryService}
/>,
);
pressKey({ name: 'return' });
await flush();
expect(fileHistoryService.getDiffStats).not.toHaveBeenCalled();
expect(lastFrame()).toContain('Restore conversation only');
expect(lastFrame()).toContain('Never mind');
expect(lastFrame()).toContain('File restore is unavailable for this turn');
expect(lastFrame()).not.toContain('Restore code and conversation');
expect(lastFrame()).not.toContain('Restore code only');
});
it('confirms a conversation-only rewind when checkpointing is disabled', () => {
const onRewind = vi.fn();
const { lastFrame } = render(
<RewindSelector
history={[userTurn(1, 'first prompt'), userTurn(2, 'second prompt')]}
onRewind={onRewind}
onCancel={vi.fn()}
fileCheckpointingEnabled={false}
fileHistoryService={fileHistoryService}
/>,
);
pressKey({ name: 'return' });
expect(lastFrame()).toContain(
'This will remove all conversation after this turn.',
);
pressKey({ sequence: 'y' });
expect(onRewind).toHaveBeenCalledWith(
expect.objectContaining({ id: 2 }),
'conversation',
);
});
});