Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai> Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai> Co-Authored-By: Kracekumar <kracethekingmaker@gmail.com>
53 lines
2 KiB
Python
53 lines
2 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from vibe.core.paths.config_paths import CONFIG_FILE
|
|
from vibe.core.paths.global_paths import GLOBAL_CONFIG_FILE, VIBE_HOME
|
|
from vibe.core.trusted_folders import trusted_folders_manager
|
|
|
|
|
|
class TestResolveConfigFile:
|
|
def test_resolves_local_config_when_exists_and_folder_is_trusted(
|
|
self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.chdir(tmp_path)
|
|
local_config_dir = tmp_path / ".vibe"
|
|
local_config_dir.mkdir()
|
|
local_config = local_config_dir / "config.toml"
|
|
local_config.write_text('active_model = "test"', encoding="utf-8")
|
|
|
|
monkeypatch.setattr(trusted_folders_manager, "is_trusted", lambda _: True)
|
|
|
|
assert CONFIG_FILE.path == local_config
|
|
assert CONFIG_FILE.path.is_file()
|
|
assert CONFIG_FILE.path.read_text(encoding="utf-8") == 'active_model = "test"'
|
|
|
|
def test_resolves_local_config_when_exists_and_folder_is_not_trusted(
|
|
self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.chdir(tmp_path)
|
|
local_config_dir = tmp_path / ".vibe"
|
|
local_config_dir.mkdir()
|
|
local_config = local_config_dir / "config.toml"
|
|
local_config.write_text('active_model = "test"', encoding="utf-8")
|
|
|
|
assert CONFIG_FILE.path == GLOBAL_CONFIG_FILE.path
|
|
|
|
def test_falls_back_to_global_config_when_local_missing(
|
|
self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.chdir(tmp_path)
|
|
# Ensure no local config exists
|
|
assert not (tmp_path / ".vibe" / "config.toml").exists()
|
|
|
|
assert CONFIG_FILE.path == GLOBAL_CONFIG_FILE.path
|
|
|
|
def test_respects_vibe_home_env_var(
|
|
self, tmp_path: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
assert VIBE_HOME.path != tmp_path
|
|
monkeypatch.setenv("VIBE_HOME", str(tmp_path))
|
|
assert VIBE_HOME.path == tmp_path
|