fa15fc977b
Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai> Co-Authored-By: Laure Hugo <laure.hugo@mistral.ai> Co-Authored-By: Benjamin Trom <benjamin.trom@mistral.ai> Co-Authored-By: Mathias Gesbert <mathias.gesbert@ext.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: Valentin Berard <val@mistral.ai> Co-Authored-By: Mistral Vibe <vibe@mistral.ai>
124 lines
4.1 KiB
Python
124 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
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 import config as core_config
|
|
from vibe.setup.onboarding import OnboardingApp
|
|
import vibe.setup.onboarding.screens.api_key as api_key_module
|
|
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)
|
|
|
|
|
|
@pytest.fixture()
|
|
def onboarding_app(
|
|
monkeypatch: pytest.MonkeyPatch, tmp_path: Path
|
|
) -> tuple[OnboardingApp, Path, dict[str, Any]]:
|
|
vibe_home = tmp_path / ".vibe"
|
|
env_file = vibe_home / ".env"
|
|
saved_updates: dict[str, Any] = {}
|
|
|
|
def record_updates(updates: dict[str, Any]) -> None:
|
|
saved_updates.update(updates)
|
|
|
|
monkeypatch.setenv("VIBE_HOME", str(vibe_home))
|
|
|
|
for module in (core_config, api_key_module):
|
|
monkeypatch.setattr(module, "GLOBAL_CONFIG_DIR", vibe_home, raising=False)
|
|
monkeypatch.setattr(module, "GLOBAL_ENV_FILE", env_file, raising=False)
|
|
|
|
monkeypatch.setattr(
|
|
core_config.VibeConfig,
|
|
"save_updates",
|
|
classmethod(lambda cls, updates: record_updates(updates)),
|
|
)
|
|
|
|
return OnboardingApp(), env_file, saved_updates
|
|
|
|
|
|
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(
|
|
onboarding_app: tuple[OnboardingApp, Path, dict[str, Any]],
|
|
) -> None:
|
|
app, env_file, config_updates = onboarding_app
|
|
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 env_file.is_file()
|
|
env_contents = env_file.read_text(encoding="utf-8")
|
|
assert "MISTRAL_API_KEY" in env_contents
|
|
assert api_key_value in env_contents
|
|
|
|
assert config_updates.get("textual_theme") == app.theme
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ui_can_pick_a_theme_and_saves_selection(
|
|
onboarding_app: tuple[OnboardingApp, Path, dict[str, Any]],
|
|
) -> None:
|
|
app, _, config_updates = onboarding_app
|
|
|
|
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 config_updates.get("textual_theme") == target_theme
|