From: Sarah Hoffmann <lonvia@denofr.de>
Date: Sat, 1 May 2021 08:50:39 +0000 (+0200)
Subject: mock tokenizer factory for replication tests
X-Git-Tag: v4.0.0~93^2
X-Git-Url: https://git.openstreetmap.org./nominatim.git/commitdiff_plain/8bdb9aa607ef1edd658559561c145e23b3f21010?ds=sidebyside;hp=-c

mock tokenizer factory for replication tests
---

8bdb9aa607ef1edd658559561c145e23b3f21010
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()