From 9d333760c7f7bae212091af031d6f89c24851da0 Mon Sep 17 00:00:00 2001 From: Stas Rudakou Date: Sat, 27 Oct 2018 18:39:04 +0300 Subject: [PATCH] Extend `WagtailTestUtils::login` to login other users --- wagtail/tests/utils/wagtail_tests.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/wagtail/tests/utils/wagtail_tests.py b/wagtail/tests/utils/wagtail_tests.py index be288468d..7b96b8405 100644 --- a/wagtail/tests/utils/wagtail_tests.py +++ b/wagtail/tests/utils/wagtail_tests.py @@ -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