]> git.openstreetmap.org Git - nominatim.git/blob - test/python/api/test_api_polygons_v1.py
allow None and str for project_dir in NominatimAPI init
[nominatim.git] / test / python / api / test_api_polygons_v1.py
1 # SPDX-License-Identifier: GPL-3.0-or-later
2 #
3 # This file is part of Nominatim. (https://nominatim.org)
4 #
5 # Copyright (C) 2024 by the Nominatim developer community.
6 # For a full list of authors see the git log.
7 """
8 Tests for the deletable v1 API call.
9 """
10 import json
11 import datetime as dt
12 from pathlib import Path
13
14 import pytest
15
16 from fake_adaptor import FakeAdaptor, FakeError, FakeResponse
17
18 import nominatim_api.v1.server_glue as glue
19
20 class TestPolygonsEndPoint:
21
22     @pytest.fixture(autouse=True)
23     def setup_deletable_table(self, temp_db_cursor, table_factory, temp_db_with_extensions):
24         self.now = dt.datetime.now()
25         self.recent = dt.datetime.now() - dt.timedelta(days=3)
26
27         table_factory('import_polygon_error',
28                       definition="""osm_id bigint,
29                                     osm_type character(1),
30                                     class text,
31                                     type text,
32                                     name hstore,
33                                     country_code character varying(2),
34                                     updated timestamp without time zone,
35                                     errormessage text,
36                                     prevgeometry geometry(Geometry,4326),
37                                     newgeometry geometry(Geometry,4326)""",
38                     content=[(345, 'N', 'boundary', 'administrative',
39                               {'name': 'Foo'}, 'xx', self.recent,
40                               'some text', None, None),
41                              (781, 'R', 'landuse', 'wood',
42                               None, 'ds', self.now,
43                               'Area reduced by lots', None, None)])
44
45
46     @pytest.mark.asyncio
47     async def test_polygons_simple(self, api):
48         a = FakeAdaptor()
49
50         resp = await glue.polygons_endpoint(api, a)
51         results = json.loads(resp.output)
52
53         results.sort(key=lambda r: (r['osm_type'], r['osm_id']))
54
55         assert results == [{'osm_type': 'N', 'osm_id': 345,
56                             'class': 'boundary', 'type': 'administrative',
57                             'name': 'Foo', 'country_code': 'xx',
58                             'errormessage': 'some text',
59                             'updated': self.recent.isoformat(sep=' ', timespec='seconds')},
60                            {'osm_type': 'R', 'osm_id': 781,
61                             'class': 'landuse', 'type': 'wood',
62                             'name': None, 'country_code': 'ds',
63                             'errormessage': 'Area reduced by lots',
64                             'updated': self.now.isoformat(sep=' ', timespec='seconds')}]
65
66
67     @pytest.mark.asyncio
68     async def test_polygons_days(self, api):
69         a = FakeAdaptor()
70         a.params['days'] = '2'
71
72         resp = await glue.polygons_endpoint(api, a)
73         results = json.loads(resp.output)
74
75         assert [r['osm_id'] for r in results] == [781]
76
77
78     @pytest.mark.asyncio
79     async def test_polygons_class(self, api):
80         a = FakeAdaptor()
81         a.params['class'] = 'landuse'
82
83         resp = await glue.polygons_endpoint(api, a)
84         results = json.loads(resp.output)
85
86         assert [r['osm_id'] for r in results] == [781]
87
88
89
90     @pytest.mark.asyncio
91     async def test_polygons_reduced(self, api):
92         a = FakeAdaptor()
93         a.params['reduced'] = '1'
94
95         resp = await glue.polygons_endpoint(api, a)
96         results = json.loads(resp.output)
97
98         assert [r['osm_id'] for r in results] == [781]