From a964ad05a8faa0ff2887ae9714af8dd6c61bec97 Mon Sep 17 00:00:00 2001
From: Sijis Aviles <sijis.aviles@gmail.com>
Date: Jun 03 2016 17:54:09 +0000
Subject: Add test cases for delete_build() function


---

diff --git a/tests/test_hub/test_delete_build.py b/tests/test_hub/test_delete_build.py
new file mode 100644
index 0000000..727f2ec
--- /dev/null
+++ b/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