Files
blender-cloud/tests/test_linked_roles.py
Francesco Siddi d46b5d645b Revert "Removed flamenco-user and attract-user role linking to subscriber/demo/admin"
Temporarily reverting due to an issue with missing roles and permissions.

This reverts commit c031c1e8ae.
2017-12-08 17:42:35 +01:00

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', [])))