From c6c57869fe653d2ea0502017a9fb15f2f745491b Mon Sep 17 00:00:00 2001
From: adityacp
Date: Wed, 30 Sep 2020 19:14:38 +0530
Subject: 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
---
 yaksh/test_views.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'yaksh/test_views.py')

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"')
-- 
cgit