109 lines
4.2 KiB
Python
109 lines
4.2 KiB
Python
# -*- 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))
|