fix: set enrollment url course id to advertised course run key

enterprise_catalog/apps/api/tests/test_tasks.py

@@ -959,6 +959,27 @@ def test_get_algolia_objects_from_course_metadata(self):
         test_course.catalog_queries.set(catalog_queries[0:3])
 
         algolia_objects = tasks.get_algolia_objects_from_course_content_metadata(test_course)
+
+        expected_transformed_advertised_course_run = {
+            'key': 'course-v1:edX+DemoX+2T2024',
+            'pacing_type': None,
+            'availability': 'current',
+            'start': '2024-02-12T11:00:00Z',
+            'end': '2026-02-05T11:00:00Z',
+            'min_effort': None,
+            'max_effort': None,
+            'weeks_to_complete': None,
+            'upgrade_deadline': 1769471999.0,
+            'enroll_by': 1769471999.0,
+            'has_enroll_by': True,
+            'enroll_start': None,
+            'has_enroll_start': False,
+            'content_price': 50.0,
+            'is_active': True,
+            'is_late_enrollment_eligible': False,
+            'restriction_type': None
+        }
+
         # Should look something like-
         #  [{'advertised_course_run': {'availability': 'current',
         #                             'end': None,
@@ -1004,6 +1025,7 @@ def test_get_algolia_objects_from_course_metadata(self):
         for algo_object in algolia_objects:
             assert algo_object.get('key') == test_course.content_key
             assert algo_object.get('uuid') == test_course.json_metadata.get('uuid')
+            assert algo_object.get('advertised_course_run') == expected_transformed_advertised_course_run
 
             if object_catalogs := algo_object.get('enterprise_catalog_uuids'):
                 assert set(object_catalogs) == {str(catalog.uuid) for catalog in catalogs}

enterprise_catalog/apps/catalog/algolia_utils.py

@@ -31,8 +31,8 @@
     VIDEO,
 )
 from enterprise_catalog.apps.catalog.content_metadata_utils import (
+    get_advertised_course_run,
     get_course_first_paid_enrollable_seat_price,
-    get_course_run_by_uuid,
     is_course_run_active,
 )
 from enterprise_catalog.apps.catalog.models import ContentMetadata
@@ -229,11 +229,7 @@ def _should_index_course(course_metadata):
         bool: Whether or not the course should be indexed by algolia.
     """
     course_json_metadata = course_metadata.json_metadata
-    advertised_course_run_uuid = course_json_metadata.get('advertised_course_run_uuid')
-    advertised_course_run = get_course_run_by_uuid(
-        course_json_metadata,
-        advertised_course_run_uuid,
-    )
+    advertised_course_run = get_advertised_course_run(course_json_metadata)
 
     # We define a series of no-arg functions, each of which has the property that,
     # if it returns true, means we should *not* index this record.
@@ -477,8 +473,7 @@ def get_course_language(course):
     Returns:
         string: human-readable language name parsed from a language code, or None if language name is not present.
     """
-    advertised_course_run = get_course_run_by_uuid(course, course.get('advertised_course_run_uuid'))
-    if not advertised_course_run:
+    if not (advertised_course_run := get_advertised_course_run(course)):
         return None
 
     content_language_name = advertised_course_run.get('content_language_search_facet_name')
@@ -496,8 +491,7 @@ def get_course_transcript_languages(course):
     Returns:
         list: a list of available human-readable video transcript languages parsed from a language code.
     """
-    advertised_course_run = get_course_run_by_uuid(course, course.get('advertised_course_run_uuid'))
-    if not advertised_course_run:
+    if not (advertised_course_run := get_advertised_course_run(course)):
         return None
 
     transcript_languages = advertised_course_run.get('transcript_languages_search_facet_names')
@@ -1168,23 +1162,6 @@ def _get_course_run(course, course_run):
     return course_run
 
 
-def get_advertised_course_run(course):
-    """
-    Get part of the advertised course_run as per advertised_course_run_uuid
-
-    Argument:
-        course (dict)
-
-    Returns:
-        dict: containing key, pacing_type, start, end, and upgrade deadline
-        for the course_run, or None
-    """
-    full_course_run = get_course_run_by_uuid(course, course.get('advertised_course_run_uuid'))
-    if full_course_run is None:
-        return None
-    return _get_course_run(course, full_course_run)
-
-
 def get_course_runs(course):
     """
     Extract and transform a list of course runs into what we'll index.
