refactor: replace some create_user with UserFactory to avoid non-existent profile errors

Author: mariajgrimaldi

cms/djangoapps/cms_user_tasks/tests.py

@@ -10,7 +10,6 @@
 import ddt
 from boto.exception import NoAuthHandlerFound
 from django.conf import settings
-from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
 from django.core import mail
 from django.test import override_settings
 from django.urls import reverse
@@ -20,6 +19,7 @@
 from user_tasks.serializers import ArtifactSerializer, StatusSerializer
 
 from cms.djangoapps.contentstore.toggles import BYPASS_OLX_FAILURE
+from common.djangoapps.student.tests.factories import UserFactory
 
 from .signals import user_task_stopped
 
@@ -78,7 +78,7 @@ class TestUserTasks(APITestCase):
 
     @classmethod
     def setUpTestData(cls):  # lint-amnesty, pylint: disable=super-method-not-called
-        cls.user = User.objects.create_user('test_user', '[email protected]', 'password')
+        cls.user = UserFactory.create(username='test_user', email='[email protected]', password='password')
         cls.status = UserTaskStatus.objects.create(
             user=cls.user, task_id=str(uuid4()), task_class='test_rest_api.sample_task', name='SampleTask 2',
             total_steps=5)
@@ -152,7 +152,7 @@ class TestUserTaskStopped(APITestCase):
 
     @classmethod
     def setUpTestData(cls):  # lint-amnesty, pylint: disable=super-method-not-called
-        cls.user = User.objects.create_user('test_user', '[email protected]', 'password')
+        cls.user = UserFactory.create(username='test_user', email='[email protected]', password='password')
         cls.status = UserTaskStatus.objects.create(
             user=cls.user, task_id=str(uuid4()), task_class='test_rest_api.sample_task', name='SampleTask 2',
             total_steps=5)

cms/djangoapps/contentstore/tests/test_i18n.py

@@ -6,12 +6,12 @@
 import gettext
 from unittest import mock, skip
 
-from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
 from django.utils import translation
 from django.utils.translation import get_language
 
 from cms.djangoapps.contentstore.tests.utils import AjaxEnabledTestClient
 from cms.djangoapps.contentstore.views.preview import _preview_module_system
+from common.djangoapps.student.tests.factories import UserFactory
 from openedx.core.lib.edx_six import get_gettext
 from xmodule.modulestore.django import ModuleI18nService
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
@@ -192,7 +192,7 @@ def setUp(self):
         self.password = 'foo'
 
         # Create the use so we can log them in.
-        self.user = User.objects.create_user(self.uname, self.email, self.password)
+        self.user = UserFactory.create(username=self.uname, email=self.email, password=self.password)
 
         # Note that we do not actually need to do anything
         # for registration if we directly mark them active.

cms/djangoapps/contentstore/tests/test_permissions.py

@@ -5,12 +5,11 @@
 
 import copy
 
-from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
-
 from cms.djangoapps.contentstore.tests.utils import AjaxEnabledTestClient
 from cms.djangoapps.contentstore.utils import reverse_course_url, reverse_url
 from common.djangoapps.student import auth
 from common.djangoapps.student.roles import CourseInstructorRole, CourseStaffRole, OrgInstructorRole, OrgStaffRole
+from common.djangoapps.student.tests.factories import UserFactory
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
 
 
@@ -52,7 +51,7 @@ def _create_users(self):
         for i in range(8):
             username = f"user{i}"
             email = f"test+user{i}@edx.org"
-            user = User.objects.create_user(username, email, 'foo')
+            user = UserFactory.create(username=username, email=email, password='foo')
             user.is_active = True
             user.save()
             users.append(user)

cms/djangoapps/contentstore/views/tests/test_access.py

