Django框架的社交网络网站.zip

from __future__ import absolute_import import json import uuid from datetime import timedelta import django from django import forms from django.conf import settings from django.contrib.auth.models import AbstractUser, AnonymousUser from django.contrib.sites.models import Site from django.core import mail, validators from django.core.exceptions import ValidationError from django.db import models from django.test.client import Client, RequestFactory from django.test.utils import override_settings from django.utils.timezone import now from allauth.account.forms import BaseSignupForm, SignupForm from allauth.account.models import ( EmailAddress, EmailConfirmation, EmailConfirmationHMAC, ) from allauth.tests import Mock, TestCase, patch from allauth.utils import get_user_model, get_username_max_length from . import app_settings from ..compat import is_authenticated, reverse from .adapter import get_adapter from .auth_backends import AuthenticationBackend from .signals import user_logged_out from .utils import ( filter_users_by_username, url_str_to_user_pk, user_pk_to_url_str, user_username, ) test_username_validators = [ validators.RegexValidator( regex=r'^[a-c]+$', message='not abc', flags=0)] @override_settings( ACCOUNT_DEFAULT_HTTP_PROTOCOL='https', ACCOUNT_EMAIL_VERIFICATION=app_settings.EmailVerificationMethod.MANDATORY, ACCOUNT_AUTHENTICATION_METHOD=app_settings.AuthenticationMethod.USERNAME, ACCOUNT_SIGNUP_FORM_CLASS=None, ACCOUNT_EMAIL_SUBJECT_PREFIX=None, LOGIN_REDIRECT_URL='/accounts/profile/', ACCOUNT_ADAPTER='allauth.account.adapter.DefaultAccountAdapter', ACCOUNT_USERNAME_REQUIRED=True) class AccountTests(TestCase): def setUp(self): if 'allauth.socialaccount' in settings.INSTALLED_APPS: # Otherwise ImproperlyConfigured exceptions may occur from ..socialaccount.models import SocialApp sa = SocialApp.objects.create(name='testfb', provider='facebook') sa.sites.add(Site.objects.get_current()) @override_settings( ACCOUNT_AUTHENTICATION_METHOD=app_settings.AuthenticationMethod .USERNAME_EMAIL) def test_username_containing_at(self): user = get_user_model().objects.create(username='@raymond.penners') user.set_password('psst') user.save() EmailAddress.objects.create(user=user, email="raymond.penners@example.com", primary=True, verified=True) resp = self.client.post(reverse('account_login'), {'login': '@raymond.penners', 'password': 'psst'}) self.assertRedirects(resp, settings.LOGIN_REDIRECT_URL, fetch_redirect_response=False) def test_signup_same_email_verified_externally(self): user = self._test_signup_email_verified_externally("john@example.com", "john@example.com") self.assertEqual(EmailAddress.objects.filter(user=user).count(), 1) EmailAddress.objects.get(verified=True, email="john@example.com", user=user, primary=True) def test_signup_other_email_verified_externally(self): """ John is invited on john@example.org, but signs up via john@example.com. E-mail verification is by-passed, their home e-mail address is used as a secondary. """ user = self._test_signup_email_verified_externally("john@example.com", "john@example.org") self.assertEqual(EmailAddress.objects.filter(user=user).count(), 2) EmailAddress.objects.get(verified=False, email="john@example.com", user=user, primary=False) EmailAddress.objects.get(verified=True, email="john@example.org", user=user, primary=True) def _test_signup_email_verified_externally(self, signup_email, verified_email): username = 'johndoe' request = RequestFactory().post(reverse('account_signup'), {'username': username, 'email': signup_email, 'password1': 'johndoe', 'password2': 'johndoe'}) # Fake stash_verified_email from django.contrib.messages.middleware import MessageMiddleware from django.contrib.sessions.middleware import SessionMiddleware SessionMiddleware().process_request(request) MessageMiddleware().process_request(request) request.user = AnonymousUser() request.session['account_verified_email'] = verified_email from .views import signup resp = signup(request) self.assertEqual(resp.status_code, 302) self.assertEqual(resp['location'], get_adapter().get_login_redirect_url(request)) self.assertEqual(len(mail.outbox), 0) return get_user_model().objects.get(username=username) @override_settings( ACCOUNT_USERNAME_REQUIRED=True, ACCOUNT_SIGNUP_PASSOWRD_ENTER_TWICE=True) def test_signup_password_twice_form_error(self): resp = self.client.post( reverse('account_signup'), data={ 'username': 'johndoe', 'email': 'john@example.org', 'password1': 'johndoe', 'password2': 'janedoe'}) self.assertFormError( resp, 'form', 'password2', 'You must type the same password each time.' ) @override_settings( ACCOUNT_USERNAME_REQUIRED=True, ACCOUNT_SIGNUP_EMAIL_ENTER_TWICE=True) def test_signup_email_twice(self): request = RequestFactory().post(reverse('account_signup'), {'username': 'johndoe', 'email': 'john@example.org', 'email2': 'john@example.org', 'password1': 'johndoe', 'password2': 'johndoe'}) from django.contrib.messages.middleware import MessageMiddleware from django.contrib.sessions.middleware import SessionMiddleware SessionMiddleware().process_request(request) MessageMiddleware().process_request(request) request.user = AnonymousUser() from .views import signup signup(request) user = get_user_model().objects.get(username='johndoe') self.assertEqual(user.email, "john@example.org") def _create_user(self, username='john', password='doe'): user = get_user_model().objects.create( username=username, is_active=True) if password: user.set_password(password) else: user.set_unusable_password() user.save() return user def _create_user_and_login(self, usable_password=True): password = 'doe' if usable_password else False user = self._create_user(password=password) self.client_force_login(user) return user def test_redirect_when_authenticated(self): self._create_user_and_login() c = self.client resp = c.get(reverse('account_login')) self.assertRedirects( resp, "/accounts