a560a47ce8
Co-authored-by: Quentin Torroba <quentin.torroba@mistral.ai> Co-authored-by: Vincent Guilloux <vincent.guilloux@mistral.ai> Co-authored-by: Thomas Kenbeek <thomas.kenbeek@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import 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
|
|
|
|
|
|
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, vibe_app: VibeApp
|
|
) -> None:
|
|
async with vibe_app.run_test(notifications=True) as pilot:
|
|
await vibe_app.mount(ClipboardSelectionWidget("[/]"))
|
|
with patch("vibe.cli.clipboard._copy_to_clipboard"):
|
|
copy_selection_to_clipboard(vibe_app)
|
|
|
|
await pilot.pause(0.1)
|
|
notifications = list(vibe_app._notifications)
|
|
assert notifications
|
|
notification = notifications[-1]
|
|
assert notification.markup is False
|
|
assert "copied to clipboard" in notification.message
|