57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
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):
|
|
context.session.assertPerm = mock.MagicMock()
|
|
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):
|
|
context.session.assertPerm = mock.MagicMock()
|
|
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):
|
|
context.session.assertPerm = mock.MagicMock()
|
|
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
|
|
self.assertFalse(kojihub.delete_build(build='', strict=False))
|