fix deprecated assertDictContainsSubset

Terralego · Nov 25, 2024 · 59f23d6 · 59f23d6
1 parent fca1fc7
commit 59f23d6
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/project/geosource/tests/test_api.py b/project/geosource/tests/test_api.py
@@ -144,9 +144,9 @@ def test_postgis_source_creation(self):
             {**self.source_example, "db_password": "test_password"},
             format="json",
         )
-
+        data = response.json()
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
-        self.assertDictContainsSubset(self.source_example, response.json())
+        self.assertEqual(data, data | self.source_example)
 
     @patch(
         "project.geosource.serializers.PostGISSourceSerializer._first_record",
@@ -188,7 +188,8 @@ def test_postgis_source_creation_no_geom_field_good_geom(self):
         )
         self.source_example["geom_field"] = "foo"
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
-        self.assertDictContainsSubset(self.source_example, response.json())
+        data = response.json()
+        self.assertEqual(data, data | self.source_example)
 
     @patch(
         "project.geosource.serializers.requests.get",
@@ -209,7 +210,8 @@ def test_wmts_source_creation(self, mocked_request_get):
         )
 
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
-        self.assertDictContainsSubset(wmts_source, response.json())
+        data = response.json()
+        self.assertEqual(data, data | wmts_source)
 
     @patch(
         "project.geosource.serializers.PostGISSourceSerializer._first_record",