mirror of
https://github.com/kevinveenbirkenbach/docker-volume-backup.git
synced 2026-07-17 06:05:13 +00:00
Kevin Veen-Birkenbach
79214e64e8
requires_stop now probes is_swarm_task per container, which runs a real docker inspect; the unit CI container has no docker socket, so the three whitelist tests died with BackupException. Mock the probe to False so they assert the unchanged whitelist logic, and add a swarm case proving a task container never triggers a stop and skips the image check entirely. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from baudolo.backup.app import requires_stop
|
|
|
|
|
|
@patch("baudolo.backup.app.is_swarm_task", return_value=False)
|
|
class TestRequiresStop(unittest.TestCase):
|
|
@patch("baudolo.backup.app.get_image_info")
|
|
def test_requires_stop_false_when_all_images_are_whitelisted(
|
|
self, mock_get_image_info, _mock_is_swarm_task
|
|
):
|
|
# All containers use images containing allowed substrings
|
|
mock_get_image_info.side_effect = [
|
|
"repo/mastodon:v4",
|
|
"repo/wordpress:latest",
|
|
]
|
|
containers = ["c1", "c2"]
|
|
whitelist = ["mastodon", "wordpress"]
|
|
self.assertFalse(requires_stop(containers, whitelist))
|
|
|
|
@patch("baudolo.backup.app.get_image_info")
|
|
def test_requires_stop_true_when_any_image_is_not_whitelisted(
|
|
self, mock_get_image_info, _mock_is_swarm_task
|
|
):
|
|
mock_get_image_info.side_effect = [
|
|
"repo/mastodon:v4",
|
|
"repo/nginx:latest",
|
|
]
|
|
containers = ["c1", "c2"]
|
|
whitelist = ["mastodon", "wordpress"]
|
|
self.assertTrue(requires_stop(containers, whitelist))
|
|
|
|
@patch("baudolo.backup.app.get_image_info")
|
|
def test_requires_stop_true_when_whitelist_empty(
|
|
self, mock_get_image_info, _mock_is_swarm_task
|
|
):
|
|
mock_get_image_info.return_value = "repo/anything:latest"
|
|
self.assertTrue(requires_stop(["c1"], []))
|
|
|
|
|
|
class TestRequiresStopSwarm(unittest.TestCase):
|
|
@patch("baudolo.backup.app.get_image_info")
|
|
@patch("baudolo.backup.app.is_swarm_task", return_value=True)
|
|
def test_swarm_tasks_never_require_stop(
|
|
self, _mock_is_swarm_task, mock_get_image_info
|
|
):
|
|
mock_get_image_info.return_value = "repo/not-whitelisted:latest"
|
|
self.assertFalse(requires_stop(["c1"], []))
|
|
mock_get_image_info.assert_not_called()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|