archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

Added unittests for caching of file links.

Browse Source 
Also converted test_auth.py to use the new AbstractPillarTest class.
This class ensures that we test against the testing database, which
is dropped at every setUp()/tearDown().
This commit is contained in:
Sybren A. Stüvel
2016-03-25 15:57:17 +01:00
parent cb4b0f1e4d
commit adb4f5b39e
5 changed files with 440 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
tests/test_auth.py
View File

@@ -1,23 +1,7 @@
import unittest
import os
import base64
import httpretty
import json
BLENDER_ID_ENDPOINT = 'http://127.0.0.1:8001' # nonexistant server, no trailing slash!
TEST_EMAIL_USER = 'koro'
TEST_EMAIL_ADDRESS = '%s@testing.blender.org' % TEST_EMAIL_USER
os.environ['BLENDER_ID_ENDPOINT'] = BLENDER_ID_ENDPOINT
os.environ['MONGO_DBNAME'] = 'unittest'
os.environ['EVE_SETTINGS'] = os.path.join(
os.path.dirname(os.path.dirname(__file__)),
'pillar', 'settings.py')
from application import app
from application.utils import authentication as auth
app.config['BLENDER_ID_ENDPOINT'] = BLENDER_ID_ENDPOINT
from common_test_stuff import AbstractPillarTest, TEST_EMAIL_USER, TEST_EMAIL_ADDRESS
def make_header(username, password=''):
@@ -26,19 +10,11 @@ def make_header(username, password=''):
return 'basic ' + base64.b64encode('%s:%s' % (username, password))
class AuthenticationTests(unittest.TestCase):
def setUp(self):
self.app = app.test_client()
with app.test_request_context():
self.delete_test_data()
def tearDown(self):
with app.test_request_context():
self.delete_test_data()
class AuthenticationTests(AbstractPillarTest):
def test_make_unique_username(self):
from application.utils import authentication as auth
with app.test_request_context():
with self.app.test_request_context():
# This user shouldn't exist yet.
self.assertEqual(TEST_EMAIL_USER, auth.make_unique_username(TEST_EMAIL_ADDRESS))
@@ -46,44 +22,29 @@ class AuthenticationTests(unittest.TestCase):
auth.create_new_user(TEST_EMAIL_ADDRESS, TEST_EMAIL_USER, 'test1234')
self.assertEqual('%s1' % TEST_EMAIL_USER, auth.make_unique_username(TEST_EMAIL_ADDRESS))
def delete_test_data(self):
app.data.driver.db.drop_collection('users')
app.data.driver.db.drop_collection('tokens')
def blenderid_validate_unhappy(self):
"""Sets up HTTPretty to mock unhappy validation flow."""
httpretty.register_uri(httpretty.POST,
'%s/u/validate_token' % BLENDER_ID_ENDPOINT,
body=json.dumps({'data': {'token': 'Token is invalid'}, 'status': 'fail'}),
content_type="application/json")
def blenderid_validate_happy(self):
"""Sets up HTTPretty to mock happy validation flow."""
httpretty.register_uri(httpretty.POST,
'%s/u/validate_token' % BLENDER_ID_ENDPOINT,
body=json.dumps({'data': {'user': {'email': TEST_EMAIL_ADDRESS, 'id': 5123}},
'status': 'success'}),
content_type="application/json")
@httpretty.activate
def test_validate_token__not_logged_in(self):
with app.test_request_context():
from application.utils import authentication as auth
with self.app.test_request_context():
self.assertFalse(auth.validate_token())
@httpretty.activate
def test_validate_token__unknown_token(self):
"""Test validating of invalid token, unknown both to us and Blender ID."""
self.blenderid_validate_unhappy()
with app.test_request_context(headers={'Authorization': make_header('unknowntoken')}):
from application.utils import authentication as auth
self.htp_blenderid_validate_unhappy()
with self.app.test_request_context(headers={'Authorization': make_header('unknowntoken')}):
self.assertFalse(auth.validate_token())
@httpretty.activate
def test_validate_token__unknown_but_valid_token(self):
"""Test validating of valid token, unknown to us but known to Blender ID."""
self.blenderid_validate_happy()
with app.test_request_context(headers={'Authorization': make_header('knowntoken')}):
from application.utils import authentication as auth
self.htp_blenderid_validate_happy()
with self.app.test_request_context(headers={'Authorization': make_header('knowntoken')}):
self.assertTrue(auth.validate_token())