Change multiple files

- Add download sample yaml for toc in the lesson
- Add validation for upload toc yaml
- Add tests for download sample yaml toc

1 files changed, 7 insertions, 2 deletions

diff --git a/yaksh/test_views.py b/yaksh/test_views.py
index 9ce3e8b..d80a01e 100644
--- a/yaksh/test_views.py
+++ b/yaksh/test_views.py
@@ -3320,7 +3320,7 @@ class TestCourseDetail(TestCase):
         response = self.client.get(reverse('yaksh:download_sample_csv'))
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.get('Content-Disposition'),
-                         'attachment; filename="sample_user_upload"')
+                         'attachment; filename="sample_user_upload.csv"')
 
     def test_view_course_status(self):
         """ Test to view course status """
@@ -8484,7 +8484,7 @@ class TestLessonContents(TestCase):
         self.assertEqual(response.status_code, 200)
         self.assertEqual(student_info.get("student_id"), self.student.id)
 
-    def test_upload_lesson_contents(self):
+    def test_upload_download_lesson_contents(self):
         self.client.login(
             username=self.user1.username,
             password=self.user1_plaintext_pass
@@ -8567,3 +8567,8 @@ class TestLessonContents(TestCase):
             )
             self.assertIn("Invalid time format", messages[1])
 
+        # Download yaml sample
+        response = self.client.get(reverse('yaksh:download_sample_toc'))
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.get('Content-Disposition'),
+                         'attachment; filename="sample_lesson_toc.yaml"')