661588de0c
Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai> Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai> Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai>
99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from pathlib import Path
|
|
import tomllib
|
|
|
|
import pytest
|
|
from textual.events import Resize
|
|
from textual.geometry import Size
|
|
from textual.pilot import Pilot
|
|
from textual.widgets import Input
|
|
|
|
from vibe.core.config_path import GLOBAL_CONFIG_FILE, GLOBAL_ENV_FILE
|
|
from vibe.setup.onboarding import OnboardingApp
|
|
from vibe.setup.onboarding.screens.api_key import ApiKeyScreen
|
|
from vibe.setup.onboarding.screens.theme_selection import THEMES, ThemeSelectionScreen
|
|
|
|
|
|
async def _wait_for(
|
|
condition: Callable[[], bool],
|
|
pilot: Pilot,
|
|
timeout: float = 5.0,
|
|
interval: float = 0.05,
|
|
) -> None:
|
|
elapsed = 0.0
|
|
while not condition():
|
|
await pilot.pause(interval)
|
|
if (elapsed := elapsed + interval) >= timeout:
|
|
msg = "Timed out waiting for condition."
|
|
raise AssertionError(msg)
|
|
|
|
|
|
async def pass_welcome_screen(pilot: Pilot) -> None:
|
|
welcome_screen = pilot.app.get_screen("welcome")
|
|
await _wait_for(
|
|
lambda: not welcome_screen.query_one("#enter-hint").has_class("hidden"), pilot
|
|
)
|
|
await pilot.press("enter")
|
|
await _wait_for(lambda: isinstance(pilot.app.screen, ThemeSelectionScreen), pilot)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ui_gets_through_the_onboarding_successfully() -> None:
|
|
app = OnboardingApp()
|
|
api_key_value = "sk-onboarding-test-key"
|
|
|
|
async with app.run_test() as pilot:
|
|
await pass_welcome_screen(pilot)
|
|
|
|
await pilot.press("enter")
|
|
await _wait_for(lambda: isinstance(app.screen, ApiKeyScreen), pilot)
|
|
api_screen = app.screen
|
|
input_widget = api_screen.query_one("#key", Input)
|
|
await pilot.press(*api_key_value)
|
|
assert input_widget.value == api_key_value
|
|
|
|
await pilot.press("enter")
|
|
await _wait_for(lambda: app.return_value is not None, pilot, timeout=2.0)
|
|
|
|
assert app.return_value == "completed"
|
|
|
|
assert GLOBAL_ENV_FILE.path.is_file()
|
|
env_contents = GLOBAL_ENV_FILE.path.read_text(encoding="utf-8")
|
|
assert "MISTRAL_API_KEY" in env_contents
|
|
assert api_key_value in env_contents
|
|
|
|
assert GLOBAL_CONFIG_FILE.path.is_file()
|
|
config_contents = GLOBAL_CONFIG_FILE.path.read_text(encoding="utf-8")
|
|
config_dict = tomllib.loads(config_contents)
|
|
assert config_dict.get("textual_theme") == app.theme
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ui_can_pick_a_theme_and_saves_selection(config_dir: Path) -> None:
|
|
app = OnboardingApp()
|
|
|
|
async with app.run_test() as pilot:
|
|
await pass_welcome_screen(pilot)
|
|
|
|
theme_screen = app.screen
|
|
app.post_message(
|
|
Resize(Size(40, 10), Size(40, 10))
|
|
) # trigger the resize event handler
|
|
preview = theme_screen.query_one("#preview")
|
|
assert preview.styles.max_height is not None
|
|
target_theme = "gruvbox"
|
|
assert target_theme in THEMES
|
|
start_index = THEMES.index(app.theme)
|
|
target_index = THEMES.index(target_theme)
|
|
steps_down = (target_index - start_index) % len(THEMES)
|
|
await pilot.press(*["down"] * steps_down)
|
|
assert app.theme == target_theme
|
|
await pilot.press("enter")
|
|
await _wait_for(lambda: isinstance(app.screen, ApiKeyScreen), pilot)
|
|
|
|
assert GLOBAL_CONFIG_FILE.path.is_file()
|
|
config_contents = GLOBAL_CONFIG_FILE.path.read_text(encoding="utf-8")
|
|
config_dict = tomllib.loads(config_contents)
|
|
assert config_dict.get("textual_theme") == target_theme
|