OldInsuranceMaps/tests/test_api.py at 47b83f8828a19593d98ddf5a6fd3c789e8e5d2fc · ohmg-dev/OldInsuranceMaps · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import json

from django.test import tag

from .base import OHMGTestCase


@tag("api")
class APITestCase(OHMGTestCase):
    uploaded_files = [("regions", OHMGTestCase.Files.new_iberia_p1__1)]

    fixtures = [
        OHMGTestCase.Fixtures.admin_user,
        OHMGTestCase.Fixtures.layerset_categories,
        OHMGTestCase.Fixtures.layerset_categories_sanborn,
        OHMGTestCase.Fixtures.region_categories,
        OHMGTestCase.Fixtures.region_categories_sanborn,
        OHMGTestCase.Fixtures.new_iberia_place,
        OHMGTestCase.Fixtures.new_iberia_map,
        OHMGTestCase.Fixtures.new_iberia_docs,
        OHMGTestCase.Fixtures.gcps_new_iberia_p1__1,
        OHMGTestCase.Fixtures.gcpgroup_new_iberia_p1__1,
        OHMGTestCase.Fixtures.new_iberia_reg_1__1_georef,
        OHMGTestCase.Fixtures.new_iberia_reg_1__2,
        OHMGTestCase.Fixtures.new_iberia_reg_1__3,
        OHMGTestCase.Fixtures.new_iberia_reg_2,
        OHMGTestCase.Fixtures.new_iberia_main_layerset,
        OHMGTestCase.Fixtures.new_iberia_lyr,
        OHMGTestCase.Fixtures.prepsession_new_iberia_p1_split,
        OHMGTestCase.Fixtures.prepsession_new_iberia_p2_no_split,
        OHMGTestCase.Fixtures.georef_session_new_iberia_p1__1,
    ]

    def test_users_endpointt(self):
        response = self.get_api_client().get("/api/beta2/users/")
        self.assertEqual(response.status_code, 200)

    def test_places_endpoint(self):
        response = self.get_api_client().get("/api/beta2/places/")
        self.assertEqual(response.status_code, 200)

    def test_places_geojson_endpoint(self):
        response = self.get_api_client().get("/api/beta2/places/geojson/")
        self.assertEqual(response.status_code, 200)

    def test_map_endpoint(self):
        response = self.get_api_client().get("/api/beta2/map/?map=sanborn03375_001")
        self.assertEqual(response.status_code, 200)

    def test_maps_endpoint(self):
        response = self.get_api_client().get("/api/beta2/maps2/")
        self.assertEqual(response.status_code, 200)

        data = json.loads(response.content)
        self.assertEqual(len(data["items"]), 1)

        response = self.get_api_client().get("/api/beta2/maps2/", {"place": "new-iberia-la"})
        self.assertEqual(response.status_code, 200)

        data = json.loads(response.content)
        self.assertEqual(len(data["items"]), 1)

    def test_get_sessions_endpoint(self):
        response = self.get_api_client().get("/api/beta2/sessions/")
        self.assertEqual(response.status_code, 200)

        data = json.loads(response.content)
        self.assertEqual(data["count"], 3)

        response2 = self.get_api_client().get(
            "/api/beta2/sessions/?offset=0&limit=10&date_range=2024-12-28,2024-12-29"
        )

        data2 = json.loads(response2.content)
        self.assertEqual(data2["count"], 0)

        response3 = self.get_api_client().get(
            "/api/beta2/sessions/?offset=0&limit=10&date_range=2024-12-30,2024-12-30"
        )

        data3 = json.loads(response3.content)
        self.assertEqual(data3["count"], 3)

    def test_get_layerset_endpoint(self):
        response = self.get_api_client().get(
            "/api/beta2/layerset/?map=sanborn03375_001&category=main-content"
        )
        self.assertEqual(response.status_code, 200)

        data = json.loads(response.content)
        self.assertEqual(data["map_id"], "sanborn03375_001")