f3006ebef9
* refactor: re-architect tests to mirror the codebase
* Update tests.yml
* fix: add missing tool_error imports after registry refactor
* fix(tests): replace patch.dict with monkeypatch to prevent env var leaks under xdist
patch.dict(os.environ) can leak TERMINAL_ENV across xdist workers,
causing test_code_execution tests to hit the Modal remote path.
* fix(tests): fix update_check and telegram xdist failures
- test_update_check: replace patch("hermes_cli.banner.os.getenv") with
monkeypatch.setenv("HERMES_HOME") — banner.py no longer imports os
directly, it uses get_hermes_home() from hermes_constants.
- test_telegram_conflict/approval_buttons: provide real exception classes
for telegram.error mock (NetworkError, TimedOut, BadRequest) so the
except clause in connect() doesn't fail with "catching classes that do
not inherit from BaseException" when xdist pollutes sys.modules.
* fix(tests): accept unavailable_models kwarg in _prompt_model_selection mock
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
"""Tests for utils.atomic_yaml_write — crash-safe YAML file writes."""
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
import yaml
|
|
|
|
from utils import atomic_yaml_write
|
|
|
|
|
|
class TestAtomicYamlWrite:
|
|
def test_writes_valid_yaml(self, tmp_path):
|
|
target = tmp_path / "data.yaml"
|
|
data = {"key": "value", "nested": {"a": 1}}
|
|
|
|
atomic_yaml_write(target, data)
|
|
|
|
assert yaml.safe_load(target.read_text(encoding="utf-8")) == data
|
|
|
|
def test_cleans_up_temp_file_on_baseexception(self, tmp_path):
|
|
class SimulatedAbort(BaseException):
|
|
pass
|
|
|
|
target = tmp_path / "data.yaml"
|
|
original = {"preserved": True}
|
|
target.write_text(yaml.safe_dump(original), encoding="utf-8")
|
|
|
|
with patch("utils.yaml.dump", side_effect=SimulatedAbort):
|
|
with pytest.raises(SimulatedAbort):
|
|
atomic_yaml_write(target, {"new": True})
|
|
|
|
tmp_files = [f for f in tmp_path.iterdir() if ".tmp" in f.name]
|
|
assert len(tmp_files) == 0
|
|
assert yaml.safe_load(target.read_text(encoding="utf-8")) == original
|
|
|
|
def test_appends_extra_content(self, tmp_path):
|
|
target = tmp_path / "data.yaml"
|
|
|
|
atomic_yaml_write(target, {"key": "value"}, extra_content="\n# comment\n")
|
|
|
|
text = target.read_text(encoding="utf-8")
|
|
assert "key: value" in text
|
|
assert "# comment" in text
|