add of get_geo_items in API + test

1 files changed, 24 insertions, 7 deletions

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index e06d5faa0..ad78646b9 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -2185,11 +2185,13 @@ class GeomaticTest(FindInit, TestCase):
 
     def setUp(self):
         self.create_finds(data_base={"label": "Find 1"}, force=True)
-
-    def test_point_precision(self):
-        self.username = "myuser"
         self.password = "mypassword"
+        self.username = "myuser"
         User.objects.create_superuser(self.username, "myemail@test.com", self.password)
+        self.client = Client()
+        self.client.login(username=self.username, password=self.password)
+
+    def test_point_precision(self):
         profile, created = IshtarSiteProfile.objects.get_or_create(
             slug="default", active=True
         )
@@ -2205,14 +2207,13 @@ class GeomaticTest(FindInit, TestCase):
         base_find.save()
 
         find = base_find.find.all()[0]
-        c = Client()
-        c.login(username=self.username, password=self.password)
-        response = c.get(reverse("show-find", kwargs={"pk": find.pk}))
+        response = self.client.get(reverse("show-find", kwargs={"pk": find.pk}))
         self.assertIn(b"33.12999", response.content)
+
         profile.point_precision = 2
         profile.save()
 
-        response = c.get(reverse("show-find", kwargs={"pk": find.pk}))
+        response = self.client.get(reverse("show-find", kwargs={"pk": find.pk}))
         self.assertIn(b"33.13", response.content)
 
     def test_update_container_localisation_on_warehouse_update(self):
@@ -2535,6 +2536,12 @@ class GeomaticTest(FindInit, TestCase):
         res = base_find.get_geo_items(get_polygons=False)
         self.assertEqual(dic, res)
 
+            # test API
+        find = base_find.find.all()[0]
+        response = self.client.get(reverse("api-get-geo-items", kwargs={"pk": find.pk}))
+        self.assertEqual(response.status_code, 200)
+        self.assertIn(json.dumps(dic).encode('utf-8'), response.content)
+
         # with multi-polygon
         base_find = models.BaseFind.objects.get(pk=base_find.pk)
         limit = (
@@ -2575,6 +2582,16 @@ class GeomaticTest(FindInit, TestCase):
         res_centroid = base_find.get_geo_items(get_polygons=False)
         self.assertEqual(dict_centroid, res_centroid)
 
+            # test API
+        response = self.client.get(reverse("api-get-geo-items", kwargs={"pk": base_find.pk, "get_polygons": True}))
+        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-items", kwargs={"pk": base_find.pk, "get_polygons": False}))
+        self.assertEqual(response.status_code, 200)
+        self.assertIn(json.dumps(dict_centroid).encode('utf-8'), response.content)
+
+
+
 class AutocompleteTest(AutocompleteTestBase, TestCase):
     fixtures = FIND_FIXTURES
     models = [