v2.9.4 (#669)

Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Jean-Baptiste Muscat <jeanbaptiste.muscatdupuis@mistral.ai>
Co-authored-by: JeroenvdV <JeroenvdV@users.noreply.github.com>
Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com>
Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai>
Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai>
Co-authored-by: Val <102326092+vdeva@users.noreply.github.com>
Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com>
Co-authored-by: allansimon-mistral <allan.simon@ext.mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>

tests/e2e/test_cli_tui_fresh_install.py

@@ -0,0 +1,110 @@
+from __future__ import annotations
+
+import io
+import os
+from pathlib import Path
+import subprocess
+import sys
+
+import pexpect
+import pytest
+
+from tests import TESTS_ROOT
+from tests.e2e.common import (
+    ansi_tolerant_pattern,
+    send_ctrl_c_until_quit_confirmation,
+    wait_for_main_screen,
+    wait_for_request_count,
+)
+from tests.e2e.mock_server import StreamingMockServer
+
+
+def _venv_executable(venv_path: Path, name: str) -> Path:
+    if os.name == "nt":
+        return venv_path / "Scripts" / f"{name}.exe"
+    return venv_path / "bin" / name
+
+
+def _build_wheel(dist_dir: Path) -> Path:
+    subprocess.run(
+        ["uv", "build", "--wheel", "--out-dir", str(dist_dir)],
+        cwd=TESTS_ROOT.parent,
+        check=True,
+    )
+    wheels = sorted(dist_dir.glob("mistral_vibe-*.whl"))
+    assert len(wheels) == 1
+    return wheels[0]
+
+
+def _install_fresh_wheel(tmp_path: Path, wheel_path: Path) -> Path:
+    venv_path = tmp_path / "fresh-install-venv"
+    subprocess.run(
+        ["uv", "venv", "--no-config", "--python", sys.executable, str(venv_path)],
+        cwd=tmp_path,
+        check=True,
+    )
+
+    python_path = _venv_executable(venv_path, "python")
+    subprocess.run(
+        [
+            "uv",
+            "pip",
+            "install",
+            "--no-config",
+            "--refresh",
+            "--python",
+            str(python_path),
+            str(wheel_path),
+        ],
+        cwd=tmp_path,
+        check=True,
+    )
+    return _venv_executable(venv_path, "vibe")
+
+
+@pytest.mark.timeout(90)
+def test_fresh_wheel_install_can_spawn_cli_and_complete_happy_path(
+    streaming_mock_server: StreamingMockServer,
+    setup_e2e_env: None,
+    e2e_workdir: Path,
+    tmp_path: Path,
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    wheel_path = _build_wheel(tmp_path / "dist")
+    vibe_executable = _install_fresh_wheel(tmp_path, wheel_path)
+
+    monkeypatch.delenv("PYTHONPATH", raising=False)
+
+    captured = io.StringIO()
+    child = pexpect.spawn(
+        str(vibe_executable),
+        ["--workdir", str(e2e_workdir)],
+        cwd=str(tmp_path),
+        env=os.environ,
+        encoding="utf-8",
+        timeout=30,
+        dimensions=(36, 120),
+    )
+    child.logfile_read = captured
+
+    try:
+        wait_for_main_screen(child, timeout=20)
+        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)
+
+        send_ctrl_c_until_quit_confirmation(child, captured, timeout=5)
+        child.expect(pexpect.EOF, timeout=10)
+    finally:
+        if child.isalive():
+            child.terminate(force=True)
+        if not child.closed:
+            child.close()
+
+    output = captured.getvalue()
+    assert "Welcome to Mistral Vibe" not in output
+    assert streaming_mock_server.requests[-1].get("model") == "mock-model"