v2.7.0 (#534)

Co-authored-by: Quentin Torroba <quentin.torroba@mistral.ai>
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>

tests/snapshots/test_ui_snapshot_rewind.py

@@ -0,0 +1,142 @@
+from __future__ import annotations
+
+from unittest.mock import patch
+
+import pytest
+from textual.pilot import Pilot
+
+from tests.mock.utils import mock_llm_chunk
+from tests.snapshots.base_snapshot_test_app import BaseSnapshotTestApp
+from tests.snapshots.snap_compare import SnapCompare
+from tests.stubs.fake_backend import FakeBackend
+from vibe.core.rewind import RewindError
+
+
+class RewindSnapshotApp(BaseSnapshotTestApp):
+    """Test app with a multi-turn conversation for rewind snapshots."""
+
+    def __init__(self) -> None:
+        fake_backend = FakeBackend([
+            mock_llm_chunk(content="Hello! How can I help you?")
+        ])
+        super().__init__(backend=fake_backend)
+
+
+async def _send_messages(pilot: Pilot) -> None:
+    """Send three messages to build up conversation history.
+
+    Also patches ``has_file_changes_at`` to always return True so the
+    rewind panel shows the "restore files" option.
+    """
+    for msg in ["first message", "second message", "third message"]:
+        await pilot.press(*msg)
+        await pilot.press("enter")
+        await pilot.pause(0.4)
+
+    app: RewindSnapshotApp = pilot.app  # type: ignore[assignment]
+    rm = app.agent_loop.rewind_manager
+    patch.object(rm, "has_file_changes_at", return_value=True).start()
+
+
+def test_snapshot_rewind_panel_shown(snap_compare: SnapCompare) -> None:
+    """Pressing alt+up enters rewind mode and shows the panel."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await _send_messages(pilot)
+        await pilot.press("alt+up")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+
+    assert snap_compare(
+        "test_ui_snapshot_rewind.py:RewindSnapshotApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_rewind_navigate_up(snap_compare: SnapCompare) -> None:
+    """Pressing alt+up twice selects the second-to-last message."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await _send_messages(pilot)
+        await pilot.press("alt+up")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+        await pilot.press("alt+up")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+
+    assert snap_compare(
+        "test_ui_snapshot_rewind.py:RewindSnapshotApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_rewind_navigate_down(snap_compare: SnapCompare) -> None:
+    """Navigate up twice then down once returns to the last message."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await _send_messages(pilot)
+        await pilot.press("alt+up")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+        await pilot.press("alt+up")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+        await pilot.press("alt+down")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+
+    assert snap_compare(
+        "test_ui_snapshot_rewind.py:RewindSnapshotApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_rewind_exit_on_escape(snap_compare: SnapCompare) -> None:
+    """Pressing escape exits rewind mode and restores the input panel."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await _send_messages(pilot)
+        await pilot.press("alt+up")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+        await pilot.press("escape")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+
+    assert snap_compare(
+        "test_ui_snapshot_rewind.py:RewindSnapshotApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_rewind_error_shows_toast(
+    snap_compare: SnapCompare, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    """When rewind_to_message fails, a toast is shown and rewind mode stays active."""
+
+    async def failing_rewind(*_args, **_kwargs):
+        raise RewindError("Invalid message index: 99")
+
+    async def run_before(pilot: Pilot) -> None:
+        await _send_messages(pilot)
+        app: RewindSnapshotApp = pilot.app  # type: ignore[assignment]
+        monkeypatch.setattr(
+            app.agent_loop.rewind_manager, "rewind_to_message", failing_rewind
+        )
+        await pilot.press("alt+up")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.2)
+        await pilot.press("enter")
+        await pilot.app.workers.wait_for_complete()
+        await pilot.pause(0.3)
+
+    assert snap_compare(
+        "test_ui_snapshot_rewind.py:RewindSnapshotApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )