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>
82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Callable, Iterator
|
|
from contextlib import AbstractContextManager, contextmanager
|
|
import io
|
|
import os
|
|
from pathlib import Path
|
|
from typing import cast
|
|
|
|
import pexpect
|
|
import pytest
|
|
|
|
from tests import TESTS_ROOT
|
|
from tests.e2e.common import write_e2e_config
|
|
from tests.e2e.mock_server import ChunkFactory, StreamingMockServer
|
|
|
|
|
|
@pytest.fixture
|
|
def streaming_mock_server(
|
|
request: pytest.FixtureRequest,
|
|
) -> Iterator[StreamingMockServer]:
|
|
chunk_factory = cast(ChunkFactory | None, getattr(request, "param", None))
|
|
server = StreamingMockServer(chunk_factory=chunk_factory)
|
|
server.start()
|
|
try:
|
|
yield server
|
|
finally:
|
|
server.stop()
|
|
|
|
|
|
@pytest.fixture
|
|
def setup_e2e_env(
|
|
tmp_path: Path,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
streaming_mock_server: StreamingMockServer,
|
|
) -> None:
|
|
vibe_home = tmp_path / "vibe-home"
|
|
write_e2e_config(vibe_home, streaming_mock_server.api_base)
|
|
monkeypatch.setenv("MISTRAL_API_KEY", "fake-key")
|
|
monkeypatch.setenv("VIBE_HOME", str(vibe_home))
|
|
monkeypatch.setenv("TERM", "xterm-256color")
|
|
|
|
|
|
@pytest.fixture
|
|
def e2e_workdir(tmp_path: Path) -> Path:
|
|
workdir = tmp_path / "workdir"
|
|
workdir.mkdir()
|
|
return workdir
|
|
|
|
|
|
type SpawnedVibeContext = Iterator[tuple[pexpect.spawn, io.StringIO]]
|
|
type SpawnedVibeContextManager = AbstractContextManager[
|
|
tuple[pexpect.spawn, io.StringIO]
|
|
]
|
|
type SpawnedVibeFactory = Callable[[Path], SpawnedVibeContextManager]
|
|
|
|
|
|
@pytest.fixture
|
|
def spawned_vibe_process() -> SpawnedVibeFactory:
|
|
@contextmanager
|
|
def spawn(workdir: Path) -> SpawnedVibeContext:
|
|
captured = io.StringIO()
|
|
child = pexpect.spawn(
|
|
"uv",
|
|
["run", "vibe", "--workdir", str(workdir)],
|
|
cwd=str(TESTS_ROOT.parent),
|
|
env=os.environ,
|
|
encoding="utf-8",
|
|
timeout=30,
|
|
dimensions=(36, 120),
|
|
)
|
|
child.logfile_read = captured
|
|
|
|
try:
|
|
yield child, captured
|
|
finally:
|
|
if child.isalive():
|
|
child.terminate(force=True)
|
|
if not child.closed:
|
|
child.close()
|
|
|
|
return spawn
|