From dd58d4ad043d7d6c834897ae1a698b0bc3a524fc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= Date: Fri, 30 Sep 2016 12:54:15 +0200 Subject: [PATCH] Created AbstractPillarTest.create_project_admin() function. --- pillar/tests/__init__.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/pillar/tests/__init__.py b/pillar/tests/__init__.py index 2c451fce..707be923 100644 --- a/pillar/tests/__init__.py +++ b/pillar/tests/__init__.py @@ -229,12 +229,24 @@ class AbstractPillarTest(TestMinimal): :rtype: tuple """ project_id, proj = self.ensure_project_exists() - admin_group_id = proj['permissions']['groups'][0]['group'] - - user_id = self.create_user(user_id=user_id, roles=roles, groups=[admin_group_id]) + user_id = self.create_project_admin(proj, user_id, roles) return project_id, user_id + def create_project_admin(self, proj, user_id='cafef00dc379cf10c4aaceaf', roles=('subscriber', )): + """Creates a user that's member of the project's admin group. + + :param proj: project document, or at least a dict with permissions in it. + :type proj: dict + :returns: user_id + :rtype: ObjectId + """ + + admin_group_id = proj['permissions']['groups'][0]['group'] + user_id = self.create_user(user_id=user_id, roles=roles, groups=[admin_group_id]) + + return user_id + def badger(self, user_email, roles, action, srv_token=None): """Creates a service account, and uses it to grant or revoke a role to the user.