@@ -1514,6 +1491,8 @@ def _algolia_object_from_product(product, algolia_fields):
     """
     searchable_product = copy.deepcopy(product)
     if searchable_product.get('content_type') == COURSE:
+        advertised_course_run = get_advertised_course_run(searchable_product)
+        transformed_advertised_course_run = _get_course_run(searchable_product, advertised_course_run)
         searchable_product.update({
             'language': get_course_language(searchable_product),
             'availability': get_course_availability(searchable_product),
@@ -1522,7 +1501,7 @@ def _algolia_object_from_product(product, algolia_fields):
             'program_titles': get_course_program_titles(searchable_product),
             'subjects': get_course_subjects(searchable_product),
             'card_image_url': get_course_card_image_url(searchable_product),
-            'advertised_course_run': get_advertised_course_run(searchable_product),
+            'advertised_course_run': transformed_advertised_course_run,
             'course_runs': get_course_runs(searchable_product),
             'upcoming_course_runs': get_upcoming_course_runs(searchable_product),
             'skill_names': get_course_skill_names(searchable_product),

enterprise_catalog/apps/catalog/content_metadata_utils.py

@@ -89,7 +89,7 @@ def get_course_first_paid_enrollable_seat_price(course):
     # Use advertised course run.
     # If that fails use one of the other active course runs.
     # (The latter is what Discovery does)
-    advertised_course_run = get_course_run_by_uuid(course, course.get('advertised_course_run_uuid'))
+    advertised_course_run = get_advertised_course_run(course)
     if advertised_course_run and advertised_course_run.get('first_enrollable_paid_seat_price'):
         return advertised_course_run.get('first_enrollable_paid_seat_price')
 
@@ -102,3 +102,20 @@ def get_course_first_paid_enrollable_seat_price(course):
         if 'first_enrollable_paid_seat_price' in course_run:
             return course_run['first_enrollable_paid_seat_price']
     return None
+
+
+def get_advertised_course_run(course):
+    """
+    Get part of the advertised course_run as per advertised_course_run_uuid
+
+    Argument:
+        course (dict)
+
+    Returns:
+        dict: containing key, pacing_type, start, end, and upgrade deadline
+        for the course_run, or None
+    """
+    full_course_run = get_course_run_by_uuid(course, course.get('advertised_course_run_uuid'))
+    if full_course_run is None:
+        return None
+    return full_course_run

enterprise_catalog/apps/catalog/models.py

@@ -45,6 +45,7 @@
     json_serialized_course_modes,
 )
 from enterprise_catalog.apps.catalog.content_metadata_utils import (
+    get_advertised_course_run,
     get_course_first_paid_enrollable_seat_price,
 )
 from enterprise_catalog.apps.catalog.utils import (
@@ -555,10 +556,15 @@ def get_content_enrollment_url(self, content_metadata):
         else:
             # Catalog param only needed for legacy (non-learner-portal) enrollment URLs
             params['catalog'] = self.uuid
+
+            course_run_key = content_key
+            if not parent_content_key:
+                if advertised_course_run := get_advertised_course_run(content_metadata.json_metadata):
+                    course_run_key = advertised_course_run['key']
             url = '{}/enterprise/{}/course/{}/enroll/'.format(
                 settings.LMS_BASE_URL,
                 self.enterprise_uuid,
-                content_key,
+                course_run_key,
             )
 
         return update_query_parameters(url, params)

enterprise_catalog/apps/catalog/serializers.py

@@ -11,7 +11,7 @@
 from enterprise_catalog.apps.api.constants import CourseMode
 from enterprise_catalog.apps.catalog.constants import EXEC_ED_2U_COURSE_TYPE
 from enterprise_catalog.apps.catalog.content_metadata_utils import (
-    get_course_run_by_uuid,
+    get_advertised_course_run,
 )
 
 
@@ -113,8 +113,7 @@ def is_exec_ed_2u_course(self):
     def course_run_metadata(self):
         if run_metadata := self.instance.get('course_run_metadata'):
             return run_metadata
-        advertised_course_run_uuid = self.course_metadata.get('advertised_course_run_uuid')
-        return get_course_run_by_uuid(self.course_metadata, advertised_course_run_uuid)
+        return get_advertised_course_run(self.course_metadata)
 
     @extend_schema_field(serializers.DateTimeField)
     def get_start_date(self, obj) -> str:  # pylint: disable=unused-argument

enterprise_catalog/apps/catalog/tests/factories.py

@@ -99,7 +99,7 @@ def _json_metadata(self):
                 'logo_image_url': fake.image_url() + '.jpg',
             }]
             course_runs = [{
-                'key': 'course-v1:edX+DemoX',
+                'key': 'course-v1:edX+DemoX+2T2024',
                 'uuid': str(FAKE_ADVERTISED_COURSE_RUN_UUID),
                 'content_language': 'en-us',
                 'status': 'published',