68 lines
2.7 KiB
Python
68 lines
2.7 KiB
Python
|
"""Contract of the rules deciding what is backed up and what must stop."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from unittest import mock
|
||
|
|
|
||
|
|
from baudolo.backup import policy as mod
|
||
|
|
|
||
|
|
|
||
|
|
class TestIsImageIgnored(unittest.TestCase):
|
||
|
|
def test_an_empty_whitelist_ignores_nothing(self) -> None:
|
||
|
|
self.assertFalse(mod.is_image_ignored("c1", []))
|
||
|
|
|
||
|
|
def test_a_listed_image_is_ignored(self) -> None:
|
||
|
|
with mock.patch.object(mod, "get_image_info", return_value="alpine:3.20"):
|
||
|
|
self.assertTrue(mod.is_image_ignored("c1", ["alpine:3.20"]))
|
||
|
|
|
||
|
|
def test_matching_is_exact(self) -> None:
|
||
|
|
with mock.patch.object(mod, "get_image_info", return_value="alpine:3.21"):
|
||
|
|
self.assertFalse(mod.is_image_ignored("c1", ["alpine:3.20"]))
|
||
|
|
|
||
|
|
|
||
|
|
class TestVolumeIsFullyIgnored(unittest.TestCase):
|
||
|
|
def test_a_volume_without_containers_is_kept(self) -> None:
|
||
|
|
self.assertFalse(mod.volume_is_fully_ignored([], ["alpine:3.20"]))
|
||
|
|
|
||
|
|
def test_it_needs_every_container_to_be_ignored(self) -> None:
|
||
|
|
with mock.patch.object(mod, "get_image_info", side_effect=["a", "b"]):
|
||
|
|
self.assertFalse(mod.volume_is_fully_ignored(["c1", "c2"], ["a"]))
|
||
|
|
|
||
|
|
def test_all_ignored_skips_the_volume(self) -> None:
|
||
|
|
with mock.patch.object(mod, "get_image_info", side_effect=["a", "a"]):
|
||
|
|
self.assertTrue(mod.volume_is_fully_ignored(["c1", "c2"], ["a"]))
|
||
|
|
|
||
|
|
|
||
|
|
class TestRequiresStop(unittest.TestCase):
|
||
|
|
def test_no_containers_means_no_stop(self) -> None:
|
||
|
|
self.assertFalse(mod.requires_stop([], []))
|
||
|
|
|
||
|
|
def test_a_swarm_task_never_forces_a_stop(self) -> None:
|
||
|
|
with mock.patch.object(mod, "is_swarm_task", return_value=True):
|
||
|
|
self.assertFalse(mod.requires_stop(["c1"], []))
|
||
|
|
|
||
|
|
def test_a_whitelisted_image_does_not_force_a_stop(self) -> None:
|
||
|
|
with (
|
||
|
|
mock.patch.object(mod, "is_swarm_task", return_value=False),
|
||
|
|
mock.patch.object(mod, "get_image_info", return_value="alpine:3.20"),
|
||
|
|
):
|
||
|
|
self.assertFalse(mod.requires_stop(["c1"], ["alpine:3.20"]))
|
||
|
|
|
||
|
|
def test_an_unlisted_image_forces_a_stop(self) -> None:
|
||
|
|
with (
|
||
|
|
mock.patch.object(mod, "is_swarm_task", return_value=False),
|
||
|
|
mock.patch.object(mod, "get_image_info", return_value="postgres:17"),
|
||
|
|
):
|
||
|
|
self.assertTrue(mod.requires_stop(["c1"], ["alpine:3.20"]))
|
||
|
|
|
||
|
|
def test_one_unlisted_container_is_enough(self) -> None:
|
||
|
|
with (
|
||
|
|
mock.patch.object(mod, "is_swarm_task", return_value=False),
|
||
|
|
mock.patch.object(mod, "get_image_info", side_effect=["alpine:3.20", "x"]),
|
||
|
|
):
|
||
|
|
self.assertTrue(mod.requires_stop(["c1", "c2"], ["alpine:3.20"]))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|