66 lines
2 KiB
Python
66 lines
2 KiB
Python
from django.contrib.auth import get_user_model
|
|
|
|
from allauth_cas.test.testcases import CASTestCase, CASViewTestCase
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class ClipperProviderTests(CASTestCase):
|
|
|
|
def setUp(self):
|
|
self.u = User.objects.create_user('user', 'user@mail.net', 'user')
|
|
|
|
def test_auto_signup(self):
|
|
self.client_cas_login(
|
|
self.client, provider_id='clipper', username='clipper_uid')
|
|
|
|
u = User.objects.get(username='clipper_uid')
|
|
self.assertEqual(u.email, 'clipper_uid@clipper.ens.fr')
|
|
|
|
def test_extra_data_keeps_ldap_data(self):
|
|
clipper_conn = self.u.socialaccount_set.create(
|
|
uid='user', provider='clipper',
|
|
extra_data={'ldap': {'aa': 'bb'}},
|
|
)
|
|
|
|
self.client_cas_login(
|
|
self.client, provider_id='clipper', username='user')
|
|
|
|
clipper_conn.refresh_from_db()
|
|
self.assertEqual(clipper_conn.extra_data['ldap'], {'aa': 'bb'})
|
|
|
|
|
|
class ClipperViewsTests(CASViewTestCase):
|
|
|
|
def test_login_view(self):
|
|
r = self.client.get('/accounts/clipper/login/')
|
|
expected = (
|
|
"https://cas.eleves.ens.fr/login?service=http%3A%2F%2Ftestserver"
|
|
"%2Faccounts%2Fclipper%2Flogin%2Fcallback%2F"
|
|
)
|
|
self.assertRedirects(
|
|
r, expected,
|
|
fetch_redirect_response=False,
|
|
)
|
|
|
|
def test_callback_view(self):
|
|
# Required to initialize a SocialLogin.
|
|
r = self.client.get('/accounts/clipper/login/')
|
|
|
|
# Tests.
|
|
self.patch_cas_response(valid_ticket='__all__')
|
|
r = self.client.get('/accounts/clipper/login/callback/', {
|
|
'ticket': '123456',
|
|
})
|
|
self.assertLoginSuccess(r)
|
|
|
|
def test_logout_view(self):
|
|
r = self.client.get('/accounts/clipper/logout/')
|
|
expected = (
|
|
"https://cas.eleves.ens.fr/logout?service=http%3A%2F%2Ftestserver"
|
|
"%2F"
|
|
)
|
|
self.assertRedirects(
|
|
r, expected,
|
|
fetch_redirect_response=False,
|
|
)
|