diff --git a/course_discovery/apps/api/tests/test_serializers.py b/course_discovery/apps/api/tests/test_serializers.py index 285f1bbd47..ec023e9adf 100644 --- a/course_discovery/apps/api/tests/test_serializers.py +++ b/course_discovery/apps/api/tests/test_serializers.py @@ -2554,6 +2554,9 @@ def get_expected_data(cls, course_run, course_skill, request): 'first_enrollable_paid_seat_sku': course_run.first_enrollable_paid_seat_sku(), 'first_enrollable_paid_seat_price': course_run.first_enrollable_paid_seat_price, 'is_enrollable': course_run.is_enrollable, + 'restriction_type': ( + course_run.restricted_run.restriction_type if hasattr(course_run, 'restricted_run') else None + ) } diff --git a/course_discovery/apps/course_metadata/search_indexes/serializers/course_run.py b/course_discovery/apps/course_metadata/search_indexes/serializers/course_run.py index c7f23279be..1878915293 100644 --- a/course_discovery/apps/course_metadata/search_indexes/serializers/course_run.py +++ b/course_discovery/apps/course_metadata/search_indexes/serializers/course_run.py @@ -90,6 +90,7 @@ class Meta: 'transcript_languages', 'type', 'weeks_to_complete', + 'restriction_type', )