228f3c65a9
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Corentin André <corentin.andre@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@users.noreply.github.com> Co-authored-by: Peter Evers <pevers90@gmail.com> 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: MichisGitIsKing <MichisGitIsKing@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from typing import cast
|
|
|
|
from textual.pilot import Pilot
|
|
|
|
from tests.snapshots.base_snapshot_test_app import BaseSnapshotTestApp
|
|
from tests.snapshots.snap_compare import SnapCompare
|
|
from vibe.cli.textual_ui.widgets.messages import PlanFileMessage
|
|
|
|
PLAN_CONTENT = """\
|
|
# Implementation Plan
|
|
|
|
## 1. Add user authentication
|
|
- JWT token validation
|
|
- Session management
|
|
|
|
## 2. Database migrations
|
|
- Create users table
|
|
- Add indexes
|
|
"""
|
|
|
|
|
|
class PlanFileMessageTestApp(BaseSnapshotTestApp):
|
|
pass
|
|
|
|
|
|
def test_snapshot_plan_file_message(snap_compare: SnapCompare, tmp_path: Path) -> None:
|
|
plan_file = tmp_path / "plan.md"
|
|
plan_file.write_text(PLAN_CONTENT)
|
|
|
|
async def run_before(pilot: Pilot) -> None:
|
|
app = cast(PlanFileMessageTestApp, pilot.app)
|
|
plan_widget = PlanFileMessage(file_path=plan_file)
|
|
await app._mount_and_scroll(plan_widget)
|
|
await pilot.pause(0.3)
|
|
|
|
assert snap_compare(
|
|
"test_ui_snapshot_plan_file_message.py:PlanFileMessageTestApp",
|
|
terminal_size=(120, 36),
|
|
run_before=run_before,
|
|
)
|