mirror of
https://github.com/Tivix/django-rest-auth.git
synced 2025-02-06 21:20:32 +03:00
Replaced references to contrib.auth.model.User with get_user_model(). Closes #68
This commit is contained in:
parent
d0c5f0a6d9
commit
bef1074052
|
@ -5,7 +5,6 @@ from django.conf import settings
|
|||
from django.core.urlresolvers import reverse
|
||||
from django.test.client import Client, MULTIPART_CONTENT
|
||||
from django.test import TestCase
|
||||
from django.contrib.auth.models import User
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.core import mail
|
||||
from django.test.utils import override_settings
|
||||
|
@ -194,7 +193,7 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
self.post(self.password_change_url, status_code=403)
|
||||
|
||||
# create user
|
||||
user = User.objects.create_user(self.USERNAME, '', self.PASS)
|
||||
user = get_user_model().objects.create_user(self.USERNAME, '', self.PASS)
|
||||
|
||||
self.post(self.login_url, data=payload, status_code=200)
|
||||
self.assertEqual('key' in self.response.json.keys(), True)
|
||||
|
@ -222,7 +221,7 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
"username": self.USERNAME,
|
||||
"password": self.PASS
|
||||
}
|
||||
User.objects.create_user(self.USERNAME, '', self.PASS)
|
||||
get_user_model().objects.create_user(self.USERNAME, '', self.PASS)
|
||||
self.post(self.login_url, data=login_payload, status_code=200)
|
||||
self.token = self.response.json['key']
|
||||
|
||||
|
@ -257,7 +256,7 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
"username": self.USERNAME,
|
||||
"password": self.PASS
|
||||
}
|
||||
User.objects.create_user(self.USERNAME, '', self.PASS)
|
||||
get_user_model().objects.create_user(self.USERNAME, '', self.PASS)
|
||||
self.post(self.login_url, data=login_payload, status_code=200)
|
||||
self.token = self.response.json['key']
|
||||
|
||||
|
@ -285,7 +284,7 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
self.post(self.login_url, data=login_payload, status_code=200)
|
||||
|
||||
def test_password_reset(self):
|
||||
user = User.objects.create_user(self.USERNAME, self.EMAIL, self.PASS)
|
||||
user = get_user_model().objects.create_user(self.USERNAME, self.EMAIL, self.PASS)
|
||||
|
||||
# call password reset
|
||||
mail_count = len(mail.outbox)
|
||||
|
@ -340,7 +339,7 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
self.post(self.login_url, data=payload, status_code=200)
|
||||
|
||||
def test_user_details(self):
|
||||
user = User.objects.create_user(self.USERNAME, self.EMAIL, self.PASS)
|
||||
user = get_user_model().objects.create_user(self.USERNAME, self.EMAIL, self.PASS)
|
||||
payload = {
|
||||
"username": self.USERNAME,
|
||||
"password": self.PASS
|
||||
|
@ -350,19 +349,19 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
self.get(self.user_url, status_code=200)
|
||||
|
||||
self.patch(self.user_url, data=self.BASIC_USER_DATA, status_code=200)
|
||||
user = User.objects.get(pk=user.pk)
|
||||
user = get_user_model().objects.get(pk=user.pk)
|
||||
self.assertEqual(user.first_name, self.response.json['first_name'])
|
||||
self.assertEqual(user.last_name, self.response.json['last_name'])
|
||||
self.assertEqual(user.email, self.response.json['email'])
|
||||
|
||||
def test_registration(self):
|
||||
user_count = User.objects.all().count()
|
||||
user_count = get_user_model().objects.all().count()
|
||||
|
||||
# test empty payload
|
||||
self.post(self.register_url, data={}, status_code=400)
|
||||
|
||||
self.post(self.register_url, data=self.REGISTRATION_DATA, status_code=201)
|
||||
self.assertEqual(User.objects.all().count(), user_count + 1)
|
||||
self.assertEqual(get_user_model().objects.all().count(), user_count + 1)
|
||||
new_user = get_user_model().objects.latest('id')
|
||||
self.assertEqual(new_user.username, self.REGISTRATION_DATA['username'])
|
||||
|
||||
|
@ -374,7 +373,7 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
ACCOUNT_EMAIL_REQUIRED=True
|
||||
)
|
||||
def test_registration_with_email_verification(self):
|
||||
user_count = User.objects.all().count()
|
||||
user_count = get_user_model().objects.all().count()
|
||||
mail_count = len(mail.outbox)
|
||||
|
||||
# test empty payload
|
||||
|
@ -383,7 +382,7 @@ class APITestCase1(TestCase, BaseAPITestCase):
|
|||
|
||||
self.post(self.register_url, data=self.REGISTRATION_DATA_WITH_EMAIL,
|
||||
status_code=status.HTTP_201_CREATED)
|
||||
self.assertEqual(User.objects.all().count(), user_count + 1)
|
||||
self.assertEqual(get_user_model().objects.all().count(), user_count + 1)
|
||||
self.assertEqual(len(mail.outbox), mail_count + 1)
|
||||
new_user = get_user_model().objects.latest('id')
|
||||
self.assertEqual(new_user.username, self.REGISTRATION_DATA['username'])
|
||||
|
@ -452,19 +451,19 @@ class TestSocialAuth(TestCase, BaseAPITestCase):
|
|||
responses.add(responses.GET, self.graph_api_url, body=resp_body,
|
||||
status=200, content_type='application/json')
|
||||
|
||||
users_count = User.objects.all().count()
|
||||
users_count = get_user_model().objects.all().count()
|
||||
payload = {
|
||||
'access_token': 'abc123'
|
||||
}
|
||||
|
||||
self.post(self.fb_login_url, data=payload, status_code=200)
|
||||
self.assertIn('key', self.response.json.keys())
|
||||
self.assertEqual(User.objects.all().count(), users_count + 1)
|
||||
self.assertEqual(get_user_model().objects.all().count(), users_count + 1)
|
||||
|
||||
# make sure that second request will not create a new user
|
||||
self.post(self.fb_login_url, data=payload, status_code=200)
|
||||
self.assertIn('key', self.response.json.keys())
|
||||
self.assertEqual(User.objects.all().count(), users_count + 1)
|
||||
self.assertEqual(get_user_model().objects.all().count(), users_count + 1)
|
||||
|
||||
@responses.activate
|
||||
@override_settings(
|
||||
|
|
Loading…
Reference in New Issue
Block a user