From 8bdb9aa607ef1edd658559561c145e23b3f21010 Mon Sep 17 00:00:00 2001 From: Sarah Hoffmann Date: Sat, 1 May 2021 10:50:39 +0200 Subject: [PATCH 1/1] mock tokenizer factory for replication tests --- test/python/test_cli_replication.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/test/python/test_cli_replication.py b/test/python/test_cli_replication.py index d4f8290f..b95e6ede 100644 --- a/test/python/test_cli_replication.py +++ b/test/python/test_cli_replication.py @@ -26,6 +26,28 @@ def call_nominatim(*args): config_dir=str(SRC_DIR / 'settings'), cli_args=['replication'] + list(args)) +@pytest.fixture +def tokenizer_mock(monkeypatch): + class DummyTokenizer: + def __init__(self, *args, **kwargs): + self.update_sql_functions_called = False + self.finalize_import_called = False + + def update_sql_functions(self, *args): + self.update_sql_functions_called = True + + def finalize_import(self, *args): + self.finalize_import_called = True + + tok = DummyTokenizer() + monkeypatch.setattr(nominatim.tokenizer.factory, 'get_tokenizer_for_db' , + lambda *args: tok) + monkeypatch.setattr(nominatim.tokenizer.factory, 'create_tokenizer' , + lambda *args: tok) + + return tok + + @pytest.fixture def index_mock(monkeypatch, tokenizer_mock): mock = MockParamCapture() -- 2.45.2