Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-05-24 06:43:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Extend WagtailTestUtils::login to login other users

Browse source
This commit is contained in:
Stas Rudakou 2018-10-27 18:39:04 +03:00 committed by Mikalai Radchuk
parent f1148e7f76
commit 9d333760c7
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
wagtail/tests/utils/wagtail_tests.py
View file

@ -26,13 +26,14 @@ class WagtailTestUtils:
return user_model.objects.create_superuser(**user_data)
def login(self):
user = self.create_test_user()
def login(self, user=None):
if user is None:
user = self.create_test_user()
user_model = get_user_model()
# Login
self.assertTrue(
self.client.login(password='password', **{user_model.USERNAME_FIELD: 'test@email.com'})
self.client.login(password='password', **{user_model.USERNAME_FIELD: user.username})
)
return user