diff --git a/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_account_deletion.py b/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_account_deletion.py index 197e9cf6aa5..23f4a5f292a 100644 --- a/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_account_deletion.py +++ b/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_account_deletion.py @@ -9,9 +9,13 @@ from django.db.models.signals import post_save from django.urls import reverse from django.contrib.auth import get_user_model +from django.contrib.auth.models import User + from rest_framework.test import APITestCase from rest_framework import status +from student.models import UserProfile + from ...multi_tenant_emails.tests.test_utils import with_organization_context from openedx.core.djangoapps.user_api.accounts.tests.retirement_helpers import ( diff --git a/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_reset_password.py b/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_reset_password.py index 4a0593a0363..253b4b53462 100644 --- a/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_reset_password.py +++ b/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_idp_reset_password.py @@ -5,6 +5,7 @@ import ddt from django.conf import settings +from django.contrib.auth.models import User from django.core import mail from django.db.models.signals import post_save from django.test import RequestFactory, TestCase @@ -12,6 +13,7 @@ from openedx.core.djangoapps.user_authn.views.password_reset import password_reset from openedx.core.djangolib.testing.utils import skip_unless_lms +from student.models import UserProfile from student.tests.factories import UserFactory diff --git a/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_registration_api.py b/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_registration_api.py index 40358b17632..937f032ddf5 100644 --- a/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_registration_api.py +++ b/openedx/core/djangoapps/appsembler/tahoe_idp/tests/test_tahoe_registration_api.py @@ -4,12 +4,14 @@ import ddt from mock import patch +from django.contrib.auth.models import User from django.db.models.signals import post_save from django.urls import reverse_lazy from rest_framework import status from rest_framework.test import APITestCase from openedx.core.djangolib.testing.utils import skip_unless_lms +from student.models import User from ...multi_tenant_emails.tests.test_utils import with_organization_context