admin/registry-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test delete_tags function

Browse Source
This commit is contained in:
Andrey Pohilko
2017-04-28 13:48:54 +06:00
parent 9b4f8b6893
commit d621c5abd3
1 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
test.py
View File

@@ -408,14 +408,35 @@ class TestDeleteTagsFunction(unittest.TestCase):
) )
def test_delete_tags_keep(self): 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( self.delete_mock.assert_called_with(
"imagename", "imagename",
"tag2", "tag1",
False, 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): class TestArgParser(unittest.TestCase):
def test_no_args(self): def test_no_args(self):
with self.assertRaises(SystemExit): with self.assertRaises(SystemExit):