Python UserFactory.create方法代码示例

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_put_invalid_token(self):
        user = UserFactory.create()
        other_user = UserFactory.create()
        token = default_token_generator.make_token(other_user)
        uid = urlsafe_base64_encode(force_bytes(user.pk))

        request = self.create_request('put', auth=False)
        view = self.view_class.as_view()
        response = view(request, uidb64=uid, token=token)
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_post(self):
        """Assert user can sign in"""
        UserFactory.create(email=self.username, password=self.password)

        request = self.create_request('post', auth=False, data=self.data)
        view = self.view_class.as_view()
        response = view(request)
        self.assertEqual(
            response.status_code, status.HTTP_200_OK, msg=response.data)

        # Ensure user has a token now
        token = self.model.objects.get()
        self.assertEqual(response.data['token'], token.key)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
 def test_deserialize_update_email_in_use(self):
     user = UserFactory.create()
     other_user = UserFactory.create()
     data = {
         'name': "Robert'); DROP TABLE Students;--'",
         'email': other_user.email,
         'password': 'Sup3RSecre7paSSw0rD',
         'password2': 'Sup3RSecre7paSSw0rD',
     }
     serializer = serializers.RegistrationSerializer(user, data=data)
     self.assertFalse(serializer.is_valid())
     self.assertEqual(
         serializer._errors['email'],
         ['That email address has already been registered.'])

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_post_rate_limit(self):
        """Ensure the POST requests are rate limited."""
        email = '[email protected]'
        UserFactory.create(email=email)

        data = {'email': email}
        request = self.create_request('post', data=data, auth=False)
        view = self.view_class.as_view()

        response = view(request)
        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)

        # request is throttled
        response = view(request)
        self.assertEqual(response.status_code, status.HTTP_429_TOO_MANY_REQUESTS)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_post_user_not_confirmed(self):
        """Assert non active users can not log in."""
        UserFactory.create(email=self.username, password=self.password, is_active=False)

        request = self.create_request('post', auth=False, data=self.data)
        view = self.view_class.as_view()
        response = view(request)
        self.assertEqual(
            response.status_code,
            status.HTTP_400_BAD_REQUEST,
            msg=response.data,
        )
        expected = 'User account is disabled.'
        self.assertIn(expected, response.data['non_field_errors'])
        self.assertNotIn('token', response.data)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_post(self):
        username = '[email protected]'
        password = 'myepicstrongpassword'
        UserFactory.create(email=username.lower(), password=password)

        data = {'username': username, 'password': password}
        request = self.create_request('post', auth=False, data=data)
        view = self.view_class.as_view()
        response = view(request)
        self.assertEqual(
            response.status_code, status.HTTP_200_OK, msg=response.data)

        # Ensure user has a token now
        token = self.model.objects.get(user__email=username.lower())
        self.assertEqual(response.data['token'], token.key)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_post_different_user_logged_in(self):
        """Assert a user can have amny accounts and verify one."""
        user = UserFactory.create()
        other_user = UserFactory.create()
        token = user.generate_validation_token()

        request = self.create_request('post', user=other_user)
        view = self.view_class.as_view()
        response = view(request, token=token)
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

        updated_user = User.objects.get(pk=user.pk)
        self.assertTrue(updated_user.email_verified)

        logged_in_user = User.objects.get(pk=other_user.pk)
        self.assertFalse(logged_in_user.email_verified)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_duplicate_email(self):
        """Emails should be unique regardless of case."""
        # First create a user with the same email.
        self.data.pop('password2')
        UserFactory.create(**self.data)

        # Just for kicks, lets try changing the email case.
        self.data['email'] = self.data['email'].upper()

        request = self.create_request('post', auth=False, data=self.data)
        response = self.view_class.as_view()(request)
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)

        self.assertTrue('email' in response.data, msg=response.data)

        self.assertEqual(User.objects.count(), 1)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_delete_without_avatar(self):
        user = UserFactory.create()
        request = self.create_request('delete', user=user)
        view = self.view_class.as_view()
        response = view(request)

        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_existent_email(self):
        email = '[email protected]'
        user = UserFactory.create(email=email)
        context = {}
        site = Site.objects.get_current()

        request = self.create_request(
            'post',
            data={'email': email},
            auth=False,
        )
        view = self.view_class.as_view()
        with patch(EMAIL_CONTEXT) as get_context:
            get_context.return_value = context
            with patch(SEND_METHOD) as send_email:
                response = view(request)
        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)

        expected = {
            'to': user.email,
            'template_name': 'user_management/password_reset_email.txt',
            'html_template_name': 'user_management/password_reset_email.html',
            'subject': '{} password reset'.format(site.domain),
            'context': context,
            'headers': {},
        }
        send_email.assert_called_once_with(**expected)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
 def test_post_email_already_verified(self):
     """Assert email already verified does not trigger another email."""
     user = UserFactory.create(email_verified=True)
     request = self.create_request('post', auth=False, data={'email': user.email})
     view = self.view_class.as_view()
     response = view(request)
     self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
     self.assertIn('email', response.data)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_options(self):
        user = UserFactory.create()
        token = default_token_generator.make_token(user)
        uid = urlsafe_base64_encode(force_bytes(user.pk))

        request = self.create_request('options', auth=False)
        view = self.view_class.as_view()
        response = view(request, uidb64=uid, token=token)
        self.assertEqual(response.status_code, status.HTTP_200_OK)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_post_verified_email(self):
        """Assert verified user cannot verify email."""
        user = UserFactory.create(email_verified=True)
        token = user.generate_validation_token()

        request = self.create_request('post')
        view = self.view_class.as_view()
        response = view(request, token=token)
        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_post_expired_token(self):
        """Assert token expires."""
        user = UserFactory.create()
        token = user.generate_validation_token()
        request = self.create_request('post', auth=False)
        view = self.view_class.as_view()

        response = view(request, token=token)

        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)

# 需要导入模块: from user_management.models.tests.factories import UserFactory [as 别名]
# 或者: from user_management.models.tests.factories.UserFactory import create [as 别名]
    def test_get(self):
        user = UserFactory.create()

        request = self.create_request(user=user)
        view = self.view_class.as_view()
        response = view(request)
        self.assertEqual(response.status_code, status.HTTP_200_OK)

        expected = [self.expected_data(user)]
        self.assertEqual(response.data, expected)