tests updated, additional test added

Author: ihsaan-ullah

src/apps/api/tests/test_datasets.py

@@ -30,8 +30,8 @@ def test_dataset_api_checks_duplicate_names_for_same_user(self):
         resp = self.client.post(reverse("data-list"), {
             'name': 'Test!',
             'type': Data.COMPETITION_BUNDLE,
-            'request_sassy_file_name': faker.file_name(),
-            'file_name': faker.file_name(),
+            'request_sassy_file_name': faker.file_name(extension='.zip'),
+            'file_name': faker.file_name(extension='.zip'),
             'file_size': 1000,
         })
 
@@ -42,7 +42,7 @@ def test_dataset_api_checks_duplicate_names_for_same_user(self):
         resp = self.client.put(reverse("data-detail", args=(self.existing_dataset.pk,)), {
             'name': 'Test!',
             'type': Data.COMPETITION_BUNDLE,
-            'request_sassy_file_name': faker.file_name(),
+            'request_sassy_file_name': faker.file_name(extension='.zip'),
             'file_size': 1000,
         })
         assert resp.status_code == 200
@@ -77,8 +77,8 @@ def test_dataset_api_check_quota(self):
         resp = self.client.post(reverse("data-list"), {
             'name': 'new-file-test',
             'type': Data.COMPETITION_BUNDLE,
-            'request_sassy_file_name': faker.file_name(),
-            'file_name': faker.file_name(),
+            'request_sassy_file_name': faker.file_name(extension='.zip'),
+            'file_name': faker.file_name(extension='.zip'),
             'file_size': file_size,
         })
 
@@ -90,12 +90,28 @@ def test_dataset_api_check_quota(self):
         resp = self.client.post(reverse("data-list"), {
             'name': 'new-file-test',
             'type': Data.COMPETITION_BUNDLE,
-            'request_sassy_file_name': faker.file_name(),
+            'request_sassy_file_name': faker.file_name(extension='.zip'),
             'file_name': faker.file_name(),
             'file_size': file_size,
         })
         assert resp.status_code == 201
 
+    def test_dataset_api_rejects_non_zip_files(self):
+        self.client.login(username='creator', password='creator')
+
+        # Attempt to upload a non-zip file
+        resp = self.client.post(reverse("data-list"), {
+            'name': 'non-zip-test',
+            'type': Data.COMPETITION_BUNDLE,
+            'request_sassy_file_name': faker.file_name(extension='.py'),
+            'file_name': faker.file_name(extension='.py'),
+            'file_size': 1000,
+        })
+
+        assert resp.status_code == 400
+        assert "non_field_errors" in resp.data
+        assert resp.data["non_field_errors"][0] == "Only zip files are allowed!"
+
 
 class DatasetDetailTests(TestCase):
     def setUp(self):