v1.2.0

Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai>
Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai>
Co-Authored-By: Kracekumar <kracethekingmaker@gmail.com>

tests/snapshots/base_snapshot_test_app.py

@@ -37,7 +37,7 @@ class BaseSnapshotTestApp(VibeApp):
 
         self.agent = Agent(
             config,
-            auto_approve=self.auto_approve,
+            mode=self._current_agent_mode,
             enable_streaming=self.enable_streaming,
             backend=FakeBackend(),
         )

tests/snapshots/test_ui_snapshot_basic_conversation.py

@@ -24,7 +24,7 @@ class SnapshotTestAppWithConversation(BaseSnapshotTestApp):
         super().__init__(config=config)
         self.agent = Agent(
             config,
-            auto_approve=self.auto_approve,
+            mode=self._current_agent_mode,
             enable_streaming=self.enable_streaming,
             backend=fake_backend,
         )

tests/snapshots/test_ui_snapshot_modes.py

@@ -0,0 +1,84 @@
+from __future__ import annotations
+
+from textual.pilot import Pilot
+
+from tests.snapshots.snap_compare import SnapCompare
+
+
+def test_snapshot_default_mode(snap_compare: SnapCompare) -> None:
+    """Test that default mode is displayed correctly at startup."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.1)
+
+    assert snap_compare(
+        "base_snapshot_test_app.py:BaseSnapshotTestApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_cycle_to_plan_mode(snap_compare: SnapCompare) -> None:
+    """Test that shift+tab cycles from default to plan mode."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.1)
+        await pilot.press("shift+tab")  # default -> plan
+        await pilot.pause(0.1)
+
+    assert snap_compare(
+        "base_snapshot_test_app.py:BaseSnapshotTestApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_cycle_to_accept_edits_mode(snap_compare: SnapCompare) -> None:
+    """Test that shift+tab cycles from plan to accept edits mode."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.1)
+        await pilot.press("shift+tab")  # default -> plan
+        await pilot.press("shift+tab")  # plan -> accept edits
+        await pilot.pause(0.1)
+
+    assert snap_compare(
+        "base_snapshot_test_app.py:BaseSnapshotTestApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_cycle_to_auto_approve_mode(snap_compare: SnapCompare) -> None:
+    """Test that shift+tab cycles to auto approve mode."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.1)
+        await pilot.press("shift+tab")  # default -> plan
+        await pilot.press("shift+tab")  # plan -> accept edits
+        await pilot.press("shift+tab")  # accept edits -> auto approve
+        await pilot.pause(0.1)
+
+    assert snap_compare(
+        "base_snapshot_test_app.py:BaseSnapshotTestApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_cycle_wraps_to_default(snap_compare: SnapCompare) -> None:
+    """Test that shift+tab cycles back to default mode after auto approve."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.1)
+        await pilot.press("shift+tab")  # default -> plan
+        await pilot.press("shift+tab")  # plan -> accept edits
+        await pilot.press("shift+tab")  # accept edits -> auto approve
+        await pilot.press("shift+tab")  # auto approve -> default (wrap)
+        await pilot.pause(0.1)
+
+    assert snap_compare(
+        "base_snapshot_test_app.py:BaseSnapshotTestApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )