mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
51fecc67d9
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: Nicolas Karolak <nicolas@karolak.fr>
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
"""Tests for the external editor UI integration (Ctrl+G keybind)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from contextlib import contextmanager
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from vibe.cli.textual_ui.app import VibeApp
|
|
from vibe.cli.textual_ui.widgets.chat_input.container import ChatInputContainer
|
|
|
|
|
|
@contextmanager
|
|
def mock_suspend():
|
|
"""Mock context manager to replace app.suspend()."""
|
|
yield
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ctrl_g_opens_external_editor_and_updates_input(
|
|
vibe_app: VibeApp,
|
|
) -> None:
|
|
"""Test that Ctrl+G triggers external editor and updates input with result."""
|
|
with patch(
|
|
"vibe.cli.textual_ui.widgets.chat_input.text_area.ExternalEditor"
|
|
) as MockEditor:
|
|
mock_instance = MagicMock()
|
|
mock_instance.is_available.return_value = True
|
|
mock_instance.edit.return_value = "edited content"
|
|
MockEditor.return_value = mock_instance
|
|
|
|
async with vibe_app.run_test() as pilot:
|
|
chat_input = vibe_app.query_one(ChatInputContainer)
|
|
chat_input.value = "original"
|
|
chat_input.focus_input()
|
|
await pilot.pause()
|
|
|
|
with patch.object(vibe_app, "suspend", mock_suspend):
|
|
await pilot.press("ctrl+g")
|
|
await pilot.pause()
|
|
|
|
mock_instance.edit.assert_called_once_with("original")
|
|
assert chat_input.value == "edited content"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ctrl_g_works_with_empty_input(vibe_app: VibeApp) -> None:
|
|
"""Test that Ctrl+G works when input is empty."""
|
|
with patch(
|
|
"vibe.cli.textual_ui.widgets.chat_input.text_area.ExternalEditor"
|
|
) as MockEditor:
|
|
mock_instance = MagicMock()
|
|
mock_instance.is_available.return_value = True
|
|
mock_instance.edit.return_value = "new content"
|
|
MockEditor.return_value = mock_instance
|
|
|
|
async with vibe_app.run_test() as pilot:
|
|
chat_input = vibe_app.query_one(ChatInputContainer)
|
|
assert chat_input.value == ""
|
|
chat_input.focus_input()
|
|
await pilot.pause()
|
|
|
|
with patch.object(vibe_app, "suspend", mock_suspend):
|
|
await pilot.press("ctrl+g")
|
|
await pilot.pause()
|
|
|
|
mock_instance.edit.assert_called_once_with("")
|
|
assert chat_input.value == "new content"
|