feat: add Rewind Confirmation dialog and Rewind Viewer component (#15717)

2026-05-03 08:24:10 -07:00 · 2026-01-14 10:22:21 -05:00
parent 3b55581aaf
commit dfb7dc7069
19 changed files with 1318 additions and 27 deletions
@@ -0,0 +1,91 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { describe, it, expect, vi, afterEach } from 'vitest';
+import { act } from 'react';
+import { renderWithProviders } from '../../test-utils/render.js';
+import { waitFor } from '../../test-utils/async.js';
+import { RewindConfirmation, RewindOutcome } from './RewindConfirmation.js';
+
+describe('RewindConfirmation', () => {
+  afterEach(() => {
+    vi.restoreAllMocks();
+  });
+
+  it('renders correctly with stats', () => {
+    const stats = {
+      addedLines: 10,
+      removedLines: 5,
+      fileCount: 1,
+      details: [{ fileName: 'test.ts', diff: '' }],
+    };
+    const onConfirm = vi.fn();
+    const { lastFrame } = renderWithProviders(
+      <RewindConfirmation
+        stats={stats}
+        onConfirm={onConfirm}
+        terminalWidth={80}
+      />,
+      { width: 80 },
+    );
+
+    expect(lastFrame()).toMatchSnapshot();
+    expect(lastFrame()).toContain('Revert code changes');
+  });
+
+  it('renders correctly without stats', () => {
+    const onConfirm = vi.fn();
+    const { lastFrame } = renderWithProviders(
+      <RewindConfirmation
+        stats={null}
+        onConfirm={onConfirm}
+        terminalWidth={80}
+      />,
+      { width: 80 },
+    );
+
+    expect(lastFrame()).toMatchSnapshot();
+    expect(lastFrame()).not.toContain('Revert code changes');
+    expect(lastFrame()).toContain('Rewind conversation');
+  });
+
+  it('calls onConfirm with Cancel on Escape', async () => {
+    const onConfirm = vi.fn();
+    const { stdin } = renderWithProviders(
+      <RewindConfirmation
+        stats={null}
+        onConfirm={onConfirm}
+        terminalWidth={80}
+      />,
+      { width: 80 },
+    );
+
+    await act(async () => {
+      stdin.write('\x1b');
+    });
+
+    await waitFor(() => {
+      expect(onConfirm).toHaveBeenCalledWith(RewindOutcome.Cancel);
+    });
+  });
+
+  it('renders timestamp when provided', () => {
+    const onConfirm = vi.fn();
+    const timestamp = new Date().toISOString();
+    const { lastFrame } = renderWithProviders(
+      <RewindConfirmation
+        stats={null}
+        onConfirm={onConfirm}
+        terminalWidth={80}
+        timestamp={timestamp}
+      />,
+      { width: 80 },
+    );
+
+    expect(lastFrame()).toMatchSnapshot();
+    expect(lastFrame()).not.toContain('Revert code changes');
+  });
+});