mirror of
https://github.com/osm-search/Nominatim.git
synced 2024-12-01 22:34:52 +03:00
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
"""
|
|
Tests for property table manpulation.
|
|
"""
|
|
import pytest
|
|
|
|
from nominatim.db import properties
|
|
|
|
@pytest.fixture
|
|
def prop_table(table_factory):
|
|
table_factory('nominatim_properties', 'property TEXT, value TEXT')
|
|
|
|
|
|
def test_get_property_existing(prop_table, temp_db_conn, temp_db_cursor):
|
|
temp_db_cursor.execute("INSERT INTO nominatim_properties VALUES('foo', 'bar')")
|
|
|
|
assert properties.get_property(temp_db_conn, 'foo') == 'bar'
|
|
|
|
|
|
def test_get_property_unknown(prop_table, temp_db_conn, temp_db_cursor):
|
|
temp_db_cursor.execute("INSERT INTO nominatim_properties VALUES('other', 'bar')")
|
|
|
|
assert properties.get_property(temp_db_conn, 'foo') is None
|
|
|
|
|
|
@pytest.mark.parametrize("prefill", (True, False))
|
|
def test_set_property_new(prop_table, temp_db_conn, temp_db_cursor, prefill):
|
|
if prefill:
|
|
temp_db_cursor.execute("INSERT INTO nominatim_properties VALUES('something', 'bar')")
|
|
|
|
properties.set_property(temp_db_conn, 'something', 'else')
|
|
|
|
assert temp_db_cursor.scalar("""SELECT value FROM nominatim_properties
|
|
WHERE property = 'something'""") == 'else'
|
|
|
|
assert properties.get_property(temp_db_conn, 'something') == 'else'
|