From d621c5abd36a04f4330aef30327249bc7a5d1ea2 Mon Sep 17 00:00:00 2001
From: Andrey Pohilko <anoxis@gmail.com>
Date: Fri, 28 Apr 2017 13:48:54 +0600
Subject: [PATCH] test delete_tags function

---
 test.py | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/test.py b/test.py
index 85cbf0a..a773bb8 100644
--- 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):