]> git.openstreetmap.org Git - nominatim.git/blob - test/python/test_db_sql_preprocessor.py
Merge pull request #2200 from lonvia/migrations-for-current-version
[nominatim.git] / test / python / test_db_sql_preprocessor.py
1 """
2 Tests for SQL preprocessing.
3 """
4 from pathlib import Path
5
6 import pytest
7
8 from nominatim.db.sql_preprocessor import SQLPreprocessor
9
10 @pytest.fixture
11 def sql_factory(tmp_path):
12     def _mk_sql(sql_body):
13         (tmp_path / 'test.sql').write_text("""
14           CREATE OR REPLACE FUNCTION test() RETURNS TEXT
15           AS $$
16           BEGIN
17             {}
18           END;
19           $$ LANGUAGE plpgsql IMMUTABLE;""".format(sql_body))
20         return 'test.sql'
21
22     return _mk_sql
23
24
25 @pytest.fixture
26 def sql_preprocessor(temp_db_conn, tmp_path, def_config, monkeypatch, table_factory):
27     monkeypatch.setenv('NOMINATIM_DATABASE_MODULE_PATH', '.')
28     table_factory('country_name', 'partition INT', (0, 1, 2))
29     return SQLPreprocessor(temp_db_conn, def_config, tmp_path)
30
31 @pytest.mark.parametrize("expr,ret", [
32     ("'a'", 'a'),
33     ("'{{db.partitions|join}}'", '012'),
34     ("{% if 'country_name' in db.tables %}'yes'{% else %}'no'{% endif %}", "yes"),
35     ("{% if 'xxx' in db.tables %}'yes'{% else %}'no'{% endif %}", "no"),
36     ("'{{config.DATABASE_MODULE_PATH}}'", '.')
37     ])
38 def test_load_file_simple(sql_preprocessor, sql_factory, temp_db_conn, temp_db_cursor, expr, ret):
39     sqlfile = sql_factory("RETURN {};".format(expr))
40
41     sql_preprocessor.run_sql_file(temp_db_conn, sqlfile)
42
43     assert temp_db_cursor.scalar('SELECT test()') == ret
44
45
46 def test_load_file_with_params(sql_preprocessor, sql_factory, temp_db_conn, temp_db_cursor):
47     sqlfile = sql_factory("RETURN '{{ foo }} {{ bar }}';")
48
49     sql_preprocessor.run_sql_file(temp_db_conn, sqlfile, bar='XX', foo='ZZ')
50
51     assert temp_db_cursor.scalar('SELECT test()') == 'ZZ XX'