5d2e01a6d7
Co-authored-by: Carlo <carloantonio.patti@mistral.ai> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Clement Sirieix <clem.sirieix@gmail.com> 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: Thomas Kenbeek <thomas.kenbeek@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pexpect
|
|
import pytest
|
|
|
|
from tests.e2e.common import (
|
|
SpawnedVibeProcessFixture,
|
|
ansi_tolerant_pattern,
|
|
wait_for_main_screen,
|
|
wait_for_request_count,
|
|
)
|
|
from tests.e2e.mock_server import StreamingMockServer
|
|
|
|
|
|
@pytest.mark.timeout(15)
|
|
def test_spawn_cli_to_send_and_receive_message(
|
|
streaming_mock_server: StreamingMockServer,
|
|
setup_e2e_env: None,
|
|
e2e_workdir: Path,
|
|
spawned_vibe_process: SpawnedVibeProcessFixture,
|
|
) -> None:
|
|
with spawned_vibe_process(e2e_workdir) as (child, captured):
|
|
wait_for_main_screen(child, timeout=15)
|
|
child.send("Greet")
|
|
child.send("\r")
|
|
|
|
wait_for_request_count(
|
|
lambda: len(streaming_mock_server.requests), expected_count=1, timeout=10
|
|
)
|
|
child.expect(ansi_tolerant_pattern("Hello from mock server"), timeout=10)
|
|
|
|
child.sendcontrol("c")
|
|
child.expect(pexpect.EOF, timeout=10)
|
|
|
|
output = captured.getvalue()
|
|
assert "Welcome to Mistral Vibe" not in output
|
|
|
|
request_payload = streaming_mock_server.requests[-1]
|
|
assert request_payload.get("stream") is True
|
|
assert request_payload.get("model") == "mock-model"
|
|
stream_options = request_payload.get("stream_options")
|
|
assert stream_options is not None
|
|
assert stream_options.get("include_usage") is True
|
|
messages = request_payload.get("messages")
|
|
assert messages is not None
|
|
assert any(
|
|
message.get("role") == "user" and message.get("content") == "Greet"
|
|
for message in messages
|
|
)
|