diff options
Diffstat (limited to 'pydis_site')
| -rw-r--r-- | pydis_site/apps/home/tests/test_signal_listener.py | 9 | 
1 files changed, 9 insertions, 0 deletions
| diff --git a/pydis_site/apps/home/tests/test_signal_listener.py b/pydis_site/apps/home/tests/test_signal_listener.py index b7400558..c79a0294 100644 --- a/pydis_site/apps/home/tests/test_signal_listener.py +++ b/pydis_site/apps/home/tests/test_signal_listener.py @@ -2,6 +2,7 @@ from unittest import mock  from allauth.account.signals import user_logged_in  from allauth.socialaccount.models import SocialAccount, SocialLogin +from allauth.socialaccount.providers import registry  from allauth.socialaccount.providers.discord.provider import DiscordProvider  from allauth.socialaccount.providers.github.provider import GitHubProvider  from allauth.socialaccount.signals import ( @@ -20,6 +21,14 @@ class SignalListenerTests(TestCase):      @classmethod      def setUpTestData(cls):          """Executed when testing begins.""" +        # This needs to be registered so we can test the role linking logic with a user that +        # doesn't have a Discord account linked, but is logged in somehow with another account +        # type anyway. The logic this is testing was designed so that the system would be +        # robust enough to handle that case, but it's impossible to fully test (and therefore +        # to have coverage of) those lines without an extra provider, and GH was the second +        # provider it was built with in mind. +        registry.register(GitHubProvider) +          cls.admin_role = Role.objects.create(              id=0,              name="admin", | 
