Python django.contrib.messages.storage.base.BaseStorage() Examples

The following are 1 code examples of django.contrib.messages.storage.base.BaseStorage(). You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may also want to check out all available functions/classes of the module django.contrib.messages.storage.base , or try the search function

Example #1

Source File: test_utils_account.py From site with MIT License

5 votes

def test_social_account_adapter_signup(self):
        """Test that our Allauth social account adapter correctly handles signups."""
        adapter = SocialAccountAdapter()

        discord_login = SocialLogin(account=self.discord_account)
        discord_login_role = SocialLogin(account=self.discord_account_role)
        discord_login_two_roles = SocialLogin(account=self.discord_account_two_roles)
        discord_login_not_present = SocialLogin(account=self.discord_account_not_present)

        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)

                with self.assertRaises(ImmediateHttpResponse):
                    adapter.is_open_for_signup(messages_request, discord_login)

                with self.assertRaises(ImmediateHttpResponse):
                    adapter.is_open_for_signup(messages_request, discord_login_role)

                with self.assertRaises(ImmediateHttpResponse):
                    adapter.is_open_for_signup(messages_request, discord_login_not_present)

                self.assertEqual(len(messages_request._messages._queued_messages), 4)
                self.assertEqual(mock_redirect.call_count, 4)
            self.assertEqual(mock_reverse.call_count, 4)

        self.assertTrue(adapter.is_open_for_signup(HttpRequest(), discord_login_two_roles))