Also allow demo users access to Flamenco & Attract

tests/test_linked_roles.py

@@ -2,7 +2,7 @@ from abstract_cloud_test import AbstractCloudTest
 
 
 class LinkedRolesTest(AbstractCloudTest):
-    def test_linked_roles(self):
+    def test_linked_roles_subscriber(self):
         user_id = self.create_user(roles=[])
         db_user = self.fetch_user_from_db(user_id)
 
@@ -15,3 +15,17 @@ class LinkedRolesTest(AbstractCloudTest):
         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', [])))