blob: 91c2808c17d03f3392dd20a50ac7f9cb709cd92d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
from unittest.mock import patch
from allauth.exceptions import ImmediateHttpResponse
from allauth.socialaccount.models import SocialAccount, SocialLogin
from django.contrib.auth.models import User
from django.contrib.messages.storage.base import BaseStorage
from django.http import HttpRequest
from django.test import TestCase
from pydis_site.utils.account import AccountAdapter, SocialAccountAdapter
class AccountUtilsTests(TestCase):
def setUp(self):
self.django_user = User.objects.create(username="user")
self.discord_account = SocialAccount.objects.create(
user=self.django_user, provider="discord", uid=0
)
self.github_account = SocialAccount.objects.create(
user=self.django_user, provider="github", uid=0
)
def test_account_adapter(self):
"""Test that our Allauth account adapter functions correctly."""
adapter = AccountAdapter()
self.assertFalse(adapter.is_open_for_signup(HttpRequest()))
def test_social_account_adapter(self):
"""Test that our Allauth social account adapter functions correctly."""
adapter = SocialAccountAdapter()
discord_login = SocialLogin(account=self.discord_account)
github_login = SocialLogin(account=self.github_account)
messages_request = HttpRequest()
messages_request._messages = BaseStorage(messages_request)
with patch("pydis_site.utils.account.reverse") as mock_reverse:
with patch("pydis_site.utils.account.redirect") as mock_redirect:
with self.assertRaises(ImmediateHttpResponse):
adapter.is_open_for_signup(messages_request, github_login)
self.assertEqual(len(messages_request._messages._queued_messages), 1)
self.assertEqual(mock_redirect.call_count, 1)
self.assertEqual(mock_reverse.call_count, 1)
self.assertTrue(adapter.is_open_for_signup(HttpRequest(), discord_login))
|
