32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
from abstract_cloud_test import AbstractCloudTest
|
|
|
|
|
|
class LinkedRolesTest(AbstractCloudTest):
|
|
def test_linked_roles_subscriber(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', [])))
|
|
|
|
def test_linked_roles_demo(self):
|
|
user_id = self.create_user(roles=[])
|
|
db_user = self.fetch_user_from_db(user_id)
|
|
|
|
self.badger(db_user['email'], {'demo'}, 'grant')
|
|
db_user = self.fetch_user_from_db(user_id)
|
|
self.assertEqual({'demo', 'flamenco-user', 'attract-user'},
|
|
set(db_user['roles']))
|
|
|
|
self.badger(db_user['email'], {'demo'}, 'revoke')
|
|
db_user = self.fetch_user_from_db(user_id)
|
|
self.assertEqual(set(),
|
|
set(db_user.get('roles', [])))
|