sartography/cr-connect-workflow
1
0
Fork 0
mirror of https://github.com/sartography/cr-connect-workflow.git synced 2025-02-23 05:08:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updates tests to match changing sso_backdoor to query string params

Browse Source
This commit is contained in:
Aaron Louie 2020-02-21 12:03:14 -05:00
parent e4efc5e4e2
commit 1ac9180304
1 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/test_authentication.py
View File

@ -1,21 +1,25 @@
import urllib.parse
from crc import db from crc import db
from crc.models.user import UserModel from crc.models.user import UserModel
from tests.base_test import BaseTest from tests.base_test import BaseTest
class TestAuthentication(BaseTest): class TestAuthentication(BaseTest):
test_uid = "dhf8r" test_uid = "dhf8r"
def logged_in_headers(self, user=None): def logged_in_headers(self, user=None):
if user is None: if user is None:
uid = self.test_uid uid = self.test_uid
headers = {'uid': self.test_uid, 'first_name': 'Daniel', 'last_name': 'Funk', 'email_address': 'dhf8r@virginia.edu'} user_info = {'uid': self.test_uid, 'first_name': 'Daniel', 'last_name': 'Funk',
'email_address': 'dhf8r@virginia.edu'}
else: else:
uid = user.uid uid = user.uid
headers = {'uid': user.uid, 'first_name': user.first_name, 'last_name': user.last_name, 'email_address': user.email_address} user_info = {'uid': user.uid, 'first_name': user.first_name, 'last_name': user.last_name,
'email_address': user.email_address}
rv = self.app.get("/v1.0/sso_backdoor", headers=headers, follow_redirects=True, query_string = self.user_info_to_query_string(user_info)
rv = self.app.get("/v1.0/sso_backdoor%s" % query_string, follow_redirects=True,
content_type="application/json") content_type="application/json")
user_model = UserModel.query.filter_by(uid=uid).first() user_model = UserModel.query.filter_by(uid=uid).first()
self.assertIsNotNone(user_model.display_name) self.assertIsNotNone(user_model.display_name)
@ -33,18 +37,20 @@ class TestAuthentication(BaseTest):
user = db.session.query(UserModel).filter(UserModel.uid == self.test_uid).first() user = db.session.query(UserModel).filter(UserModel.uid == self.test_uid).first()
self.assertIsNone(user) self.assertIsNone(user)
headers = {'uid': self.test_uid, 'first_name': 'Daniel', 'email_address': 'dhf8r@virginia.edu'} user_info = {'uid': self.test_uid, 'first_name': 'Daniel', 'last_name': 'Funk',
rv_1 = self.app.get("/v1.0/sso_backdoor", headers=headers, follow_redirects=False) 'email_address': 'dhf8r@virginia.edu'}
query_string = self.user_info_to_query_string(user_info)
url = '/v1.0/sso_backdoor%s' % query_string
rv_1 = self.app.get(url, follow_redirects=False)
self.assertTrue(rv_1.status_code == 302) self.assertTrue(rv_1.status_code == 302)
user = db.session.query(UserModel).filter(UserModel.uid == self.test_uid).first() user = db.session.query(UserModel).filter(UserModel.uid == self.test_uid).first()
self.assertIsNotNone(user) self.assertIsNotNone(user)
self.assertIsNotNone(user.display_name) self.assertIsNotNone(user.display_name)
self.assertIsNotNone(user.email_address) self.assertIsNotNone(user.email_address)
# Hitting the same endpoint again with the same info should not cause an error # Hitting the same endpoint again with the same info should not cause an error
rv_2 = self.app.get("/v1.0/sso_backdoor", headers=headers, follow_redirects=False) rv_2 = self.app.get(url, follow_redirects=False)
self.assertTrue(rv_1.status_code == 302) self.assertTrue(rv_1.status_code == 302)
def test_current_user_status(self): def test_current_user_status(self):
@ -58,3 +64,13 @@ class TestAuthentication(BaseTest):
user = UserModel(uid="ajl2j", first_name='Aaron', last_name='Louie', email_address='ajl2j@virginia.edu') user = UserModel(uid="ajl2j", first_name='Aaron', last_name='Louie', email_address='ajl2j@virginia.edu')
rv = self.app.get('/v1.0/user', headers=self.logged_in_headers(user)) rv = self.app.get('/v1.0/user', headers=self.logged_in_headers(user))
self.assert_success(rv) self.assert_success(rv)
def user_info_to_query_string(self, user_info):
query_string_list = []
items = user_info.items()
for key, value in items:
query_string_list.append('%s=%s' % (key, urllib.parse.quote(value)))
return '?%s' % '&'.join(query_string_list)