From bf09cbc228cf0ac6457b1200e7e0f5d833cae956 Mon Sep 17 00:00:00 2001
From: Tomas Kopecek <tkopecek@redhat.com>
Date: Jun 29 2023 12:36:14 +0000
Subject: Fix user_in_group policy test


Related: https://pagure.io/koji/issue/3858

---

diff --git a/kojihub/auth.py b/kojihub/auth.py
index a858db2..dc52b81 100644
--- a/kojihub/auth.py
+++ b/kojihub/auth.py
@@ -776,7 +776,10 @@ def get_user_groups(user_id):
                                     'user_id=%(user_id)i'],
                            joins=['users ON group_id = users.id'],
                            values={'t_group': t_group, 'user_id': user_id})
-    return query.execute()
+    groups = {}
+    for gdata in query.execute():
+        groups[gdata['group_id']] = gdata['name']
+    return groups
 
 
 def get_user_perms(user_id):
diff --git a/tests/test_hub/test_policy_tests.py b/tests/test_hub/test_policy_tests.py
index 1fab261..95d791d 100644
--- a/tests/test_hub/test_policy_tests.py
+++ b/tests/test_hub/test_policy_tests.py
@@ -742,7 +742,7 @@ class IsBuildOwnerTest(unittest.TestCase):
         self.get_build.return_value = {'build_id': data['build'], 'owner_id': 3}
         self.get_user.return_value = {'id': 2, 'name': 'testuser', 'usertype': 2}
         self.policy_get_user.return_value = {'id': 3, 'name': 'username'}
-        self.get_user_groups.return_value = [2]
+        self.get_user_groups.return_value = {2: 'group_name'}
         obj = kojihub.IsBuildOwnerTest('isbuildownertest - owner group')
         self.assertTrue(obj.run(data))
         self.get_build.assert_called_once_with(data['build'])