correction with kwargs for get_polygons

1 files changed, 8 insertions, 5 deletions

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index cf1f47ba0..ee5420ebe 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -2524,9 +2524,12 @@ class GeomaticTest(FindInit, TestCase):
         base_find = models.BaseFind.objects.get(pk=base_find.pk)
         dic = {
             "type": "Feature",
-            "geometry": {"type": "Point", "coordinates": [2.0, 43.0]},
+            "geometry": {"type": "MultiPolygon", "coordinates": []},
         }
-        res = base_find.get_geo_items()
+        res = base_find.get_geo_items(get_polygons=True)
+        self.assertEqual(dic, res)
+        dic["geometry"] = {"type": "Point", "coordinates": [2.0, 43.0]}
+        res = base_find.get_geo_items(get_polygons=False)
         self.assertEqual(dic, res)
 
         # test API
@@ -2574,15 +2577,15 @@ class GeomaticTest(FindInit, TestCase):
         response = self.client.get(
             reverse(
                 "api-get-geo-polygons",
-                kwargs={"pk": base_find.pk, "get_polygons": True},
+                kwargs={"pk": base_find.pk},
             )
         )
         self.assertEqual(response.status_code, 200)
         self.assertIn(json.dumps(dict_poly).encode("utf-8"), response.content)
         response = self.client.get(
             reverse(
-                "api-get-geo-polygons",
-                kwargs={"pk": base_find.pk, "get_polygons": False},
+                "api-get-geo-point",
+                kwargs={"pk": base_find.pk},
             )
         )
         self.assertEqual(response.status_code, 200)