From b7d16f51f45a137b3c514aaa65d9c667b92bcfa7 Mon Sep 17 00:00:00 2001 From: Matt Westcott Date: Mon, 28 Jul 2014 13:12:28 +0100 Subject: [PATCH] remove newly-added references to User in test_account_management --- wagtail/wagtailadmin/tests/test_account_management.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wagtail/wagtailadmin/tests/test_account_management.py b/wagtail/wagtailadmin/tests/test_account_management.py index 8cb187ad6..c398e17cb 100644 --- a/wagtail/wagtailadmin/tests/test_account_management.py +++ b/wagtail/wagtailadmin/tests/test_account_management.py @@ -32,7 +32,7 @@ class TestAuthentication(TestCase, WagtailTestUtils): the user was logged in successfully """ # Create user to log in with - user = User.objects.create_superuser(username='test', email='test@email.com', password='password') + user = get_user_model().objects.create_superuser(username='test', email='test@email.com', password='password') # Post credentials to the login page post_data = { @@ -71,7 +71,7 @@ class TestAuthentication(TestCase, WagtailTestUtils): This tests issue #431 """ # Login as unprivileged user - User.objects.create(username='unprivileged', password='123') + get_user_model().objects.create(username='unprivileged', password='123') self.client.login(username='unprivileged', password='123') # Get login page