24 lines
751 B
Python
24 lines
751 B
Python
import mock
|
|
import unittest
|
|
import koji
|
|
import kojihub
|
|
|
|
|
|
class TestGetUserPerms(unittest.TestCase):
|
|
def setUp(self):
|
|
self.get_user = mock.patch('kojihub.get_user').start()
|
|
self.get_user_perms = mock.patch('koji.auth.get_user_perms').start()
|
|
|
|
def tearDown(self):
|
|
mock.patch.stopall()
|
|
|
|
def test_no_user(self):
|
|
self.get_user.side_effect = koji.GenericError
|
|
with self.assertRaises(koji.GenericError):
|
|
kojihub.RootExports().getUserPerms(123)
|
|
self.get_user_perms.assert_not_called()
|
|
|
|
def test_normal(self):
|
|
self.get_user.return_value = {'id': 123, 'name': 'testuser'}
|
|
kojihub.RootExports().getUserPerms(123)
|
|
self.get_user_perms.assert_called_once_with(123)
|