test delete_tags function

2017-04-28 13:48:54 +06:00
parent 9b4f8b6893
commit d621c5abd3
1 changed files with 24 additions and 3 deletions
--- a/test.py
+++ b/test.py
@@ -408,14 +408,35 @@ class TestDeleteTagsFunction(unittest.TestCase):
        )

    def test_delete_tags_keep(self):
-        delete_tags(self.registry, "imagename", False, ["tag1", "tag2"], ["tag_to_keep"])
+        digest_mock = MagicMock(return_value = "DIGEST_MOCK")
+        self.registry.get_tag_digest = digest_mock
+
+        delete_tags(self.registry, "imagename", False, ["tag1", "tag2"], ["tag2"])
+
+        digest_mock.assert_called_with("imagename", "tag2")
+
        self.delete_mock.assert_called_with(
            "imagename",
-            "tag2",
+            "tag1",
            False,
-            ['MOCK_DIGEST_HEADER']
+            ['DIGEST_MOCK']
        )

+    def test_delete_tags_digest_none(self):
+        digest_mock = MagicMock(return_value = None)
+        self.registry.get_tag_digest = digest_mock
+        delete_tags(self.registry, "imagename", False, ["tag1", "tag2"], ["tag2"])
+
+        digest_mock.assert_called_with("imagename", "tag2")
+
+        self.delete_mock.assert_called_with(
+            "imagename",
+            "tag1",
+            False,
+            []
+        )
+
+
 class TestArgParser(unittest.TestCase):
    def test_no_args(self):
        with self.assertRaises(SystemExit):