blender-cloud/tests/test_linked_roles.py
Sybren A. Stüvel 0a9f0ebddb Roles '{flamenco,attract}-user' are now linked to 'subscriber'
All users who get 'subscriber' role automatically get 'flamenco-user' and
'attract-user', and when 'subscriber' is revoked, so are
'{flamenco,attract}-user'.
2017-07-11 12:40:13 +02:00

18 lines
664 B
Python

from abstract_cloud_test import AbstractCloudTest
class LinkedRolesTest(AbstractCloudTest):
def test_linked_roles(self):
user_id = self.create_user(roles=[])
db_user = self.fetch_user_from_db(user_id)
self.badger(db_user['email'], {'subscriber'}, 'grant')
db_user = self.fetch_user_from_db(user_id)
self.assertEqual({'subscriber', 'flamenco-user', 'attract-user'},
set(db_user['roles']))
self.badger(db_user['email'], {'subscriber'}, 'revoke')
db_user = self.fetch_user_from_db(user_id)
self.assertEqual(set(),
set(db_user.get('roles', [])))