632ea8c032
Co-authored-by: Antoine <33425718+anth2o@users.noreply.github.com> Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> 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: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@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: Robin Gullo <robin.gullo@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
import tomllib
|
|
|
|
from vibe.cli.cache import read_cache, write_cache
|
|
|
|
|
|
class TestReadCache:
|
|
def test_reads_valid_toml(self, tmp_path: Path) -> None:
|
|
cache_path = tmp_path / "cache.toml"
|
|
cache_path.write_text('[update_cache]\nlatest_version = "1.0.0"\n')
|
|
|
|
result = read_cache(cache_path)
|
|
|
|
assert result["update_cache"]["latest_version"] == "1.0.0"
|
|
|
|
def test_returns_empty_dict_when_missing(self, tmp_path: Path) -> None:
|
|
assert read_cache(tmp_path / "missing.toml") == {}
|
|
|
|
def test_returns_empty_dict_when_corrupted(self, tmp_path: Path) -> None:
|
|
cache_path = tmp_path / "cache.toml"
|
|
cache_path.write_text("{bad toml")
|
|
|
|
assert read_cache(cache_path) == {}
|
|
|
|
|
|
class TestWriteCache:
|
|
def test_writes_new_file(self, tmp_path: Path) -> None:
|
|
cache_path = tmp_path / "cache.toml"
|
|
|
|
write_cache(cache_path, "feedback", {"last_shown_at": 100.0})
|
|
|
|
with cache_path.open("rb") as f:
|
|
data = tomllib.load(f)
|
|
assert data["feedback"]["last_shown_at"] == 100.0
|
|
|
|
def test_merges_with_existing(self, tmp_path: Path) -> None:
|
|
cache_path = tmp_path / "cache.toml"
|
|
cache_path.write_text('[update_cache]\nlatest_version = "1.0.0"\n')
|
|
|
|
write_cache(cache_path, "feedback", {"last_shown_at": 200.0})
|
|
|
|
with cache_path.open("rb") as f:
|
|
data = tomllib.load(f)
|
|
assert data["update_cache"]["latest_version"] == "1.0.0"
|
|
assert data["feedback"]["last_shown_at"] == 200.0
|
|
|
|
def test_merges_within_section_and_leaves_other_sections_alone(
|
|
self, tmp_path: Path
|
|
) -> None:
|
|
cache_path = tmp_path / "cache.toml"
|
|
cache_path.write_text(
|
|
"[update_cache]\n"
|
|
'latest_version = "1.0.0"\n'
|
|
"stored_at_timestamp = 1\n"
|
|
'seen_whats_new_version = "1.0.0"\n\n'
|
|
"[feedback]\n"
|
|
"last_shown_at = 100.0\n"
|
|
)
|
|
|
|
write_cache(
|
|
cache_path,
|
|
"update_cache",
|
|
{"latest_version": "2.0.0", "stored_at_timestamp": 2},
|
|
)
|
|
|
|
with cache_path.open("rb") as f:
|
|
data = tomllib.load(f)
|
|
assert data["update_cache"]["latest_version"] == "2.0.0"
|
|
assert data["update_cache"]["stored_at_timestamp"] == 2
|
|
assert data["update_cache"]["seen_whats_new_version"] == "1.0.0"
|
|
assert data["feedback"]["last_shown_at"] == 100.0
|