702d0f412e
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Gauthier Guinet <43207538+Gguinet@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Stanislas <stanislas.lange@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from textual.pilot import Pilot
|
|
|
|
from tests.snapshots.snap_compare import SnapCompare
|
|
from vibe.setup.update_prompt import update_prompt_dialog
|
|
from vibe.setup.update_prompt.update_prompt_dialog import UpdatePromptApp
|
|
|
|
|
|
class UpdatePromptSnapshotApp(UpdatePromptApp):
|
|
CSS_PATH = str(
|
|
Path(update_prompt_dialog.__file__).parent / "update_prompt_dialog.tcss"
|
|
)
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__(current_version="1.0.0", latest_version="2.0.0")
|
|
|
|
|
|
def test_snapshot_update_prompt_initial(snap_compare: SnapCompare) -> None:
|
|
async def run_before(pilot: Pilot) -> None:
|
|
await pilot.pause(0.2)
|
|
|
|
assert snap_compare(
|
|
"test_ui_snapshot_update_prompt.py:UpdatePromptSnapshotApp",
|
|
terminal_size=(80, 20),
|
|
run_before=run_before,
|
|
)
|
|
|
|
|
|
def test_snapshot_update_prompt_continue_selected(snap_compare: SnapCompare) -> None:
|
|
async def run_before(pilot: Pilot) -> None:
|
|
await pilot.pause(0.2)
|
|
await pilot.press("right")
|
|
await pilot.pause(0.1)
|
|
|
|
assert snap_compare(
|
|
"test_ui_snapshot_update_prompt.py:UpdatePromptSnapshotApp",
|
|
terminal_size=(80, 20),
|
|
run_before=run_before,
|
|
)
|