mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
e1a25caa52
Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Julien Legrand <72564015+JulienLGRD@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
from typing import override
|
|
|
|
import pytest
|
|
from textual.app import App
|
|
|
|
from vibe.setup import onboarding
|
|
|
|
|
|
class StubApp(App[str | None]):
|
|
def __init__(self, return_value: str | None) -> None:
|
|
super().__init__()
|
|
self._return_value = return_value
|
|
|
|
@override
|
|
def run(self, *args: object, **kwargs: object) -> str | None:
|
|
return self._return_value
|
|
|
|
|
|
def _exit_raiser(code: int = 0) -> None:
|
|
raise SystemExit(code)
|
|
|
|
|
|
def test_exits_on_cancel(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
monkeypatch.setattr(sys, "exit", _exit_raiser)
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
onboarding.run_onboarding(StubApp(None))
|
|
|
|
assert excinfo.value.code == 0
|
|
out = capsys.readouterr().out
|
|
assert "Setup cancelled. See you next time!" in out
|
|
|
|
|
|
def test_warns_on_save_error(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
monkeypatch.setattr(sys, "exit", _exit_raiser)
|
|
|
|
onboarding.run_onboarding(StubApp("save_error:disk full"))
|
|
|
|
out = capsys.readouterr().out
|
|
assert "Could not save API key" in out
|
|
assert "disk full" in out
|
|
|
|
|
|
def test_exits_on_invalid_api_key_env_var(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
monkeypatch.setattr(sys, "exit", _exit_raiser)
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
onboarding.run_onboarding(StubApp("env_var_error:BAD=NAME"))
|
|
|
|
assert excinfo.value.code == 1
|
|
out = capsys.readouterr().out
|
|
assert "Could not save the API key because this provider is configured" in out
|
|
assert "invalid" in out
|
|
assert "environment variable name: BAD=NAME" in out
|
|
assert "was not saved for this session" in out
|
|
assert "set for this session only" not in out
|
|
|
|
|
|
def test_successfully_completes(
|
|
monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]
|
|
) -> None:
|
|
monkeypatch.setattr(sys, "exit", _exit_raiser)
|
|
|
|
onboarding.run_onboarding(StubApp("completed"))
|
|
|
|
out = capsys.readouterr().out
|
|
assert 'Setup complete 🎉. Run "vibe" to start using the Mistral Vibe CLI.' in out
|