Revision e892db2d
| b/snf-cyclades-app/synnefo/helpdesk/tests.py | ||
|---|---|---|
| 78 | 78 |
return super(AuthClient, self).request(**request) |
| 79 | 79 |
|
| 80 | 80 |
|
| 81 |
def get_user_mock(request, *args, **kwargs): |
|
| 82 |
request.user_uniq = None |
|
| 83 |
request.user = None |
|
| 84 |
if request.META.get('HTTP_X_AUTH_TOKEN', None) == '0000':
|
|
| 85 |
request.user_uniq = 'test' |
|
| 86 |
request.user = {'uniq': 'test', 'auth_token': '0000'}
|
|
| 87 |
if request.META.get('HTTP_X_AUTH_TOKEN', None) == '0001':
|
|
| 88 |
request.user_uniq = 'test' |
|
| 89 |
request.user = {'uniq': 'test', 'groups': ['default',
|
|
| 90 |
'helpdesk'], |
|
| 91 |
'auth_token': '0001'} |
|
| 92 |
|
|
| 93 |
|
|
| 94 |
@mock.patch("astakosclient.AstakosClient", new=AstakosClientMock)
|
|
| 95 |
@mock.patch("snf_django.lib.astakos.get_user", new=get_user_mock)
|
|
| 81 | 96 |
class HelpdeskTests(TestCase): |
| 82 | 97 |
""" |
| 83 | 98 |
Helpdesk tests. Test correctness of permissions and returned data. |
| ... | ... | |
| 86 | 101 |
fixtures = ['helpdesk_test'] |
| 87 | 102 |
|
| 88 | 103 |
def setUp(self): |
| 89 |
|
|
| 90 |
def get_user_mock(request, *args, **kwargs): |
|
| 91 |
request.user_uniq = None |
|
| 92 |
request.user = None |
|
| 93 |
if request.META.get('HTTP_X_AUTH_TOKEN', None) == '0000':
|
|
| 94 |
request.user_uniq = 'test' |
|
| 95 |
request.user = {'uniq': 'test', 'auth_token': '0000'}
|
|
| 96 |
if request.META.get('HTTP_X_AUTH_TOKEN', None) == '0001':
|
|
| 97 |
request.user_uniq = 'test' |
|
| 98 |
request.user = {'uniq': 'test', 'groups': ['default',
|
|
| 99 |
'helpdesk'], |
|
| 100 |
'auth_token': '0001'} |
|
| 101 |
|
|
| 102 |
# mock the astakos authentication function |
|
| 103 |
from snf_django.lib import astakos |
|
| 104 |
astakos.get_user = get_user_mock |
|
| 105 |
import astakosclient |
|
| 106 |
astakosclient.AstakosClient = AstakosClientMock |
|
| 107 |
|
|
| 108 | 104 |
settings.SKIP_SSH_VALIDATION = True |
| 109 | 105 |
settings.HELPDESK_ENABLED = True |
| 110 | 106 |
self.client = AuthClient() |
Also available in: Unified diff