64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
import unittest
|
|
|
|
import mock
|
|
|
|
import koji
|
|
import kojihub
|
|
|
|
|
|
class TestGetUser(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.exports = kojihub.RootExports()
|
|
|
|
def test_wrong_format_user_info(self):
|
|
userinfo = ['test-user']
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
self.exports.getUser(userinfo)
|
|
self.assertEqual("Invalid type for userInfo: %s" % type(userinfo), str(cm.exception))
|
|
|
|
def test_wrong_format_userid(self):
|
|
userinfo = {'id': '123456'}
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
self.exports.getUser(userinfo)
|
|
self.assertEqual("Invalid type for userid: %s" % type(userinfo['id']), str(cm.exception))
|
|
|
|
def test_wrong_format_username(self):
|
|
userinfo = {'name': 57896}
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
self.exports.getUser(userinfo)
|
|
self.assertEqual("Invalid type for username: %s" % type(userinfo['name']),
|
|
str(cm.exception))
|
|
|
|
def test_wrong_format_krb_principal(self):
|
|
userinfo = {'krb_principal': 57896}
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
self.exports.getUser(userinfo)
|
|
self.assertEqual("Invalid type for krb_principal: %s" % type(userinfo['krb_principal']),
|
|
str(cm.exception))
|
|
|
|
|
|
class TestGetUserByKrbPrincipal(unittest.TestCase):
|
|
def setUp(self):
|
|
self.get_user = mock.patch('kojihub.get_user').start()
|
|
|
|
def test_wrong_type_krb_principal(self):
|
|
krb_principal = ['test-user']
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
kojihub.get_user_by_krb_principal(krb_principal)
|
|
self.assertEqual("Invalid type for krb_principal: %s" % type(krb_principal),
|
|
str(cm.exception))
|
|
|
|
def test_krb_principal_none(self):
|
|
krb_principal = None
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
kojihub.get_user_by_krb_principal(krb_principal)
|
|
self.assertEqual("No kerberos principal provided", str(cm.exception))
|
|
|
|
def test_valid(self):
|
|
krb_principal = 'test-user@test.org'
|
|
user_info = {'id': 1, 'krb_principals': ['test-user@test.org'],
|
|
'name': 'test-user', 'status': 0, 'usertype': 0}
|
|
self.get_user.return_value = user_info
|
|
rv = kojihub.get_user_by_krb_principal(krb_principal)
|
|
self.assertEqual(user_info, rv)
|