92 lines
2.7 KiB
Python
92 lines
2.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
try:
|
||
|
|
from unittest.mock import patch
|
||
|
|
except ImportError:
|
||
|
|
from mock import patch
|
||
|
|
|
||
|
|
from django.contrib import messages
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
from django.contrib.messages.api import get_messages
|
||
|
|
from django.contrib.messages.storage.base import Message
|
||
|
|
from django.test import TestCase, override_settings
|
||
|
|
|
||
|
|
from .cas_clients import AcceptCASClient
|
||
|
|
|
||
|
|
User = get_user_model()
|
||
|
|
|
||
|
|
|
||
|
|
@patch('allauth_cas.views.cas.CASClient', AcceptCASClient)
|
||
|
|
def client_cas_login(client):
|
||
|
|
"""
|
||
|
|
Sign in client through the example CAS provider.
|
||
|
|
Returns the response of callbacK view.
|
||
|
|
"""
|
||
|
|
r = client.get('/accounts/theid/login/callback/', {
|
||
|
|
'ticket': 'fake-ticket',
|
||
|
|
})
|
||
|
|
return r
|
||
|
|
|
||
|
|
|
||
|
|
class LogoutFlowTests(TestCase):
|
||
|
|
expected_msg_str = (
|
||
|
|
"To logout of CAS, please close your browser, or visit this <a "
|
||
|
|
"href=\"/accounts/theid/logout/?next=%2F\">link</a>."
|
||
|
|
)
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
client_cas_login(self.client)
|
||
|
|
|
||
|
|
def assertCASLogoutNotInMessages(self, response):
|
||
|
|
r_messages = get_messages(response.wsgi_request)
|
||
|
|
self.assertNotIn(
|
||
|
|
self.expected_msg_str,
|
||
|
|
(str(msg) for msg in r_messages),
|
||
|
|
)
|
||
|
|
self.assertTemplateNotUsed(
|
||
|
|
response,
|
||
|
|
'cas_account/messages/logged_out.txt',
|
||
|
|
)
|
||
|
|
|
||
|
|
@override_settings(SOCIALACCOUNT_PROVIDERS={
|
||
|
|
'theid': {
|
||
|
|
'MESSAGE_ON_LOGOUT': True,
|
||
|
|
'MESSAGE_ON_LOGOUT_LEVEL': messages.WARNING,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
def test_message_on_logout(self):
|
||
|
|
"""
|
||
|
|
Message is sent to propose user to logout of CAS.
|
||
|
|
"""
|
||
|
|
r = self.client.post('/accounts/logout/')
|
||
|
|
r_messages = get_messages(r.wsgi_request)
|
||
|
|
|
||
|
|
expected_msg = Message(messages.WARNING, self.expected_msg_str)
|
||
|
|
|
||
|
|
self.assertIn(expected_msg, r_messages)
|
||
|
|
self.assertTemplateUsed(r, 'cas_account/messages/logged_out.txt')
|
||
|
|
|
||
|
|
@override_settings(SOCIALACCOUNT_PROVIDERS={
|
||
|
|
'theid': {
|
||
|
|
'MESSAGE_ON_LOGOUT': False,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
def test_message_on_logout_disabled(self):
|
||
|
|
"""
|
||
|
|
The logout message can be disabled in settings.
|
||
|
|
"""
|
||
|
|
r = self.client.post('/accounts/logout/')
|
||
|
|
self.assertCASLogoutNotInMessages(r)
|
||
|
|
|
||
|
|
@override_settings(SOCIALACCOUNT_PROVIDERS={
|
||
|
|
'theid': {'MESSAGE_ON_LOGOUT': True},
|
||
|
|
})
|
||
|
|
def test_default_logout(self):
|
||
|
|
"""
|
||
|
|
The CAS logout message doesn't appear with other login methods.
|
||
|
|
"""
|
||
|
|
User.objects.create_user('user', '', 'user')
|
||
|
|
self.client.login(username='user', password='user')
|
||
|
|
|
||
|
|
r = self.client.post('/accounts/logout/')
|
||
|
|
self.assertCASLogoutNotInMessages(r)
|