fix unit test

This commit is contained in:
Mike McLean 2019-05-20 12:29:57 -04:00
parent 5a36065d62
commit 41628eb5e3

View file

@ -16,7 +16,7 @@ class TestDeleteBuild(unittest.TestCase):
@mock.patch('kojihub.get_build')
def test_delete_build_raise_error(self, build, context):
context.session.assertPerm = mock.MagicMock()
references = ['tags', 'rpms', 'archives', 'images']
references = ['tags', 'rpms', 'archives', 'component_of']
for ref in references:
context = mock.MagicMock()
context.session.return_value = context
@ -32,7 +32,7 @@ class TestDeleteBuild(unittest.TestCase):
@mock.patch('kojihub.get_build')
def test_delete_build_return_false(self, build, context):
context.session.assertPerm = mock.MagicMock()
references = ['tags', 'rpms', 'archives', 'images']
references = ['tags', 'rpms', 'archives', 'component_of']
for ref in references:
context = mock.MagicMock()
context.session.return_value = context
@ -48,7 +48,7 @@ class TestDeleteBuild(unittest.TestCase):
@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']
references = ['tags', 'rpms', 'archives', 'component_of', 'last_used']
for ref in references:
context = mock.MagicMock()
context.session.return_value = context