2.0.0

Co-Authored-By: Quentin Torroba <quentin.torroba@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: Clément Siriex <clement.sirieix@mistral.ai>
Co-Authored-By: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai>
Co-Authored-By: Thaddee Tyl <thaddee.tyl@gmail.com>
Co-Authored-By: David Brochart <david.brochart@gmail.com>
Co-Authored-By: Joseph Guhlin <joseph.guhlin@gmail.com>
Co-Authored-By: Thomas Kenbeek <thomaskenbeek@gmail.com>
Co-Authored-By: Remenby31 <baptiste.cruvellier31@gmail.com>

tests/cli/test_ui_clipboard_notifications.py

@@ -0,0 +1,51 @@
+from __future__ import annotations
+
+from unittest.mock import MagicMock, patch
+
+import pytest
+from textual.selection import Selection
+from textual.widget import Widget
+
+from vibe.cli.clipboard import copy_selection_to_clipboard
+from vibe.cli.textual_ui.app import VibeApp
+from vibe.core.agent_loop import AgentLoop
+from vibe.core.config import SessionLoggingConfig, VibeConfig
+
+
+class ClipboardSelectionWidget(Widget):
+    def __init__(self, selected_text: str) -> None:
+        super().__init__()
+        self._selected_text = selected_text
+
+    @property
+    def text_selection(self) -> Selection | None:
+        return Selection(None, None)
+
+    def get_selection(self, selection: Selection) -> tuple[str, str] | None:
+        return (self._selected_text, "\n")
+
+
+@pytest.mark.asyncio
+async def test_ui_clipboard_notification_does_not_crash_on_markup_text(
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    agent_loop = AgentLoop(
+        config=VibeConfig(
+            session_logging=SessionLoggingConfig(enabled=False),
+            enable_update_checks=False,
+        )
+    )
+    app = VibeApp(agent_loop=agent_loop)
+
+    async with app.run_test(notifications=True) as pilot:
+        await app.mount(ClipboardSelectionWidget("[/]"))
+        with patch("vibe.cli.clipboard._get_copy_fns") as mock_get_copy_fns:
+            mock_get_copy_fns.return_value = [MagicMock()]
+            copy_selection_to_clipboard(app)
+
+        await pilot.pause(0.1)
+        notifications = list(app._notifications)
+        assert notifications
+        notification = notifications[-1]
+        assert notification.markup is False
+        assert "copied to clipboard" in notification.message