Add test cases for delete_build() function

tests/test_hub/test_delete_build.py

@@ -0,0 +1,73 @@
+import mock
+import unittest
+import kojihub
+import time
+from koji import GenericError
+from collections import defaultdict
+
+
+class TestDeleteBuild(unittest.TestCase):
+
+    @mock.patch('kojihub.context')
+    @mock.patch('kojihub.get_build')
+    def test_delete_build_raise_error(self, build, context):
+
+        references = ['tags', 'rpms', 'archives', 'images']
+        for ref in references:
+            context = mock.MagicMock()
+            context.session.return_value = context
+
+            with mock.patch('kojihub.build_references') as refs:
+                retval = defaultdict(dict)
+                retval[ref] = True
+                refs.return_value = retval
+                with self.assertRaises(GenericError):
+                    kojihub.delete_build(build='', strict=True)
+
+    @mock.patch('kojihub.context')
+    @mock.patch('kojihub.get_build')
+    def test_delete_build_return_false(self, build, context):
+
+        references = ['tags', 'rpms', 'archives', 'images']
+        for ref in references:
+            context = mock.MagicMock()
+            context.session.return_value = context
+
+            with mock.patch('kojihub.build_references') as refs:
+                retval = defaultdict(dict)
+                retval[ref] = True
+                refs.return_value = retval
+                assert kojihub.delete_build(build='', strict=False) is False
+
+    @mock.patch('kojihub.context')
+    @mock.patch('kojihub.get_build')
+    def test_delete_build_check_last_used_raise_error(self, build, context):
+
+        references = ['tags', 'rpms', 'archives', 'images', 'last_used']
+        for ref in references:
+            context = mock.MagicMock()
+            context.session.return_value = context
+
+            with mock.patch('kojihub.build_references') as refs:
+                retval = defaultdict(dict)
+                if ref == 'last_used':
+                    retval[ref] = time.time()+100
+                    refs.return_value = retval
+                    with self.assertRaises(GenericError):
+                        kojihub.delete_build(build='', strict=True)
+
+    @mock.patch('kojihub.context')
+    @mock.patch('kojihub.get_build')
+    def test_delete_build_check_last_used_raise_error(self, build, context):
+
+        references = ['tags', 'rpms', 'archives', 'images', 'last_used']
+        for ref in references:
+            context = mock.MagicMock()
+            context.session.return_value = context
+
+            with mock.patch('kojihub.build_references') as refs:
+                retval = defaultdict(dict)
+                if ref == 'last_used':
+                    retval[ref] = time.time()+100
+                    refs.return_value = retval
+                    assert kojihub.delete_build(build='', strict=False) is False