test: fix test (duplicate project JSON schema now yields 400)

bento-platform · Jul 18, 2024 · f2ab668 · f2ab668
1 parent d2b4c50
commit f2ab668
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/chord_metadata_service/chord/tests/test_api.py b/chord_metadata_service/chord/tests/test_api.py
@@ -1,5 +1,4 @@
 import json
-from django.db.utils import IntegrityError
 
 from django.urls import reverse
 from rest_framework import status
@@ -187,8 +186,9 @@ def test_create_constraint(self):
                              data=json.dumps(self.project_json_schema_valid_payload),
                              content_type="application/json")
         self.assertEqual(r.status_code, status.HTTP_201_CREATED)
-        with self.assertRaises(IntegrityError):
-            r_duplicate = self.client.post('/api/project_json_schemas',
-                                           data=json.dumps(self.project_json_schema_valid_payload),
-                                           content_type="application/json")
-            self.assertEqual(r_duplicate, status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+        r_duplicate = self.client.post('/api/project_json_schemas',
+                                       data=json.dumps(self.project_json_schema_valid_payload),
+                                       content_type="application/json")
+        # used to be an IntegrityError raised; upgrade to DRF 3.15 made this a 400:
+        self.assertEqual(r_duplicate.status_code, status.HTTP_400_BAD_REQUEST)