@@ -3,13 +3,12 @@
 """
 
 
-from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
 from django.test import TestCase
 from opaque_keys.edx.locator import CourseLocator
 
 from common.djangoapps.student.auth import add_users
 from common.djangoapps.student.roles import CourseInstructorRole, CourseStaffRole
-from common.djangoapps.student.tests.factories import AdminFactory
+from common.djangoapps.student.tests.factories import AdminFactory, UserFactory
 
 from ..access import get_user_role
 
@@ -23,8 +22,16 @@ def setUp(self):
         super().setUp()
 
         self.global_admin = AdminFactory()
-        self.instructor = User.objects.create_user('testinstructor', '[email protected]', 'foo')
-        self.staff = User.objects.create_user('teststaff', '[email protected]', 'foo')
+        self.instructor = UserFactory.create(
+            username='testinstructor',
+            email='[email protected]',
+            password='foo',
+        )
+        self.staff = UserFactory.create(
+            username='teststaff',
+            email='[email protected]',
+            password='foo',
+        )
         self.course_key = CourseLocator('mitX', '101', 'test')
 
     def test_get_user_role_instructor(self):

cms/djangoapps/contentstore/views/tests/test_user.py

@@ -12,18 +12,21 @@
 from common.djangoapps.student import auth
 from common.djangoapps.student.models import CourseEnrollment
 from common.djangoapps.student.roles import CourseInstructorRole, CourseStaffRole
+from common.djangoapps.student.tests.factories import UserFactory
 
 
 class UsersTestCase(CourseTestCase):  # lint-amnesty, pylint: disable=missing-class-docstring
     def setUp(self):
         super().setUp()
-        self.ext_user = User.objects.create_user(
-            "joe", "[email protected]", "haha")
+        self.ext_user = UserFactory.create(
+            username="joe", email="[email protected]", password="haha",
+        )
         self.ext_user.is_active = True
         self.ext_user.is_staff = False
         self.ext_user.save()
-        self.inactive_user = User.objects.create_user(
-            "carl", "[email protected]", "haha")
+        self.inactive_user = UserFactory.create(
+            username="carl", email="[email protected]", password="haha",
+        )
         self.inactive_user.is_active = False
         self.inactive_user.is_staff = False
         self.inactive_user.save()

cms/djangoapps/course_creators/tests/test_admin.py

@@ -6,7 +6,6 @@
 from unittest import mock
 
 from django.contrib.admin.sites import AdminSite
-from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
 from django.core import mail
 from django.http import HttpRequest
 from django.test import TestCase
@@ -15,6 +14,7 @@
 from cms.djangoapps.course_creators.models import CourseCreator
 from common.djangoapps.student import auth
 from common.djangoapps.student.roles import CourseCreatorRole
+from common.djangoapps.student.tests.factories import UserFactory
 
 
 def mock_render_to_string(template_name, context):
@@ -30,11 +30,19 @@ class CourseCreatorAdminTest(TestCase):
     def setUp(self):
         """ Test case setup """
         super().setUp()
-        self.user = User.objects.create_user('test_user', '[email protected]', 'foo')
+        self.user = UserFactory.create(
+            username='test_user',
+            email='[email protected]',
+            password='foo',
+        )
         self.table_entry = CourseCreator(user=self.user)
         self.table_entry.save()
 
-        self.admin = User.objects.create_user('Mark', '[email protected]', 'foo')
+        self.admin = UserFactory.create(
+            username='Mark',
+            email='[email protected]',
+            password='foo',
+        )
         self.admin.is_staff = True
 
         self.request = HttpRequest()

cms/djangoapps/course_creators/tests/test_views.py

@@ -5,7 +5,6 @@
 
 from unittest import mock
 
-from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
 from django.core.exceptions import PermissionDenied
 from django.test import TestCase
 from django.urls import reverse
@@ -19,6 +18,7 @@
 )
 from common.djangoapps.student import auth
 from common.djangoapps.student.roles import CourseCreatorRole
+from common.djangoapps.student.tests.factories import UserFactory
 
 
 class CourseCreatorView(TestCase):
@@ -29,8 +29,16 @@ class CourseCreatorView(TestCase):
     def setUp(self):
         """ Test case setup """
         super().setUp()
-        self.user = User.objects.create_user('test_user', '[email protected]', 'foo')
-        self.admin = User.objects.create_user('Mark', '[email protected]', 'foo')
+        self.user = UserFactory.create(
+            username='test_user',
+            email='[email protected]',
+            password='foo',
+        )
+        self.admin = UserFactory.create(
+            username='Mark',
+            email='[email protected]',
+            password='foo',
+        )
         self.admin.is_staff = True
 
     def test_staff_permission_required(self):

common/djangoapps/student/tests/test_authz.py

@@ -6,14 +6,14 @@
 import pytest
 
 from ccx_keys.locator import CCXLocator
-from django.contrib.auth.models import AnonymousUser, User  # lint-amnesty, pylint: disable=imported-auth-user
+from django.contrib.auth.models import AnonymousUser
 from django.core.exceptions import PermissionDenied
 from django.test import TestCase
 from opaque_keys.edx.locator import CourseLocator
 
 from common.djangoapps.student.auth import add_users, has_studio_read_access, has_studio_write_access, remove_users, user_has_role  # lint-amnesty, pylint: disable=line-too-long
 from common.djangoapps.student.roles import CourseCreatorRole, CourseInstructorRole, CourseStaffRole
-from common.djangoapps.student.tests.factories import AdminFactory
+from common.djangoapps.student.tests.factories import AdminFactory, UserFactory
 
 
 class CreatorGroupTest(TestCase):
@@ -24,8 +24,12 @@ class CreatorGroupTest(TestCase):
     def setUp(self):
         """ Test case setup """
         super().setUp()
-        self.user = User.objects.create_user('testuser', '[email protected]', 'foo')
-        self.admin = User.objects.create_user('Mark', '[email protected]', 'foo')
+        self.user = UserFactory.create(
+            username='testuser', email='[email protected]', password='foo',
+        )
+        self.admin = UserFactory.create(
+            username='Mark', email='[email protected]', password='foo',
+        )
         self.admin.is_staff = True
 
     def test_creator_group_not_enabled(self):
@@ -51,7 +55,7 @@ def test_creator_group_enabled_nonempty(self):
             assert user_has_role(self.user, CourseCreatorRole())
 
             # check that a user who has not been added to the group still returns false
-            user_not_added = User.objects.create_user('testuser2', '[email protected]', 'foo2')
+            user_not_added = UserFactory.create(username='testuser2', email='[email protected]', password='foo2')
             assert not user_has_role(user_not_added, CourseCreatorRole())
 
             # remove first user from the group and verify that CourseCreatorRole().has_user now returns false
@@ -153,7 +157,7 @@ def setUp(self):
         """
         super().setUp()
         self.global_admin = AdminFactory()
