Revision 2cd99e7a aai/tests.py
| b/aai/tests.py | ||
|---|---|---|
| 5 | 5 |
# |
| 6 | 6 |
# Copyright 2011 Greek Research and Technology Network |
| 7 | 7 |
# |
| 8 |
from Cookie import Cookie |
|
| 9 | 8 |
|
| 10 | 9 |
from django.test import TestCase |
| 11 | 10 |
from django.test.client import Client |
| ... | ... | |
| 15 | 14 |
|
| 16 | 15 |
from datetime import datetime, timedelta |
| 17 | 16 |
|
| 18 |
from synnefo.aai.shibboleth import Tokens, NoUniqueToken |
|
| 17 |
from synnefo.aai.shibboleth import Tokens |
|
| 18 |
|
|
| 19 | 19 |
|
| 20 | 20 |
class AaiTestCase(TestCase): |
| 21 | 21 |
fixtures = ['api_test_data', 'auth_test_data'] |
| ... | ... | |
| 45 | 45 |
self.assertEquals(response['X-Auth-Token'], user.auth_token) |
| 46 | 46 |
#self.assertNotEquals(response.cookies['X-Auth-Token'].find(user.auth_token), -1) |
| 47 | 47 |
|
| 48 |
|
|
| 49 | 48 |
def test_shibboleth_no_uniq_request(self): |
| 50 | 49 |
"""test a request with no unique field |
| 51 | 50 |
""" |
| ... | ... | |
| 55 | 54 |
'TEST-AAI': 'true'}) |
| 56 | 55 |
self._test_redirect(response) |
| 57 | 56 |
|
| 58 |
|
|
| 59 | 57 |
def test_shibboleth_expired_token(self): |
| 60 | 58 |
""" test request from expired token |
| 61 | 59 |
""" |
| 62 | 60 |
user = SynnefoUser.objects.get(uniq="test@synnefo.gr") |
| 63 | 61 |
self.assertNotEqual(user.auth_token_created, None) |
| 64 |
self._update_user_ts(user)
|
|
| 62 |
_update_user_ts(user) |
|
| 65 | 63 |
response = self.client.get('/index.html', {},
|
| 66 | 64 |
**{'X-Auth-Token': user.auth_token,
|
| 67 | 65 |
'TEST-AAI': 'true'}) |
| ... | ... | |
| 89 | 87 |
self.assertTrue('Location' in response)
|
| 90 | 88 |
self.assertTrue(response['Location'].endswith(settings.LOGIN_PATH)) |
| 91 | 89 |
|
| 92 |
def _update_user_ts(self, user): |
|
| 93 |
user.auth_token_created = (datetime.now() - |
|
| 94 |
timedelta(hours = settings.AUTH_TOKEN_DURATION)) |
|
| 95 |
user.save() |
|
| 90 |
def _update_user_ts(user): |
|
| 91 |
user.auth_token_created = (datetime.now() - |
|
| 92 |
timedelta(hours = settings.AUTH_TOKEN_DURATION)) |
|
| 93 |
user.save() |
|
Also available in: Unified diff