Revision 8f377cd6 aai/tests.py
| b/aai/tests.py | ||
|---|---|---|
| 1 | 1 |
# |
| 2 |
# Unit Tests for api
|
|
| 2 |
# Unit Tests for aai
|
|
| 3 | 3 |
# |
| 4 |
# Provides automated tests for api module
|
|
| 4 |
# Provides automated tests for aai module. The tests
|
|
| 5 | 5 |
# |
| 6 | 6 |
# Copyright 2011 Greek Research and Technology Network |
| 7 | 7 |
# |
| ... | ... | |
| 10 | 10 |
from django.test.client import Client |
| 11 | 11 |
from django.conf import settings |
| 12 | 12 |
|
| 13 |
from synnefo.logic.shibboleth import Tokens, NoUniqueToken
|
|
| 13 |
from synnefo.aai.shibboleth import Tokens, NoUniqueToken
|
|
| 14 | 14 |
from synnefo.db.models import SynnefoUser |
| 15 | 15 |
|
| 16 | 16 |
from datetime import datetime, timedelta |
| ... | ... | |
| 28 | 28 |
response = self.client.get(self.apibase + '/servers', {},
|
| 29 | 29 |
**{Tokens.SIB_GIVEN_NAME: 'Jimmy',
|
| 30 | 30 |
Tokens.SIB_EDU_PERSON_PRINCIPAL_NAME: 'jh@gmail.com', |
| 31 |
Tokens.SIB_DISPLAY_NAME: 'Jimmy Hendrix'}) |
|
| 31 |
Tokens.SIB_DISPLAY_NAME: 'Jimmy Hendrix', |
|
| 32 |
'TEST-AAI' : 'true'}) |
|
| 32 | 33 |
user = None |
| 33 | 34 |
try: |
| 34 | 35 |
user = SynnefoUser.objects.get(uniq = "jh@gmail.com") |
| ... | ... | |
| 45 | 46 |
""" |
| 46 | 47 |
response = self.client.get(self.apibase + '/servers', {},
|
| 47 | 48 |
**{Tokens.SIB_GIVEN_NAME: 'Jimmy',
|
| 48 |
Tokens.SIB_DISPLAY_NAME: 'Jimmy Hendrix'}) |
|
| 49 |
Tokens.SIB_DISPLAY_NAME: 'Jimmy Hendrix', |
|
| 50 |
'TEST-AAI' : 'true'}) |
|
| 49 | 51 |
self._test_redirect(response) |
| 50 | 52 |
|
| 51 | 53 |
def test_shibboleth_wrong_from_request(self): |
| ... | ... | |
| 56 | 58 |
Tokens.SIB_EDU_PERSON_PRINCIPAL_NAME: 'jh@gmail.com', |
| 57 | 59 |
Tokens.SIB_DISPLAY_NAME: 'Jimmy Hendrix', |
| 58 | 60 |
'REMOTE_ADDR': '1.2.3.4', |
| 59 |
'SERVER_NAME': 'nohost.nodomain'}) |
|
| 61 |
'SERVER_NAME': 'nohost.nodomain', |
|
| 62 |
'TEST-AAI' : 'true'}) |
|
| 60 | 63 |
self._test_redirect(response) |
| 61 | 64 |
|
| 62 | 65 |
def test_shibboleth_expired_token(self): |
| ... | ... | |
| 66 | 69 |
self.assertNotEqual(user.auth_token_created, None) |
| 67 | 70 |
self._update_user_ts(user) |
| 68 | 71 |
response = self.client.get(self.apibase + '/servers', {},
|
| 69 |
**{'X-Auth-Token': user.auth_token})
|
|
| 72 |
**{'X-Auth-Token': user.auth_token,
|
|
| 73 |
'TEST-AAI' : 'true'}) |
|
| 70 | 74 |
self._test_redirect(response) |
| 71 | 75 |
|
| 72 | 76 |
def test_shibboleth_redirect(self): |
| 73 | 77 |
""" test redirect to Sibboleth page |
| 74 | 78 |
""" |
| 75 |
response = self.client.get(self.apibase + '/servers') |
|
| 79 |
response = self.client.get(self.apibase + '/servers', {}, **{'TEST-AAI' : 'true'})
|
|
| 76 | 80 |
self._test_redirect(response) |
| 77 | 81 |
|
| 78 | 82 |
def test_shibboleth_auth(self): |
| ... | ... | |
| 80 | 84 |
""" |
| 81 | 85 |
user = SynnefoUser.objects.get(uniq = "test@synnefo.gr") |
| 82 | 86 |
response = self.client.get(self.apibase + '/servers', {},
|
| 83 |
**{'X-Auth-Token': user.auth_token})
|
|
| 87 |
**{'X-Auth-Token': user.auth_token,
|
|
| 88 |
'TEST-AAI' : 'true'}) |
|
| 84 | 89 |
self.assertTrue(response.status_code, 200) |
| 85 | 90 |
self.assertTrue('Vary' in response)
|
| 86 | 91 |
self.assertTrue('X-Auth-Token' in response['Vary'])
|
| ... | ... | |
| 90 | 95 |
""" |
| 91 | 96 |
response = self.client.get(self.apibase + '/servers', {},
|
| 92 | 97 |
**{'X-Auth-User': 'notme',
|
| 93 |
'X-Auth-Key': '0xdeadbabe'}) |
|
| 98 |
'X-Auth-Key': '0xdeadbabe', |
|
| 99 |
'TEST-AAI' : 'true'}) |
|
| 94 | 100 |
self.assertEquals(response.status_code, 401) |
| 95 | 101 |
|
| 96 | 102 |
def test_oapi_auth(self): |
| ... | ... | |
| 98 | 104 |
""" |
| 99 | 105 |
response = self.client.get(self.apibase + '/', {},
|
| 100 | 106 |
**{'X-Auth-User': 'testdbuser',
|
| 101 |
'X-Auth-Key': 'test@synnefo.gr'}) |
|
| 107 |
'X-Auth-Key': 'test@synnefo.gr', |
|
| 108 |
'TEST-AAI' : 'true'}) |
|
| 102 | 109 |
self.assertEquals(response.status_code, 204) |
| 103 | 110 |
self.assertNotEqual(response['X-Auth-Token'], None) |
| 104 | 111 |
self.assertEquals(response['X-Server-Management-Url'], '') |
| ... | ... | |
| 114 | 121 |
user.auth_token_created = (datetime.now() - |
| 115 | 122 |
timedelta(hours = settings.AUTH_TOKEN_DURATION)) |
| 116 | 123 |
user.save() |
| 124 |
|
|
| 125 |
|
|
Also available in: Unified diff