diff options
Diffstat (limited to 'archaeological_finds/tests.py')
| -rw-r--r-- | archaeological_finds/tests.py | 13 |
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) |