From 76f2367e6609eabed901a7090cb604e236f085e3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= Date: Fri, 11 Nov 2016 15:23:25 +0100 Subject: [PATCH] Added extra role to UserAdminTest. --- tests/test_web/test_user_admin.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/tests/test_web/test_user_admin.py b/tests/test_web/test_user_admin.py index 4a1cfa51..7e953e2e 100644 --- a/tests/test_web/test_user_admin.py +++ b/tests/test_web/test_user_admin.py @@ -32,7 +32,7 @@ class UserAdminTest(AbstractPillarTest): import pillar.web.users.routes import pillar.auth - user_id = self.create_user(roles=()) + user_id = self.create_user(roles=(u'mønkeybütler', )) self.create_valid_auth_token(user_id, 'token') # Try to access the home project, creating it. @@ -69,12 +69,16 @@ class UserAdminTest(AbstractPillarTest): edit_user(['subscriber', 'demo']) # Re-check the user group membership. - groups = get_dbuser()['groups'] + dbuser = get_dbuser() self.assertEqual({home_project_gid, self.subscriber_gid, self.demo_gid}, - set(groups)) + set(dbuser['groups'])) + self.assertEqual({u'subscriber', u'demo', u'mønkeybütler'}, + set(dbuser['roles'])) # Edit user again, revoking demo role. edit_user(['subscriber']) - groups = get_dbuser()['groups'] + dbuser = get_dbuser() self.assertEqual({home_project_gid, self.subscriber_gid}, - set(groups)) + set(dbuser['groups'])) + self.assertEqual({u'subscriber', u'mønkeybütler'}, + set(dbuser['roles']))