Remove remaining direct references to the User model. Closes GitLab issue #225.

This commit is contained in:
Roberto Rosario
2016-01-28 05:55:15 -04:00
parent 27d8644438
commit 2ebb84f18f
15 changed files with 36 additions and 34 deletions

View File

@@ -6,7 +6,8 @@ import time
from json import loads
from django.contrib.auth.models import User
from django.contrib.auth import get_user_model
from django.core.files import File
from django.core.urlresolvers import reverse
from django.test import override_settings
@@ -32,7 +33,7 @@ class DocumentTypeAPITestCase(APITestCase):
"""
def setUp(self):
self.admin_user = User.objects.create_superuser(
self.admin_user = get_user_model().objects.create_superuser(
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
password=TEST_ADMIN_PASSWORD
)
@@ -95,7 +96,7 @@ class DocumentAPITestCase(APITestCase):
"""
def setUp(self):
self.admin_user = User.objects.create_superuser(
self.admin_user = get_user_model().objects.create_superuser(
username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
password=TEST_ADMIN_PASSWORD
)