X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/4cd0a4ced47c1e034194956a01418fa5884ddc3c..3742fa2929619a4c54a50d3e79e0eeadb4d6ca6f:/test/python/cli/conftest.py?ds=sidebyside diff --git a/test/python/cli/conftest.py b/test/python/cli/conftest.py index d4e1118e..1e3ca8ab 100644 --- a/test/python/cli/conftest.py +++ b/test/python/cli/conftest.py @@ -1,12 +1,12 @@ -# SPDX-License-Identifier: GPL-2.0-only +# SPDX-License-Identifier: GPL-3.0-or-later # # This file is part of Nominatim. (https://nominatim.org) # -# Copyright (C) 2022 by the Nominatim developer community. +# Copyright (C) 2024 by the Nominatim developer community. # For a full list of authors see the git log. import pytest -import nominatim.cli +import nominatim_db.cli class MockParamCapture: """ Mock that records the parameters with which a function was called @@ -32,29 +32,28 @@ class DummyTokenizer: self.update_statistics_called = False self.update_word_tokens_called = False - def update_sql_functions(self, *args): + def update_sql_functions(self, *args, **kwargs): self.update_sql_functions_called = True - def finalize_import(self, *args): + def finalize_import(self, *args, **kwargs): self.finalize_import_called = True - def update_statistics(self): + def update_statistics(self, *args, **kwargs): self.update_statistics_called = True - def update_word_tokens(self): + def update_word_tokens(self, *args, **kwargs): self.update_word_tokens_called = True @pytest.fixture -def cli_call(src_dir): +def cli_call(): """ Call the nominatim main function with the correct paths set. Returns a function that can be called with the desired CLI arguments. """ def _call_nominatim(*args): - return nominatim.cli.nominatim(module_dir='MODULE NOT AVAILABLE', - osm2pgsql_path='OSM2PGSQL NOT AVAILABLE', - phpcgi_path='/usr/bin/php-cgi', - cli_args=args) + return nominatim_db.cli.nominatim(module_dir='MODULE NOT AVAILABLE', + osm2pgsql_path='OSM2PGSQL NOT AVAILABLE', + cli_args=args) return _call_nominatim @@ -73,9 +72,9 @@ def mock_func_factory(monkeypatch): @pytest.fixture def cli_tokenizer_mock(monkeypatch): tok = DummyTokenizer() - monkeypatch.setattr(nominatim.tokenizer.factory, 'get_tokenizer_for_db', + monkeypatch.setattr(nominatim_db.tokenizer.factory, 'get_tokenizer_for_db', lambda *args: tok) - monkeypatch.setattr(nominatim.tokenizer.factory, 'create_tokenizer', + monkeypatch.setattr(nominatim_db.tokenizer.factory, 'create_tokenizer', lambda *args: tok) return tok