Imporve download course offline views test to check files inside zip file

1 files changed, 12 insertions, 3 deletions

diff --git a/yaksh/test_views.py b/yaksh/test_views.py
index c1511c0..e691140 100644
--- a/yaksh/test_views.py
+++ b/yaksh/test_views.py
@@ -2049,9 +2049,18 @@ class TestCourses(TestCase):
         )
         course_name = self.user1_course.name.replace(" ", "_")
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(response.get('Content-Disposition'),
-                         'attachment; filename={0}.zip'.format(course_name)
-                         )
+        zip_file = string_io(response.content)
+        zipped_file = zipfile.ZipFile(zip_file, 'r')
+        self.assertIsNone(zipped_file.testzip())
+        files_in_zip = zipped_file.namelist()
+        module_path = os.path.join(course_name, "demo_module",
+                                   "demo_module.html")
+        lesson_path = os.path.join(course_name, "demo_module", "demo_lesson",
+                                   "demo_lesson.html")
+        self.assertIn(module_path, files_in_zip)
+        self.assertIn(lesson_path, files_in_zip)
+        zip_file.close()
+        zipped_file.close()
         self.user1_course.learning_module.remove(self.learning_module1)