v2.6.0 (#524)

Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai>
Co-authored-by: Gauthier Guinet <43207538+Gguinet@users.noreply.github.com>
Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Quentin <torroba.q@gmail.com>
Co-authored-by: Simon <80467011+sorgfresser@users.noreply.github.com>
Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai>
Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com>
Co-authored-by: angelapopopo <angele.lenglemetz@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>

tests/snapshots/test_ui_snapshot_model_picker.py

@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+from unittest.mock import patch
+
+from textual.pilot import Pilot
+
+from tests.conftest import build_test_vibe_config
+from tests.snapshots.base_snapshot_test_app import BaseSnapshotTestApp
+from tests.snapshots.snap_compare import SnapCompare
+from vibe.core.config._settings import ModelConfig
+
+
+def _model_picker_config():
+    models = [
+        ModelConfig(
+            name="mistral-large-latest", provider="mistral", alias="mistral-large"
+        ),
+        ModelConfig(name="devstral-latest", provider="mistral", alias="devstral"),
+        ModelConfig(name="codestral-latest", provider="mistral", alias="codestral"),
+        ModelConfig(
+            name="mistral-small-latest", provider="mistral", alias="mistral-small"
+        ),
+        ModelConfig(name="devstral", provider="llamacpp", alias="local"),
+    ]
+    return build_test_vibe_config(
+        models=models,
+        active_model="devstral",
+        disable_welcome_banner_animation=True,
+        displayed_workdir="/test/workdir",
+    )
+
+
+class ModelPickerTestApp(BaseSnapshotTestApp):
+    def __init__(self):
+        super().__init__(config=_model_picker_config())
+
+    async def on_mount(self) -> None:
+        await super().on_mount()
+        await self._switch_to_model_picker_app()
+
+
+def test_snapshot_model_picker_initial(snap_compare: SnapCompare) -> None:
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.2)
+
+    assert snap_compare(
+        "test_ui_snapshot_model_picker.py:ModelPickerTestApp",
+        terminal_size=(100, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_model_picker_navigate_down(snap_compare: SnapCompare) -> None:
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.2)
+        await pilot.press("down")
+        await pilot.pause(0.1)
+
+    assert snap_compare(
+        "test_ui_snapshot_model_picker.py:ModelPickerTestApp",
+        terminal_size=(100, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_model_picker_select_different_model(
+    snap_compare: SnapCompare,
+) -> None:
+    """Select the second model and verify the picker closes back to input."""
+
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.2)
+        await pilot.press("down")
+        await pilot.press("enter")
+        await pilot.pause(0.2)
+
+    with patch("vibe.cli.textual_ui.app.VibeConfig.save_updates"):
+        assert snap_compare(
+            "test_ui_snapshot_model_picker.py:ModelPickerTestApp",
+            terminal_size=(100, 36),
+            run_before=run_before,
+        )
+
+
+def test_snapshot_model_picker_escape_cancels(snap_compare: SnapCompare) -> None:
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.pause(0.2)
+        await pilot.press("escape")
+        await pilot.pause(0.2)
+
+    assert snap_compare(
+        "test_ui_snapshot_model_picker.py:ModelPickerTestApp",
+        terminal_size=(100, 36),
+        run_before=run_before,
+    )