DGNum/flask-cas-ng
5
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test fixes for python 2.6

Browse source
This commit is contained in:
Cameron Brandon White 2015-01-29 17:03:11 -08:00
parent 2616741ed8
commit 052566c81c
1 changed files with 10 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/test_routing.py
View file

@ -67,18 +67,16 @@ class test_routing(unittest.TestCase):
with client.session_transaction() as s:
s[self.app.config['CAS_TOKEN_SESSION_KEY']] = ticket
client.get('/login/')
self.assertNotIn(
self.app.config['CAS_USERNAME_SESSION_KEY'],
flask.session)
self.assertNotIn(
self.app.config['CAS_TOKEN_SESSION_KEY'],
flask.session)
self.assertTrue(
self.app.config['CAS_USERNAME_SESSION_KEY'] not in flask.session)
self.assertTrue(
self.app.config['CAS_TOKEN_SESSION_KEY'] not in flask.session)
@mock.patch.object(routing, 'validate', return_value=True)
def test_login_by_cas_valid(self, m):
with self.app.test_client() as client:
ticket = '12345-abcdefg-cas'
response = client.get('/login/?ticket={}'.format(ticket))
response = client.get('/login/?ticket={0}'.format(ticket))
self.assertEqual(response.status_code, 302)
self.assertEqual(
response.headers['Location'],
@ -91,7 +89,7 @@ class test_routing(unittest.TestCase):
def test_login_by_cas_invalid(self, m):
with self.app.test_client() as client:
ticket = '12345-abcdefg-cas'
response = client.get('/login/?ticket={}'.format(ticket))
response = client.get('/login/?ticket={0}'.format(ticket))
self.assertEqual(response.status_code, 302)
self.assertEqual(
response.headers['Location'],
@ -121,9 +119,7 @@ class test_routing(unittest.TestCase):
with self.app.test_request_context('/login/'):
ticket = '12345-abcdefg-cas'
self.assertEqual(routing.validate(ticket), False)
self.assertNotIn(
self.app.config['CAS_USERNAME_SESSION_KEY'],
flask.session)
self.assertNotIn(
self.app.config['CAS_TOKEN_SESSION_KEY'],
flask.session)
self.assertTrue(
self.app.config['CAS_USERNAME_SESSION_KEY'] not in flask.session)
self.assertTrue(
self.app.config['CAS_TOKEN_SESSION_KEY'] not in flask.session)