e784373000
Unify error messages for CLI Unify error messages for hub Fixes: https://pagure.io/koji/issue/2720
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
import unittest
|
|
import mock
|
|
|
|
import koji
|
|
import kojihub
|
|
|
|
|
|
class TestCreateNotification(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.exports = kojihub.RootExports()
|
|
self.exports.getLoggedInUser = mock.MagicMock()
|
|
self.context = mock.patch('kojihub.context').start()
|
|
self.cursor = mock.MagicMock()
|
|
|
|
def test_non_exist_user(self):
|
|
user_id = 999
|
|
package_id = 555
|
|
tag_id = 111
|
|
success_only = False
|
|
logged_user = {'authtype': 2,
|
|
'id': 1,
|
|
'krb_principal': None,
|
|
'krb_principals': [],
|
|
'name': 'kojiadmin',
|
|
'status': 0,
|
|
'usertype': 0}
|
|
self.cursor.fetchone.return_value = None
|
|
self.context.cnx.cursor.return_value = self.cursor
|
|
self.exports.getLoggedInUser.return_value = logged_user
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
self.exports.createNotification(user_id, package_id, tag_id, success_only)
|
|
self.assertEqual('No such user ID: %s' % user_id, str(cm.exception))
|
|
|
|
|
|
class TestCreateNotificationBlock(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.exports = kojihub.RootExports()
|
|
self.exports.getLoggedInUser = mock.MagicMock()
|
|
self.context = mock.patch('kojihub.context').start()
|
|
self.cursor = mock.MagicMock()
|
|
|
|
def test_non_exist_user(self):
|
|
user_id = 999
|
|
package_id = 555
|
|
tag_id = 111
|
|
logged_user = {'authtype': 2,
|
|
'id': 1,
|
|
'krb_principal': None,
|
|
'krb_principals': [],
|
|
'name': 'kojiadmin',
|
|
'status': 0,
|
|
'usertype': 0}
|
|
self.cursor.fetchone.return_value = None
|
|
self.context.cnx.cursor.return_value = self.cursor
|
|
self.exports.getLoggedInUser.return_value = logged_user
|
|
with self.assertRaises(koji.GenericError) as cm:
|
|
self.exports.createNotificationBlock(user_id, package_id, tag_id)
|
|
self.assertEqual('No such user ID: %s' % user_id, str(cm.exception))
|