Make getGroupMembers anonymous and add getUserGroups

This commit is contained in:
Christopher O'Brien 2024-07-09 11:49:50 -04:00 committed by Tomas Kopecek
parent d982b5ac4f
commit 0bd816c689
3 changed files with 75 additions and 5 deletions

View file

@ -0,0 +1,52 @@
import mock
import koji
import kojihub
from .utils import DBQueryTestCase
class TestGetUserGroups(DBQueryTestCase):
def setUp(self):
super(TestGetUserGroups, self).setUp()
self.exports = kojihub.RootExports()
self.context = mock.patch('kojihub.kojihub.context').start()
self.context_db = mock.patch('kojihub.db.context').start()
def tearDown(self):
mock.patch.stopall()
def test_no_such_user(self):
user = 'test-user'
self.qp_execute_return_value = []
with self.assertRaises(koji.GenericError) as cm:
self.exports.getUserGroups(user)
self.assertEqual(f"No such user: {user!r}", str(cm.exception))
self.assertEqual(len(self.queries), 1)
def test_valid(self):
get_user = mock.patch('kojihub.kojihub.get_user').start()
get_user.return_value = {'id': 23, 'usertype': 0}
user = 'test-user'
self.qp_execute_return_value = [{'group_id': 123, 'name': 'grp_123'},
{'group_id': 456, 'name': 'grp_456'}]
grps = self.exports.getUserGroups(user)
get_user.assert_called_once_with(user, strict=True)
self.assertEqual(len(self.queries), 1)
query = self.queries[0]
self.assertEqual(query.tables, ['user_groups'])
self.assertEqual(query.joins, ['users ON group_id = users.id'])
self.assertEqual(query.clauses, ['active IS TRUE',
'user_id=%(user_id)i',
'users.usertype=%(t_group)i'])
self.assertEqual(query.values, {'t_group': 2,
'user_id': 23})
self.assertEqual(query.columns, ['group_id', 'name'])
self.assertEqual(grps, {123: 'grp_123',
456: 'grp_456'})

View file

@ -262,11 +262,11 @@ class TestGrouplist(unittest.TestCase):
def test_get_group_members(self):
group, gid = 'test_group', 1
# no permission
self.context.session.assertPerm.side_effect = koji.ActionNotAllowed
with self.assertRaises(koji.ActionNotAllowed):
# no permission needed, verify that it's not being checked
self.context.session.assertPerm.side_effect = Exception
with self.assertRaises(koji.GenericError):
kojihub.get_group_members(group)
self.context.session.assertPerm.assert_called_with('admin')
self.context.session.assertPerm.assert_not_called()
self.assertEqual(len(self.inserts), 0)
self.assertEqual(len(self.updates), 0)