mirror of
https://github.com/ryan4yin/nix-config.git
synced 2026-09-10 03:42:06 +02:00
47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
import importlib.util
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
|
|
spec = importlib.util.spec_from_file_location(
|
|
"install_rules", Path(__file__).with_name("install-rules.py")
|
|
)
|
|
installer = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(installer)
|
|
|
|
|
|
class InstallRulesTests(unittest.TestCase):
|
|
def test_existing_file_is_backed_up_without_overwriting_backup(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
source = root / "source.md"
|
|
source.write_text("global rules")
|
|
target = root / "AGENTS.md"
|
|
target.write_text("personal rules")
|
|
backup = root / "AGENTS.md.bak"
|
|
backup.write_text("older rules")
|
|
installer.install_one(root, source, target.name)
|
|
self.assertEqual(target.resolve(), source)
|
|
self.assertEqual(backup.read_text(), "older rules")
|
|
self.assertEqual((root / "AGENTS.md.bak.1").read_text(), "personal rules")
|
|
|
|
def test_failed_link_creation_preserves_existing_file(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
source = root / "source.md"
|
|
source.write_text("global rules")
|
|
target = root / "AGENTS.md"
|
|
target.write_text("personal rules")
|
|
with patch.object(Path, "symlink_to", side_effect=OSError("link failed")):
|
|
with self.assertRaises(OSError):
|
|
installer.install_one(root, source, target.name)
|
|
self.assertTrue(target.is_file())
|
|
self.assertFalse(target.is_symlink())
|
|
self.assertEqual(target.read_text(), "personal rules")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|