-        self.staff = User.objects.create_user('teststaff', '[email protected]', 'foo')
+        self.staff = UserFactory.create(username='teststaff', email='[email protected]', password='foo')
         self.ccx_course_key = CCXLocator.from_string('ccx-v1:edX+DemoX+Demo_Course+ccx@1')
         add_users(self.global_admin, CourseStaffRole(self.ccx_course_key), self.staff)
 
@@ -191,8 +195,12 @@ def setUp(self):
         """ Test case setup """
         super().setUp()
         self.global_admin = AdminFactory()
-        self.creator = User.objects.create_user('testcreator', '[email protected]', 'foo')
-        self.staff = User.objects.create_user('teststaff', '[email protected]', 'foo')
+        self.creator = UserFactory.create(
+            username='testcreator', email='[email protected]', password='foo',
+        )
+        self.staff = UserFactory.create(
+            username='teststaff', email='[email protected]', password='foo',
+        )
         self.course_key = CourseLocator('mitX', '101', 'test')
 
     def test_add_user_to_course_group(self):
@@ -240,7 +248,9 @@ def test_remove_user_from_course_group_permission_denied(self):
         Verifies PermissionDenied if caller of remove_user_from_course_group is not instructor role.
         """
         add_users(self.global_admin, CourseInstructorRole(self.course_key), self.creator)
-        another_staff = User.objects.create_user('another', '[email protected]', 'foo')
+        another_staff = UserFactory.create(
+            username='another', email='[email protected]', password='foo',
+        )
         add_users(self.global_admin, CourseStaffRole(self.course_key), self.creator, self.staff, another_staff)
         with pytest.raises(PermissionDenied):
             remove_users(self.staff, CourseStaffRole(self.course_key), another_staff)

common/djangoapps/student/tests/tests.py

@@ -715,7 +715,7 @@ class EnrollInCourseTest(EnrollmentEventTestMixin, CacheIsolationTestCase):
 
     @unittest.skipUnless(settings.ROOT_URLCONF == 'lms.urls', 'Test only valid in lms')
     def test_enrollment(self):
-        user = User.objects.create_user("joe", "[email protected]", "password")
+        user = UserFactory.create(username="joe", email="[email protected]", password="password")
         course_id = CourseKey.from_string("edX/Test101/2013")
         course_id_partial = CourseKey.from_string("edX/Test101/")
         course = CourseOverviewFactory.create(id=course_id)

common/djangoapps/third_party_auth/saml_configuration/tests/test_saml_configuration.py

@@ -3,10 +3,10 @@
 """
 from django.urls import reverse
 from django.contrib.sites.models import Site
-from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
 
 from rest_framework import status
 from rest_framework.test import APITestCase
+from common.djangoapps.student.tests.factories import UserFactory
 from common.djangoapps.third_party_auth.models import SAMLConfiguration
 from common.djangoapps.third_party_auth.tests.utils import skip_unless_thirdpartyauth
 SAML_CONFIGURATIONS = [
@@ -52,7 +52,7 @@ class SAMLConfigurationTests(APITestCase):
     @classmethod
     def setUpTestData(cls):
         super().setUpTestData()
-        cls.user = User.objects.create_user(username='testuser', password=TEST_PASSWORD)
+        cls.user = UserFactory.create(username='testuser', password=TEST_PASSWORD)
         cls.site, _ = Site.objects.get_or_create(domain='example.com')
         for config in SAML_CONFIGURATIONS:
             cls.samlconfiguration = SAMLConfiguration.objects.get_or_create(