v2.3.0 (#429)

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>

tests/e2e/conftest.py

@@ -0,0 +1,82 @@
+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