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 47ac7e72c..03ad0314e 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -2542,9 +2542,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
@@ -2592,15 +2595,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)