ed5b7192e6
Co-authored-by: Charles-Edouard Cady <charles.edouard.cady@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Sylvain Afchain <safchain@gmail.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from tests.conftest import build_test_vibe_app
|
|
from vibe.cli.textual_ui.widgets.chat_input.container import ChatInputContainer
|
|
from vibe.cli.textual_ui.widgets.messages import SlashCommandMessage, UserMessage
|
|
|
|
|
|
@pytest.mark.parametrize("alias", ["/exit", "exit", "quit", ":q", ":quit"])
|
|
@pytest.mark.asyncio
|
|
async def test_exit_synonym_runs_exit_handler_and_is_not_sent_as_prompt(
|
|
alias: str, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
vibe_app = build_test_vibe_app()
|
|
async with vibe_app.run_test() as pilot:
|
|
await pilot.pause(0.1)
|
|
|
|
calls: list[str] = []
|
|
|
|
async def _record_exit(**_kwargs: Any) -> None:
|
|
calls.append(alias)
|
|
|
|
monkeypatch.setattr(vibe_app, "_exit_app", _record_exit)
|
|
|
|
chat_input = vibe_app.query_one(ChatInputContainer)
|
|
chat_input.post_message(ChatInputContainer.Submitted(alias))
|
|
await pilot.pause(0.2)
|
|
|
|
assert calls == [alias]
|
|
prompts = [
|
|
m
|
|
for m in vibe_app.query(UserMessage)
|
|
if not isinstance(m, SlashCommandMessage)
|
|
]
|
|
assert prompts == []
|