Change test util to handle custom user models

This should handle custom user models that use either username or email as the 'USERNAME_FIELD'
pull/2256/merge
Adam 2016-02-18 16:55:58 -05:00 zatwierdzone przez Matt Westcott
rodzic eb269cab9c
commit 416e4be27b
1 zmienionych plików z 10 dodań i 2 usunięć

Wyświetl plik

@ -14,11 +14,19 @@ from django.utils.text import slugify
class WagtailTestUtils(object):
def login(self):
user_model = get_user_model()
# Create a user
user = get_user_model().objects.create_superuser(username='test', email='test@email.com', password='password')
user_data = {}
user_data[user_model.USERNAME_FIELD] = 'test@email.com'
user_data['password'] = 'password'
for field in user_model.REQUIRED_FIELDS:
user_data[field] = field
user = user_model.objects.create_superuser(**user_data)
# Login
self.client.login(username='test', password='password')
self.client.login(password='password', **{user_model.USERNAME_FIELD: 'test@email.com'})
return user