| """Tests for utils.atomic_yaml_write — crash-safe YAML file writes.""" |
|
|
| 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_writes_unicode_unescaped_and_round_trips(self, tmp_path): |
| """Emoji/kaomoji are written as real UTF-8, not fragile escape sequences. |
| |
| Regression for GitHub #51356: without allow_unicode=True, PyYAML emitted |
| astral-plane chars (emoji) as 8-digit `\\UXXXXXXXX` escapes inside |
| multi-line double-quoted strings wrapped with `\\` continuations, which |
| stricter/non-PyYAML parsers and hand-edits broke into unclosed quotes, |
| corrupting the entire config. |
| """ |
| target = tmp_path / "config.yaml" |
| |
| data = { |
| "personalities": { |
| "kawaii": "kawaii desu~! (◕‿◕) ★ ♪ ヽ(>∀<☆)ノ", |
| "catgirl": "nya~! (=^･ω･^=) ฅ^•ﻌ•^ฅ", |
| "surfer": "Cowabunga! 🤙 totally rad bro", |
| "hype": "LET'S GOOOO!!! 🔥 LEGENDARY!", |
| }, |
| "display": {"cursor": " â–‰"}, |
| } |
|
|
| atomic_yaml_write(target, data) |
|
|
| text = target.read_text(encoding="utf-8") |
| |
| assert "\\U" not in text |
| assert "\\u" not in text |
| |
| assert "🔥" in text |
| assert "(=^･ω･^=)" in text |
| |
| assert yaml.safe_load(text) == data |
|
|
