debian-koji/tests/test_hub/test_perm_operations.py
Yuming Zhu ca05418fb5 unittest: use unittest.mock instead of mock
because the absence of unittest.mock on python2.7, we still fallback to
mock
2024-10-23 16:35:30 +00:00

26 lines
869 B
Python

from unittest import mock
import unittest
import koji
import kojihub
class TestPerm(unittest.TestCase):
@mock.patch('kojihub.kojihub.context')
@mock.patch('kojihub.kojihub.lookup_perm', return_value=None)
def test_has_perm(self, lookup_perm, context):
rv = kojihub.RootExports().hasPerm('perm')
self.assertEqual(rv, context.session.hasPerm.return_value)
lookup_perm.assert_not_called()
context.session.hasPerm.assert_called_once_with('perm')
context.session.hasPerm.reset_mock()
with self.assertRaises(koji.GenericError) as cm:
kojihub.RootExports().hasPerm('perm', strict=True)
lookup_perm.assert_called_once_with('perm')
context.session.hasPerm.assert_not_called()
self.assertEqual(cm.exception.args[0],
'No such permission perm defined')