diff --git a/lms/djangoapps/support/tests/test_tasks.py b/lms/djangoapps/support/tests/test_tasks.py index 09c2f47056..7f4818ce73 100644 --- a/lms/djangoapps/support/tests/test_tasks.py +++ b/lms/djangoapps/support/tests/test_tasks.py @@ -14,6 +14,7 @@ from lms.djangoapps.support.models import CourseResetAudit from common.djangoapps.student.models.course_enrollment import CourseEnrollment from common.djangoapps.student.roles import SupportStaffRole from common.djangoapps.student.tests.factories import UserFactory +from xmodule.video_block import VideoBlock class ResetStudentCourse(TestSubmittingProblems): @@ -41,6 +42,7 @@ class ResetStudentCourse(TestSubmittingProblems): self.p1 = '' self.p2 = '' self.p3 = '' + self.video = '' def basic_setup(self): """ @@ -72,6 +74,23 @@ class ResetStudentCourse(TestSubmittingProblems): self.p1 = self.add_dropdown_to_section(vertical.location, 'p1', 1) self.p2 = self.add_dropdown_to_section(vertical.location, 'p2', 1) self.p3 = self.add_dropdown_to_section(vertical.location, 'p3', 1) + video_sample_xml = """ + + """ + video_data = VideoBlock.parse_video_xml(video_sample_xml) + video_data.pop('source') + self.video = BlockFactory.create( + category='video', + parent_location=vertical.location, + **video_data + ) self.refresh_course() @@ -104,6 +123,13 @@ class ResetStudentCourse(TestSubmittingProblems): self.p3.location, self.user, True + ), + call( + self.course.id, + self.student_user, + self.video.location, + self.user, + True ) ])