42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
|
import unittest
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from baudolo.backup import docker as docker_mod
|
||
|
|
from baudolo.backup.shell import BackupException
|
||
|
|
|
||
|
|
|
||
|
|
class TestIsSwarmTask(unittest.TestCase):
|
||
|
|
@patch.object(docker_mod, "execute_shell_command", return_value=["task-id-123"])
|
||
|
|
def test_true_when_task_label_present(self, _mock) -> None:
|
||
|
|
self.assertTrue(docker_mod.is_swarm_task("c1"))
|
||
|
|
|
||
|
|
@patch.object(docker_mod, "execute_shell_command", return_value=[""])
|
||
|
|
def test_false_when_label_empty(self, _mock) -> None:
|
||
|
|
self.assertFalse(docker_mod.is_swarm_task("c1"))
|
||
|
|
|
||
|
|
@patch.object(docker_mod, "execute_shell_command", return_value=[])
|
||
|
|
def test_false_when_no_output(self, _mock) -> None:
|
||
|
|
self.assertFalse(docker_mod.is_swarm_task("c1"))
|
||
|
|
|
||
|
|
@patch.object(
|
||
|
|
docker_mod,
|
||
|
|
"execute_shell_command",
|
||
|
|
side_effect=BackupException("gone"),
|
||
|
|
)
|
||
|
|
def test_vanished_container_counts_as_not_stoppable(self, _mock) -> None:
|
||
|
|
# A container removed between listing and inspect must not abort the
|
||
|
|
# whole backup run; treating it as a swarm task keeps it out of every
|
||
|
|
# stop/start and image-inspect path.
|
||
|
|
self.assertTrue(docker_mod.is_swarm_task("gone-container"))
|
||
|
|
|
||
|
|
|
||
|
|
class TestFilterStoppable(unittest.TestCase):
|
||
|
|
@patch.object(docker_mod, "is_swarm_task", side_effect=[False, True, False])
|
||
|
|
def test_mixed_list_keeps_order_and_drops_tasks(self, _mock) -> None:
|
||
|
|
result = docker_mod.filter_stoppable(["plain-1", "swarm-task", "plain-2"])
|
||
|
|
self.assertEqual(result, ["plain-1", "plain-2"])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|