test/python/cli/test_cmd_admin.py

   1 """
   2 Test for the command line interface wrapper admin subcommand.
   3
   4 These tests just check that the various command line parameters route to the
   5 correct functionionality. They use a lot of monkeypatching to avoid executing
   6 the actual functions.
   7 """
   8 import pytest
   9
  10 import nominatim.tools.admin
  11 import nominatim.tools.check_database
  12 import nominatim.tools.migration
  13 import nominatim.clicmd.admin
  14
  15
  16 @pytest.mark.parametrize("params", [('--warm', ),
  17                                     ('--warm', '--reverse-only'),
  18                                     ('--warm', '--search-only')])
  19 def test_admin_command_legacy(cli_call, mock_func_factory, params):
  20     mock_run_legacy = mock_func_factory(nominatim.clicmd.admin, 'run_legacy_script')
  21
  22     assert cli_call('admin', *params) == 0
  23
  24     assert mock_run_legacy.called == 1
  25
  26
  27 def test_admin_command_check_database(cli_call, mock_func_factory):
  28     mock = mock_func_factory(nominatim.tools.check_database, 'check_database')
  29
  30     assert cli_call('admin', '--check-database') == 0
  31     assert mock.called == 1
  32
  33
  34 def test_admin_migrate(cli_call, mock_func_factory):
  35     mock = mock_func_factory(nominatim.tools.migration, 'migrate')
  36
  37     assert cli_call('admin', '--migrate') == 0
  38     assert mock.called == 1
  39
  40
  41 class TestCliAdminWithDb:
  42
  43     @pytest.fixture(autouse=True)
  44     def setup_cli_call(self, cli_call, temp_db, cli_tokenizer_mock):
  45         self.call_nominatim = cli_call
  46         self.tokenizer_mock = cli_tokenizer_mock
  47
  48
  49     @pytest.mark.parametrize("func, params", [('analyse_indexing', ('--analyse-indexing', ))])
  50     def test_analyse_indexing(self, mock_func_factory, func, params):
  51         mock = mock_func_factory(nominatim.tools.admin, func)
  52
  53         assert self.call_nominatim('admin', *params) == 0
  54         assert mock.called == 1