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