d50704e694
Co-authored-by: Cyprien <courtot.c@gmail.com> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Laure Hugo <201583486+laure0303@users.noreply.github.com> Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Peter Evers <peter.evers@mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Yousria <yousria.debaud@mistral.ai> Co-authored-by: renovate-mistral[bot] <253709520+renovate-mistral[bot]@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from tests.conftest import build_test_vibe_app, build_test_vibe_config
|
|
from vibe.cli.textual_ui.app import BottomApp
|
|
from vibe.cli.textual_ui.widgets.theme_picker import ThemePickerApp
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_theme_opens_theme_picker() -> None:
|
|
app = build_test_vibe_app(config=build_test_vibe_config())
|
|
async with app.run_test() as pilot:
|
|
await pilot.pause(0.1)
|
|
await app._show_theme()
|
|
await pilot.pause(0.2)
|
|
|
|
assert app._current_bottom_app == BottomApp.ThemePicker
|
|
assert len(app.query(ThemePickerApp)) == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_theme_picker_lists_themes_and_marks_current() -> None:
|
|
config = build_test_vibe_config()
|
|
config.theme = "dracula"
|
|
app = build_test_vibe_app(config=config)
|
|
async with app.run_test() as pilot:
|
|
await pilot.pause(0.1)
|
|
await app._show_theme()
|
|
await pilot.pause(0.2)
|
|
|
|
picker = app.query_one(ThemePickerApp)
|
|
assert "dracula" in picker._theme_names
|
|
assert "ansi-dark" in picker._theme_names
|
|
assert picker._current_theme == "dracula"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_theme_picker_escape_restores_original_theme() -> None:
|
|
config = build_test_vibe_config()
|
|
config.theme = "ansi-dark"
|
|
app = build_test_vibe_app(config=config)
|
|
async with app.run_test() as pilot:
|
|
await pilot.pause(0.1)
|
|
await app._show_theme()
|
|
await pilot.pause(0.2)
|
|
|
|
# Move highlight to a different theme to trigger preview.
|
|
await pilot.press("down")
|
|
await pilot.pause(0.2)
|
|
|
|
with patch("vibe.cli.textual_ui.app.VibeConfig.save_updates") as mock_save:
|
|
await pilot.press("escape")
|
|
await pilot.pause(0.2)
|
|
|
|
mock_save.assert_not_called()
|
|
|
|
assert app._current_bottom_app == BottomApp.Input
|
|
assert len(app.query(ThemePickerApp)) == 0
|
|
assert app.config.theme == "ansi-dark"
|
|
assert app.theme == "ansi-dark"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_theme_picker_select_persists_and_applies() -> None:
|
|
config = build_test_vibe_config()
|
|
config.theme = "ansi-dark"
|
|
app = build_test_vibe_app(config=config)
|
|
async with app.run_test() as pilot:
|
|
await pilot.pause(0.1)
|
|
await app._show_theme()
|
|
await pilot.pause(0.2)
|
|
|
|
picker = app.query_one(ThemePickerApp)
|
|
names = picker._theme_names
|
|
current_index = names.index("ansi-dark")
|
|
target_index = (current_index + 1) % len(names)
|
|
target = names[target_index]
|
|
|
|
await pilot.press("down")
|
|
|
|
with patch("vibe.cli.textual_ui.app.VibeConfig.save_updates") as mock_save:
|
|
await pilot.press("enter")
|
|
await pilot.pause(0.2)
|
|
|
|
mock_save.assert_called_once_with({"theme": target})
|
|
|
|
assert app._current_bottom_app == BottomApp.Input
|
|
assert len(app.query(ThemePickerApp)) == 0
|
|
assert app.config.theme == target
|
|
assert app.theme == target
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_theme_picker_jk_moves_cursor() -> None:
|
|
from textual.widgets import OptionList
|
|
|
|
config = build_test_vibe_config()
|
|
config.theme = "ansi-dark"
|
|
app = build_test_vibe_app(config=config)
|
|
async with app.run_test() as pilot:
|
|
await pilot.pause(0.1)
|
|
await app._show_theme()
|
|
await pilot.pause(0.2)
|
|
|
|
option_list = app.query_one(ThemePickerApp).query_one(OptionList)
|
|
start = option_list.highlighted
|
|
assert start is not None
|
|
|
|
await pilot.press("j")
|
|
await pilot.pause(0.1)
|
|
assert option_list.highlighted == (start + 1) % option_list.option_count
|
|
|
|
await pilot.press("k")
|
|
await pilot.pause(0.1)
|
|
assert option_list.highlighted == start
|