backup-docker-to-local/tests/unit/backup/test_layout.py

"""Contract of where a backup run puts its directories."""

from __future__ import annotations

import tempfile
import unittest
from pathlib import Path
from unittest import mock

from baudolo.backup import layout as mod
from baudolo.backup.shell import BackupException


class TestVersionDirectory(unittest.TestCase):
    def test_it_creates_the_generation_directory(self) -> None:
        with tempfile.TemporaryDirectory() as tmp:
            created = mod.create_version_directory(tmp, "20260731020304")
            self.assertTrue(Path(created).is_dir())
            self.assertEqual(Path(created).name, "20260731020304")

    def test_it_refuses_a_generation_another_run_already_claimed(self) -> None:
        with tempfile.TemporaryDirectory() as tmp:
            mod.create_version_directory(tmp, "20260731")
            with self.assertRaises(BackupException) as caught:
                mod.create_version_directory(tmp, "20260731")
            self.assertIn("20260731", str(caught.exception))

    def test_it_creates_missing_parents(self) -> None:
        with tempfile.TemporaryDirectory() as tmp:
            nested = str(Path(tmp) / "machine" / "repo")
            created = mod.create_version_directory(nested, "20260731")
            self.assertTrue(Path(created).is_dir())


class TestVolumeDirectory(unittest.TestCase):
    def test_it_nests_the_volume_under_the_generation(self) -> None:
        with tempfile.TemporaryDirectory() as tmp:
            created = mod.create_volume_directory(tmp, "postgres_data")
            self.assertEqual(Path(created).parent, Path(tmp))
            self.assertTrue(Path(created).is_dir())


class TestMachineId(unittest.TestCase):
    def test_it_takes_the_hash_without_the_filename(self) -> None:
        digest = "a" * 64
        with mock.patch.object(
            mod, "execute_shell_command", return_value=[f"{digest}  /etc/machine-id"]
        ):
            self.assertEqual(mod.get_machine_id(), digest)


if __name__ == "__main__":
    unittest.main()