From ee4918a011a0a4d77c5e7befda0fefb719877207 Mon Sep 17 00:00:00 2001 From: Quentin André Date: Fri, 9 Jul 2021 15:07:42 +0200 Subject: add of get_geo_items in API + test --- archaeological_finds/tests.py | 31 ++++++++++++++++++++++++------- 1 file changed, 24 insertions(+), 7 deletions(-) (limited to 'archaeological_finds/tests.py') diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index 48f0749b9..2a98ef381 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -2203,11 +2203,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 ) @@ -2223,14 +2225,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): @@ -2553,6 +2554,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 = ( @@ -2593,6 +2600,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 = [ -- cgit v1.2.3