mirror of
https://github.com/osm-search/Nominatim.git
synced 2024-11-23 13:44:36 +03:00
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
"""
|
|
Tests for creating new tokenizers.
|
|
"""
|
|
import importlib
|
|
import pytest
|
|
|
|
from nominatim.db import properties
|
|
from nominatim.tokenizer import factory
|
|
from nominatim.errors import UsageError
|
|
from dummy_tokenizer import DummyTokenizer
|
|
|
|
@pytest.fixture
|
|
def test_config(def_config, tmp_path):
|
|
def_config.project_dir = tmp_path
|
|
return def_config
|
|
|
|
|
|
def test_setup_dummy_tokenizer(temp_db_conn, test_config,
|
|
tokenizer_mock, property_table):
|
|
tokenizer = factory.create_tokenizer(test_config)
|
|
|
|
assert isinstance(tokenizer, DummyTokenizer)
|
|
assert tokenizer.init_state == "new"
|
|
assert (test_config.project_dir / 'tokenizer').is_dir()
|
|
|
|
assert properties.get_property(temp_db_conn, 'tokenizer') == 'dummy'
|
|
|
|
|
|
def test_setup_tokenizer_dir_exists(test_config, tokenizer_mock, property_table):
|
|
(test_config.project_dir / 'tokenizer').mkdir()
|
|
|
|
tokenizer = factory.create_tokenizer(test_config)
|
|
|
|
assert isinstance(tokenizer, DummyTokenizer)
|
|
assert tokenizer.init_state == "new"
|
|
|
|
|
|
def test_setup_tokenizer_dir_failure(test_config, tokenizer_mock, property_table):
|
|
(test_config.project_dir / 'tokenizer').write_text("foo")
|
|
|
|
with pytest.raises(UsageError):
|
|
factory.create_tokenizer(test_config)
|
|
|
|
|
|
def test_setup_bad_tokenizer_name(test_config, monkeypatch):
|
|
monkeypatch.setenv('NOMINATIM_TOKENIZER', 'dummy')
|
|
|
|
with pytest.raises(UsageError):
|
|
factory.create_tokenizer(test_config)
|
|
|
|
def test_load_tokenizer(temp_db_conn, test_config,
|
|
tokenizer_mock, property_table):
|
|
factory.create_tokenizer(test_config)
|
|
|
|
tokenizer = factory.get_tokenizer_for_db(test_config)
|
|
|
|
assert isinstance(tokenizer, DummyTokenizer)
|
|
assert tokenizer.init_state == "loaded"
|
|
|
|
|
|
def test_load_no_tokenizer_dir(test_config, tokenizer_mock, property_table):
|
|
factory.create_tokenizer(test_config)
|
|
|
|
test_config.project_dir = test_config.project_dir / 'foo'
|
|
|
|
with pytest.raises(UsageError):
|
|
factory.get_tokenizer_for_db(test_config)
|
|
|
|
|
|
def test_load_missing_propoerty(temp_db_cursor, test_config, tokenizer_mock, property_table):
|
|
factory.create_tokenizer(test_config)
|
|
|
|
temp_db_cursor.execute("TRUNCATE TABLE nominatim_properties")
|
|
|
|
with pytest.raises(UsageError):
|
|
factory.get_tokenizer_for_db(test_config)
|
|
|