fix: add missing CEA definition and fix imports in tests

open-craft · May 26, 2023 · 6e74b8b · 6e74b8b
1 parent 2be191d
commit 6e74b8b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/enterprise/models.py b/enterprise/models.py
@@ -78,6 +78,7 @@
     from common.djangoapps.student.models import CourseEnrollment, CourseEnrollmentAllowed
 except ImportError:
     CourseEnrollment = None
+    CourseEnrollmentAllowed = None
 
 try:
     from openedx.core.djangoapps.content.course_overviews.models import CourseOverview

diff --git a/tests/test_enterprise/api/test_views.py b/tests/test_enterprise/api/test_views.py
@@ -67,7 +67,8 @@
 )
 from test_utils.decorators import mock_api_response
 from test_utils.factories import FAKER, PendingEnterpriseCustomerUserFactory
-from test_utils.fake_enrollment_api import get_course_details, get_default_branding_object
+from test_utils.fake_enrollment_api import get_course_details
+from test_utils.fake_enterprise_api import get_default_branding_object
 
 fake = Faker()
 
@@ -3612,12 +3613,12 @@ def test_bulk_enrollment_in_bulk_courses_pending_licenses(
                     },
                     {
                         'email': 'abc@test.com',
-                        'course_run_key': 'course-v2:edX+DemoX+Second_Demo_Course',
+                        'course_run_key': 'course-v1:HarvardX+CoolScience+2016',
                         'license_uuid': '5a88bdcade7c4ecb838f8111b68e18ac'
                     },
                     {
                         'email': 'xyz@test.com',
-                        'course_run_key': 'course-v2:edX+DemoX+Second_Demo_Course',
+                        'course_run_key': 'course-v1:HarvardX+CoolScience+2016',
                         'license_uuid': '2c58acdade7c4ede838f7111b42e18ac'
                     },
                 ]