Home project: create it when user tries to GET it.

tests/common_test_class.py

@@ -106,7 +106,7 @@ class AbstractPillarTest(TestMinimal):
 
             return found['_id'], found
 
-    def create_user(self, user_id='cafef00dc379cf10c4aaceaf', roles=('subscriber', )):
+    def create_user(self, user_id='cafef00dc379cf10c4aaceaf', roles=('subscriber',)):
         from application.utils.authentication import make_unique_username
 
         with self.app.test_request_context():
@@ -141,6 +141,34 @@ class AbstractPillarTest(TestMinimal):
 
         return token_data
 
+    def badger(self, user_email, role, action, srv_token=None):
+        """Creates a service account, and uses it to grant or revoke a role to the user.
+
+        To skip creation of the service account, pass a srv_token.
+
+        :returns: the authentication token of the created service account.
+        :rtype: str
+        """
+
+        # Create a service account if needed.
+        if srv_token is None:
+            from application.modules.service import create_service_account
+            with self.app.test_request_context():
+                _, srv_token_doc = create_service_account('service@example.com',
+                                                          {'badger'},
+                                                          {'badger': [role]})
+                srv_token = srv_token_doc['token']
+
+        resp = self.client.post('/service/badger',
+                                headers={'Authorization': self.make_header(srv_token),
+                                         'Content-Type': 'application/json'},
+                                data=json.dumps({'action': action,
+                                                 'role': role,
+                                                 'user_email': user_email}))
+        self.assertEqual(204, resp.status_code, resp.data)
+
+        return srv_token
+
     def mock_blenderid_validate_unhappy(self):
         """Sets up Responses to mock unhappy validation flow."""
 

tests/test_bcloud_home_project.py

@@ -0,0 +1,108 @@
+# -*- encoding: utf-8 -*-
+
+"""Unit tests for the Blender Cloud home project module."""
+
+import functools
+import json
+import logging
+import urllib
+
+import responses
+from bson import ObjectId
+from flask import g, url_for
+
+from common_test_class import AbstractPillarTest, TEST_EMAIL_ADDRESS
+
+log = logging.getLogger(__name__)
+
+
+class HomeProjectTest(AbstractPillarTest):
+
+    def setUp(self, **kwargs):
+        AbstractPillarTest.setUp(self, **kwargs)
+        self.create_standard_groups()
+
+    def _create_user_with_token(self, roles, token, user_id='cafef00df00df00df00df00d'):
+        """Creates a user directly in MongoDB, so that it doesn't trigger any Eve hooks."""
+        user_id = self.create_user(roles=roles, user_id=user_id)
+        self.create_valid_auth_token(user_id, token)
+        return user_id
+
+    def test_create_home_project(self):
+        from application.modules.blender_cloud import home_project
+        from application.utils.authentication import validate_token
+
+        user_id = self._create_user_with_token(roles={u'subscriber'}, token='token')
+
+        # Test home project creation
+        with self.app.test_request_context(headers={'Authorization': self.make_header('token')}):
+            validate_token()
+
+            proj = home_project.create_home_project(user_id)
+            self.assertEqual('home', proj['category'])
+            self.assertEqual({u'text', u'group', u'asset'},
+                             set(nt['name'] for nt in proj['node_types']))
+
+            endpoint = url_for('blender_cloud.home_project.home_project')
+            db_proj = self.app.data.driver.db['projects'].find_one(proj['_id'])
+
+        # Test availability at end-point
+        resp = self.client.get(endpoint)
+        self.assertEqual(403, resp.status_code)
+
+        resp = self.client.get(endpoint, headers={'Authorization': self.make_header('token')})
+        self.assertEqual(200, resp.status_code)
+
+        json_proj = json.loads(resp.data)
+        self.assertEqual(ObjectId(json_proj['_id']), proj['_id'])
+        self.assertEqual(json_proj['_etag'], db_proj['_etag'])
+
+    @responses.activate
+    def test_autocreate_home_project_after_getting_subscriber_role(self):
+        from application.modules.blender_cloud import home_project
+
+        # Implicitly create user by token validation.
+        self.mock_blenderid_validate_happy()
+        resp = self.client.get('/users/me', headers={'Authorization': self.make_header('token')})
+        self.assertEqual(200, resp.status_code, resp)
+
+        # Grant subscriber role, and fetch the home project.
+        self.badger(TEST_EMAIL_ADDRESS, 'subscriber', 'grant')
+
+        resp = self.client.get('/bcloud/home-project',
+                               headers={'Authorization': self.make_header('token')})
+        self.assertEqual(200, resp.status_code)
+
+        json_proj = json.loads(resp.data)
+        self.assertEqual('home', json_proj['category'])
+
+    @responses.activate
+    def test_autocreate_home_project_after_getting_demo_role(self):
+        # Implicitly create user by token validation.
+        self.mock_blenderid_validate_happy()
+        resp = self.client.get('/users/me', headers={'Authorization': self.make_header('token')})
+        self.assertEqual(200, resp.status_code, resp)
+
+        # Grant demo role, which should also should allow creation fo the home project.
+        self.badger(TEST_EMAIL_ADDRESS, 'demo', 'grant')
+
+        resp = self.client.get('/bcloud/home-project',
+                               headers={'Authorization': self.make_header('token')})
+        self.assertEqual(200, resp.status_code)
+
+        json_proj = json.loads(resp.data)
+        self.assertEqual('home', json_proj['category'])
+
+    def test_has_home_project(self):
+        from application.modules.blender_cloud import home_project
+        from application.utils.authentication import validate_token
+
+        user_id = self._create_user_with_token(roles={u'subscriber'}, token='token')
+
+        # Test home project creation
+        with self.app.test_request_context(headers={'Authorization': self.make_header('token')}):
+            validate_token()
+
+            self.assertFalse(home_project.has_home_project(user_id))
+            home_project.create_home_project(user_id)
+            self.assertTrue(home_project.has_home_project(user_id))

tests/test_project_management.py

@@ -138,13 +138,13 @@ class ProjectCreationTest(AbstractProjectTest):
                                headers={'Authorization': self.make_header('token-a')})
         self.assertEqual(200, resp.status_code)
         proj_list = json.loads(resp.data)
-        self.assertEqual([u'Prøject A'], [p['name'] for p in proj_list['_items']])
+        self.assertEqual({u'Prøject A'}, {p['name'] for p in proj_list['_items']})
 
         resp = self.client.get('/projects',
                                headers={'Authorization': self.make_header('token-b')})
         self.assertEqual(200, resp.status_code)
         proj_list = json.loads(resp.data)
-        self.assertEqual([u'Prøject B'], [p['name'] for p in proj_list['_items']])
+        self.assertEqual({u'Prøject B'}, {p['name'] for p in proj_list['_items']})
 
         # No access to anything for user C, should result in empty list.
         self._create_user_with_token(roles={u'subscriber'}, token='token-c', user_id=12 * 'c')