Initial commit

Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai>
Co-Authored-By: Laure Hugo <laure.hugo@mistral.ai>
Co-Authored-By: Benjamin Trom <benjamin.trom@mistral.ai>
Co-Authored-By: Mathias Gesbert <mathias.gesbert@ext.mistral.ai>
Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai>
Co-Authored-By: Clément Drouin <clement.drouin@mistral.ai>
Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai>
Co-Authored-By: Valentin Berard <val@mistral.ai>
Co-Authored-By: Mistral Vibe <vibe@mistral.ai>

tests/snapshots/base_snapshot_test_app.py

@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+from rich.style import Style
+from textual.widgets.text_area import TextAreaTheme
+
+from tests.stubs.fake_backend import FakeBackend
+from vibe.cli.textual_ui.app import VibeApp
+from vibe.cli.textual_ui.widgets.chat_input import ChatTextArea
+from vibe.core.agent import Agent
+from vibe.core.config import SessionLoggingConfig, VibeConfig
+
+
+def default_config() -> VibeConfig:
+    """Default configuration for snapshot testing.
+    Remove as much interference as possible from the snapshot comparison, in order to get a clean pixel-to-pixel comparison.
+    - Injects a fake backend to prevent (or stub) LLM calls.
+    - Disables the welcome banner animation.
+    - Forces a value for the displayed workdir
+    - Hides the chat input cursor (as the blinking animation is not deterministic).
+    """
+    return VibeConfig(
+        session_logging=SessionLoggingConfig(enabled=False),
+        textual_theme="gruvbox",
+        disable_welcome_banner_animation=True,
+        displayed_workdir="/test/workdir",
+    )
+
+
+class BaseSnapshotTestApp(VibeApp):
+    CSS_PATH = "../../vibe/cli/textual_ui/app.tcss"
+
+    def __init__(self, config: VibeConfig | None = None, **kwargs):
+        config = config or default_config()
+
+        super().__init__(config=config, **kwargs)
+
+        self.agent = Agent(
+            config,
+            auto_approve=self.auto_approve,
+            enable_streaming=self.enable_streaming,
+            backend=FakeBackend(),
+        )
+
+    async def on_mount(self) -> None:
+        await super().on_mount()
+        self._hide_chat_input_cursor()
+
+    def _hide_chat_input_cursor(self) -> None:
+        text_area = self.query_one(ChatTextArea)
+        hidden_cursor_theme = TextAreaTheme(name="hidden_cursor", cursor_style=Style())
+        text_area.register_theme(hidden_cursor_theme)
+        text_area.theme = "hidden_cursor"