70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
|
import unittest
|
||
|
|
from unittest.mock import patch, MagicMock
|
||
|
|
|
||
|
|
import dockreap.__main__ as dockreap_main
|
||
|
|
|
||
|
|
|
||
|
|
class TestDockreapMain(unittest.TestCase):
|
||
|
|
def test_get_anonymous_volumes_filters_64char_hex(self):
|
||
|
|
fake_stdout = "\n".join(
|
||
|
|
[
|
||
|
|
"short",
|
||
|
|
"nothex" * 10,
|
||
|
|
"a" * 64, # valid
|
||
|
|
"A" * 64, # invalid (uppercase not matched)
|
||
|
|
"g" * 64, # invalid (not hex)
|
||
|
|
"b" * 63, # invalid (too short)
|
||
|
|
"c" * 65, # invalid (too long)
|
||
|
|
"deadbeef" * 8, # valid 64-char hex
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
completed = MagicMock()
|
||
|
|
completed.stdout = fake_stdout
|
||
|
|
|
||
|
|
with patch.object(dockreap_main.subprocess, "run", return_value=completed) as run_mock:
|
||
|
|
vols = dockreap_main.get_anonymous_volumes()
|
||
|
|
|
||
|
|
# Ensure docker command was called correctly
|
||
|
|
run_mock.assert_called_with(
|
||
|
|
["docker", "volume", "ls", "--format", "{{.Name}}"],
|
||
|
|
stdout=dockreap_main.subprocess.PIPE,
|
||
|
|
text=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertEqual(vols, ["a" * 64, "deadbeef" * 8])
|
||
|
|
|
||
|
|
def test_is_volume_used_true_when_docker_returns_container_ids(self):
|
||
|
|
completed = MagicMock()
|
||
|
|
completed.stdout = "abc123\n"
|
||
|
|
|
||
|
|
with patch.object(dockreap_main.subprocess, "run", return_value=completed):
|
||
|
|
self.assertTrue(dockreap_main.is_volume_used("x" * 64))
|
||
|
|
|
||
|
|
def test_is_volume_used_false_when_no_container_ids(self):
|
||
|
|
completed = MagicMock()
|
||
|
|
completed.stdout = "\n"
|
||
|
|
|
||
|
|
with patch.object(dockreap_main.subprocess, "run", return_value=completed):
|
||
|
|
self.assertFalse(dockreap_main.is_volume_used("x" * 64))
|
||
|
|
|
||
|
|
def test_cleanup_symlink_does_nothing_when_not_symlink(self):
|
||
|
|
# We don't want to touch the real filesystem, so we patch the Path object
|
||
|
|
fake_data_path = MagicMock()
|
||
|
|
fake_data_path.is_symlink.return_value = False
|
||
|
|
|
||
|
|
fake_volume_dir = MagicMock()
|
||
|
|
fake_volume_dir.__truediv__.side_effect = lambda x: fake_data_path if x == "_data" else MagicMock()
|
||
|
|
|
||
|
|
with patch.object(dockreap_main, "VOLUME_BASE_PATH") as base_path_mock:
|
||
|
|
base_path_mock.__truediv__.return_value = fake_volume_dir
|
||
|
|
|
||
|
|
# Should not raise; should not try to unlink/rmtree
|
||
|
|
dockreap_main.cleanup_symlink("a" * 64)
|
||
|
|
|
||
|
|
fake_data_path.unlink.assert_not_called()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|