cafb6d4147
Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@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: Quentin <quentin.torroba@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: Mistral Vibe <vibe@mistral.ai>
114 lines
3.7 KiB
Python
114 lines
3.7 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from vibe.core.config.layer import LayerImplementationError
|
|
from vibe.core.config.layers.user import UserConfigLayer
|
|
from vibe.core.config.patch import ConfigPatch
|
|
from vibe.core.config.types import MISSING_CONFIG_FILE_FINGERPRINT
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reads_toml_file(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "config.toml"
|
|
path.write_text('active_model = "mistral-large"\ncount = 42\n')
|
|
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
data = await layer.load()
|
|
assert data.model_extra == {"active_model": "mistral-large", "count": 42}
|
|
fingerprint = layer.fingerprint
|
|
assert isinstance(fingerprint, str)
|
|
assert fingerprint
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_always_trusted(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "config.toml"
|
|
path.write_text('key = "value"\n')
|
|
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
assert layer.is_trusted is None
|
|
data = await layer.load()
|
|
assert layer.is_trusted is True
|
|
assert data.model_extra == {"key": "value"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_missing_file_returns_empty(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "nonexistent.toml"
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
data = await layer.load()
|
|
assert data.model_extra == {}
|
|
assert layer.fingerprint == MISSING_CONFIG_FILE_FINGERPRINT
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_apply_raises_not_implemented(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "config.toml"
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
with pytest.raises(NotImplementedError, match="M2"):
|
|
await layer.apply(ConfigPatch(fingerprint="fp-1"))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_nested_toml_structure(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "config.toml"
|
|
path.write_text("""\
|
|
[models]
|
|
active_model = "test"
|
|
|
|
[[models.items]]
|
|
alias = "a"
|
|
provider = "p"
|
|
""")
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
data = await layer.load()
|
|
assert data.model_extra == {
|
|
"models": {"active_model": "test", "items": [{"alias": "a", "provider": "p"}]}
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_invalid_toml_raises(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "bad.toml"
|
|
path.write_text("this is not valid = = = toml [[[")
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
with pytest.raises(LayerImplementationError, match="_build_config_snapshot"):
|
|
await layer.load()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_force_reload_reads_fresh_data(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "config.toml"
|
|
path.write_text('value = "first"\n')
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
|
|
data1 = await layer.load()
|
|
fp1 = layer.fingerprint
|
|
assert data1.model_extra == {"value": "first"}
|
|
assert isinstance(fp1, str)
|
|
assert fp1
|
|
|
|
path.write_text('value = "second"\n')
|
|
data2 = await layer.load(force=True)
|
|
fp2 = layer.fingerprint
|
|
assert data2.model_extra == {"value": "second"}
|
|
assert isinstance(fp2, str)
|
|
assert fp2
|
|
assert fp1 != fp2
|
|
|
|
path.unlink()
|
|
data3 = await layer.load(force=True)
|
|
assert data3.model_extra == {}
|
|
assert layer.fingerprint == MISSING_CONFIG_FILE_FINGERPRINT
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_empty_toml_file(tmp_working_directory: Path) -> None:
|
|
path = tmp_working_directory / "empty.toml"
|
|
path.write_text("")
|
|
layer = UserConfigLayer(path=path, name="user-toml")
|
|
data = await layer.load()
|
|
assert data.model_extra == {}
|