fix(tests): update status code assertions and remove redundant test c…

…ases

backend/tests/test_endpoints.py

@@ -115,7 +115,7 @@ def test_create_training(self):
         res = self.client.post(
             f"{API_BASE}/label/osm/fetch/{self.aoi.id}/", "", headers=headersList
         )
-        self.assertEqual(res.status_code, status.HTTP_201_CREATED)
+        self.assertEqual(res.status_code, 201)
 
         # download labels from osm for 2
 
@@ -309,43 +309,3 @@ def test_get_workspace(self):
 
         res = self.client.get(f"{API_BASE}/workspace/dataset_1/", headers=headersList)
         self.assertEqual(res.status_code, 404)
-
-    def test_download_workspace(self):
-        try:
-            lookup_dir = "test_dir"
-
-            # download non-existent dir should fail
-            res = self.client.get(
-                f"{API_BASE}/workspace/download/{lookup_dir}", headers=headersList
-            )
-            self.assertEqual(res.status_code, status.HTTP_404_NOT_FOUND)
-
-            # test download workspace
-            base_dir = os.path.join(settings.TRAINING_WORKSPACE, lookup_dir)
-            os.makedirs(base_dir)
-
-            with open(os.path.join(base_dir, "file.txt"), "wb") as f:
-                f.write(b"Test file")
-
-            res = self.client.get(
-                f"{API_BASE}/workspace/download/{lookup_dir}",
-                headers=headersList,
-            )
-            self.assertEqual(res.status_code, status.HTTP_200_OK)
-
-            # test download file greater than the 200 mb limit
-
-            with open(os.path.join(base_dir, "large_file.txt"), "wb") as f:
-                f.seek(201 * 1024**2)
-                f.write(b"\0")
-
-            # download file size greater than limit should fail
-            res = self.client.get(
-                f"{API_BASE}/workspace/download/{lookup_dir}",
-                headers=headersList,
-            )
-            self.assertEqual(res.status_code, status.HTTP_403_FORBIDDEN)
-
-        finally:
-            # clean up
-            shutil.rmtree(base_dir)