fix failing tests

onaio · Oct 3, 2024 · 1ac4f0c · 1ac4f0c
1 parent 4200d1d
commit 1ac4f0c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/onadata/apps/api/tests/viewsets/test_osm_viewset.py b/onadata/apps/api/tests/viewsets/test_osm_viewset.py
@@ -147,7 +147,7 @@ def test_osm_csv_export(self):
         content_disposition = headers["Content-Disposition"]
         filename = filename_from_disposition(content_disposition)
         basename, ext = os.path.splitext(filename)
-        self.assertEqual(ext, ".csv")
+        self.assertEqual(ext, '.csv"')
 
         content = get_response_content(response)
         reader = csv.DictReader(StringIO(content))

diff --git a/onadata/apps/api/tests/viewsets/test_xform_list_viewset.py b/onadata/apps/api/tests/viewsets/test_xform_list_viewset.py
@@ -990,7 +990,7 @@ def test_retrieve_xform_media_linked_xform(self):
         )
         self.assertEqual(response.status_code, 200)
         self.assertEqual(
-            response["Content-Disposition"], "attachment; filename=transportation.csv"
+            response["Content-Disposition"], 'attachment; filename="transportation.csv"'
         )
 
     def test_retrieve_xform_media_entity_list_dataset(self):

diff --git a/onadata/apps/api/tests/viewsets/test_xform_viewset.py b/onadata/apps/api/tests/viewsets/test_xform_viewset.py
@@ -5559,7 +5559,7 @@ def test_export_zip_async(self, async_result):
             content_disposition = headers["Content-Disposition"]
             filename = filename_from_disposition(content_disposition)
             basename, ext = os.path.splitext(filename)
-            self.assertEqual(ext, ".zip")
+            self.assertEqual(ext, '.zip"')
 
     @override_settings(CELERY_TASK_ALWAYS_EAGER=True)
     @patch("onadata.libs.utils.api_export_tools.AsyncResult")