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

Added a few unit tests for user authentication.

Browse Source 
Far from complete, and we need a way to mock the Blender ID server, so
that we can auth against a well-known, fake set of users.
This commit is contained in:
Sybren A. Stüvel 2016-03-04 14:47:48 +01:00
parent 1e141492a3
commit 7ca22691eb
2 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pillar/application/utils/authentication.py
View File

@ -125,6 +125,13 @@ def validate_token():
def make_unique_username(email): def make_unique_username(email):
"""Creates a unique username from the email address.
@param email: the email address
@returns: the new username
@rtype: str
"""
username = email.split('@')[0] username = email.split('@')[0]
# Check for min length of username (otherwise validation fails) # Check for min length of username (otherwise validation fails)
username = "___{0}".format(username) if len(username) < 3 else username username = "___{0}".format(username) if len(username) < 3 else username

18
pillar/tests/test_auth.py
View File

@ -1,5 +1,6 @@
import unittest import unittest
import os import os
import base64
TEST_EMAIL_USER = 'koro' TEST_EMAIL_USER = 'koro'
TEST_EMAIL_ADDRESS = '%s@testing.blender.org' % TEST_EMAIL_USER TEST_EMAIL_ADDRESS = '%s@testing.blender.org' % TEST_EMAIL_USER
@ -7,7 +8,15 @@ TEST_EMAIL_ADDRESS = '%s@testing.blender.org' % TEST_EMAIL_USER
os.environ['MONGO_DBNAME'] = 'unittest' os.environ['MONGO_DBNAME'] = 'unittest'
os.environ['EVE_SETTINGS'] = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'settings.py') os.environ['EVE_SETTINGS'] = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'settings.py')
from application import app from application import app
from application.utils.authentication import make_unique_username, validate_token
def make_header(username, password=''):
"""Returns a Basic HTTP Authentication header value."""
return 'basic ' + base64.b64encode('%s:%s' % (username, password))
class FlaskrTestCase(unittest.TestCase): class FlaskrTestCase(unittest.TestCase):
@ -18,7 +27,6 @@ class FlaskrTestCase(unittest.TestCase):
pass pass
def test_make_unique_username(self): def test_make_unique_username(self):
from application.utils.authentication import make_unique_username
with app.test_request_context(): with app.test_request_context():
# Delete the user we want to test for # Delete the user we want to test for
@ -48,3 +56,11 @@ class FlaskrTestCase(unittest.TestCase):
self.assertEqual('%s1' % TEST_EMAIL_USER, make_unique_username(TEST_EMAIL_ADDRESS)) self.assertEqual('%s1' % TEST_EMAIL_USER, make_unique_username(TEST_EMAIL_ADDRESS))
finally: finally:
users.delete_many({'username': TEST_EMAIL_USER}) users.delete_many({'username': TEST_EMAIL_USER})
def test_validate_token__not_logged_in(self):
with app.test_request_context():
self.assertFalse(validate_token())
def test_validate_token__unknown_token(self):
with app.test_request_context(headers={'Authorization': make_header('unknowntoken')}):
self.assertFalse(validate_token())