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

mayan/apps/dynamic_search/tests/test_api.py

@@ -2,7 +2,7 @@ from __future__ import unicode_literals
 
 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
@@ -23,7 +23,7 @@ class SearchAPITestCase(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
         )

mayan/apps/dynamic_search/tests/test_models.py

@@ -1,6 +1,6 @@
 from __future__ import unicode_literals
 
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from django.core.files.base import File
 from django.test import TestCase
 
@@ -14,7 +14,7 @@ from user_management.tests import (
 
 class DocumentSearchTestCase(TestCase):
     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
         )

mayan/apps/dynamic_search/tests/test_views.py

@@ -1,6 +1,6 @@
 from __future__ import unicode_literals
 
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from django.core.files.base import File
 from django.core.urlresolvers import reverse
 from django.test import TestCase
@@ -20,7 +20,7 @@ class Issue46TestCase(TestCase):
     """
 
     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
         )