Commit 746aab5e by tasawernawaz

Updating course video source field

parent 698eb202
......@@ -430,7 +430,7 @@ class CourseMarketingSiteDataLoader(AbstractMarketingSiteDataLoader):
return level_type
def get_video(self, data):
video_url = self._get_nested_url(data.get('field_product_video'))
video_url = self._get_nested_url(data.get('field_course_video') or data.get('field_product_video'))
image_url = self._get_nested_url(data.get('field_course_image_featured_card'))
return self.get_or_create_video(video_url, image_url)
......@@ -469,6 +469,7 @@ class CourseMarketingSiteDataLoader(AbstractMarketingSiteDataLoader):
'hidden': self.get_hidden(data),
'weeks_to_complete': None,
'mobile_available': data.get('field_course_enrollment_mobile') or False,
'video': course.video,
}
if weeks_to_complete:
......
......@@ -397,9 +397,18 @@ class CourseMarketingSiteDataLoaderTests(AbstractMarketingSiteDataLoaderTestMixi
self.assertEqual(self.loader.get_description(data), 'Test')
def test_get_video(self):
"""Verify that method gets video from any of 'field_course_video' or 'field_product_video.'"""
image_url = 'https://example.com/image.jpg'
video_url = 'https://example.com/video.mp4'
data = {
'field_course_video': {'url': video_url},
'field_course_image_featured_card': {'url': image_url}
}
video = self.loader.get_video(data)
self.assertEqual(video.src, video_url)
self.assertEqual(video.image.src, image_url)
data = {
'field_product_video': {'url': video_url},
'field_course_image_featured_card': {'url': image_url}
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment