test/python/tokenizer/test_factory.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) 2025 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Tests for creating new tokenizers.
   9 """
  10 import pytest
  11
  12 from nominatim_db.db import properties
  13 from nominatim_db.tokenizer import factory
  14 from nominatim_db.errors import UsageError
  15 from dummy_tokenizer import DummyTokenizer
  16
  17
  18 def test_setup_bad_tokenizer_name(project_env, monkeypatch):
  19     monkeypatch.setenv('NOMINATIM_TOKENIZER', 'dummy')
  20
  21     with pytest.raises(UsageError):
  22         factory.create_tokenizer(project_env)
  23
  24
  25 class TestFactory:
  26     @pytest.fixture(autouse=True)
  27     def init_env(self, project_env, property_table, tokenizer_mock):
  28         self.config = project_env
  29
  30     def test_setup_dummy_tokenizer(self, temp_db_conn):
  31         tokenizer = factory.create_tokenizer(self.config)
  32
  33         assert isinstance(tokenizer, DummyTokenizer)
  34         assert tokenizer.init_state == "new"
  35
  36         assert properties.get_property(temp_db_conn, 'tokenizer') == 'dummy'
  37
  38     def test_load_tokenizer(self):
  39         factory.create_tokenizer(self.config)
  40
  41         tokenizer = factory.get_tokenizer_for_db(self.config)
  42
  43         assert isinstance(tokenizer, DummyTokenizer)
  44         assert tokenizer.init_state == "loaded"
  45
  46     def test_load_repopulate_tokenizer_dir(self):
  47         factory.create_tokenizer(self.config)
  48
  49         self.config.project_dir = self.config.project_dir
  50
  51         factory.get_tokenizer_for_db(self.config)
  52
  53     def test_load_missing_property(self, temp_db_cursor):
  54         factory.create_tokenizer(self.config)
  55
  56         temp_db_cursor.execute("TRUNCATE TABLE nominatim_properties")
  57
  58         with pytest.raises(UsageError):
  59             factory.get_tokenizer_for_db(self.config)