76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
from __future__ import absolute_import
|
|
import mock
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
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):
|
|
context.session.assertPerm = mock.MagicMock()
|
|
references = ['tags', 'rpms', 'archives', 'component_of']
|
|
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):
|
|
context.session.assertPerm = mock.MagicMock()
|
|
references = ['tags', 'rpms', 'archives', 'component_of']
|
|
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):
|
|
context.session.assertPerm = mock.MagicMock()
|
|
references = ['tags', 'rpms', 'archives', 'component_of', '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
|
|
self.assertFalse(kojihub.delete_build(build='', strict=False))
|
|
|
|
@mock.patch('kojihub._delete_build')
|
|
@mock.patch('kojihub.build_references')
|
|
@mock.patch('kojihub.context')
|
|
@mock.patch('kojihub.get_build')
|
|
def test_delete_build_lazy_refs(self, build, context, buildrefs, _delete):
|
|
'''Test that we can handle lazy return from build_references'''
|
|
context.session.assertPerm = mock.MagicMock()
|
|
buildrefs.return_value = {'tags': []}
|
|
binfo = {'id': 'BUILD ID'}
|
|
build.return_value = binfo
|
|
kojihub.delete_build(build=binfo, strict=True)
|
|
|
|
# no build refs, so we should have called _delete_build
|
|
_delete.assert_called_with(binfo)
|