Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai>
Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com>
Co-authored-by: Hdandria <henri.dandria@mistral.ai>
Co-authored-by: Ivana Dunisijevic <ivana.dunisijevic@mistral.ai>
Co-authored-by: Jean Burellier <sheplu@users.noreply.github.com>
Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai>
Co-authored-by: Mert Unsal <mert.unsal@mistral.ai>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com>
Co-authored-by: Pierre Rossinès <pierre.rossines@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: renovate-mistral[bot] <253709520+renovate-mistral[bot]@users.noreply.github.com>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Clément Drouin 2026-06-19 11:01:24 +02:00 committed by GitHub
parent 564a14365e
commit 6bedf271ce
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
223 changed files with 10533 additions and 6947 deletions

View file

@ -2,11 +2,19 @@ from __future__ import annotations
from pathlib import Path
import keyring
import pytest
from vibe.core.config import DEFAULT_MISTRAL_API_ENV_KEY, ProviderConfig
from vibe.core.types import Backend
from vibe.setup.auth import AuthState, AuthStateKind, assess_auth_state
@pytest.fixture(autouse=True)
def disable_keyring(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setattr(keyring, "get_password", lambda service, username: None)
def _mistral_provider(
*, api_key_env_var: str = DEFAULT_MISTRAL_API_ENV_KEY
) -> ProviderConfig:
@ -100,7 +108,7 @@ def test_assess_process_env_when_default_key_is_only_in_process_env(
)
def test_assess_vibe_home_env_file_overrides_process_env_when_both_sources_exist(
def test_assess_process_env_when_process_env_and_dotenv_both_exist(
tmp_path: Path,
) -> None:
env_path = tmp_path / ".env"
@ -114,9 +122,9 @@ def test_assess_vibe_home_env_file_overrides_process_env_when_both_sources_exist
)
assert state == AuthState(
kind=AuthStateKind.VIBE_HOME_ENV_FILE_OVERRIDES_PROCESS_ENV,
kind=AuthStateKind.PROCESS_ENV,
can_use_active_provider=True,
sign_out_available=True,
sign_out_available=False,
env_key=DEFAULT_MISTRAL_API_ENV_KEY,
)
@ -214,3 +222,62 @@ def test_assess_empty_dotenv_value_as_signed_out(tmp_path: Path) -> None:
sign_out_available=False,
env_key=DEFAULT_MISTRAL_API_ENV_KEY,
)
def test_assess_os_keyring_when_default_key_is_in_keyring(
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
) -> None:
monkeypatch.setattr(
keyring, "get_password", lambda service, username: "keyring-key"
)
state = assess_auth_state(
_mistral_provider(), env_path=tmp_path / ".env", environ={}
)
assert state == AuthState(
kind=AuthStateKind.OS_KEYRING,
can_use_active_provider=True,
sign_out_available=True,
env_key=DEFAULT_MISTRAL_API_ENV_KEY,
)
def test_assess_vibe_home_env_file_when_dotenv_and_keyring_both_have_value(
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
) -> None:
# resolve_api_key reads the .env-injected os.environ value before the keyring,
# so an overlap must be reported as the .env file, not OS keyring.
monkeypatch.setattr(
keyring, "get_password", lambda service, username: "keyring-key"
)
env_path = tmp_path / ".env"
_write_env_file(env_path, f"{DEFAULT_MISTRAL_API_ENV_KEY}=file-key\n")
state = assess_auth_state(_mistral_provider(), env_path=env_path, environ={})
assert state == AuthState(
kind=AuthStateKind.VIBE_HOME_ENV_FILE,
can_use_active_provider=True,
sign_out_available=True,
env_key=DEFAULT_MISTRAL_API_ENV_KEY,
)
def test_assess_unsupported_provider_when_custom_key_is_in_keyring(
tmp_path: Path, monkeypatch: pytest.MonkeyPatch
) -> None:
monkeypatch.setattr(
keyring, "get_password", lambda service, username: "keyring-key"
)
state = assess_auth_state(
_generic_provider(), env_path=tmp_path / ".env", environ={}
)
assert state == AuthState(
kind=AuthStateKind.UNSUPPORTED_PROVIDER,
can_use_active_provider=True,
sign_out_available=False,
env_key="CUSTOM_API_KEY",
)