test/python/cli/test_cmd_admin.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) 2024 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Test for the command line interface wrapper admin subcommand.
   9
  10 These tests just check that the various command line parameters route to the
  11 correct functionality. They use a lot of monkeypatching to avoid executing
  12 the actual functions.
  13 """
  14 import pytest
  15
  16 import nominatim_db.tools.admin
  17 import nominatim_db.tools.check_database
  18 import nominatim_db.tools.migration
  19 import nominatim_db.clicmd.admin
  20
  21
  22 def test_admin_command_check_database(cli_call, mock_func_factory):
  23     mock = mock_func_factory(nominatim_db.tools.check_database, 'check_database')
  24
  25     assert cli_call('admin', '--check-database') == 0
  26     assert mock.called == 1
  27
  28
  29 def test_admin_migrate(cli_call, mock_func_factory):
  30     mock = mock_func_factory(nominatim_db.tools.migration, 'migrate')
  31
  32     assert cli_call('admin', '--migrate') == 0
  33     assert mock.called == 1
  34
  35
  36 def test_admin_clean_deleted_relations(cli_call, mock_func_factory):
  37     mock = mock_func_factory(nominatim_db.tools.admin, 'clean_deleted_relations')
  38
  39     assert cli_call('admin', '--clean-deleted', '1 month') == 0
  40     assert mock.called == 1
  41
  42 def test_admin_clean_deleted_relations_no_age(cli_call, mock_func_factory):
  43     mock = mock_func_factory(nominatim_db.tools.admin, 'clean_deleted_relations')
  44
  45     assert cli_call('admin', '--clean-deleted') == 1
  46
  47 class TestCliAdminWithDb:
  48
  49     @pytest.fixture(autouse=True)
  50     def setup_cli_call(self, cli_call, temp_db, cli_tokenizer_mock):
  51         self.call_nominatim = cli_call
  52         self.tokenizer_mock = cli_tokenizer_mock
  53
  54
  55     @pytest.mark.parametrize("func, params", [('analyse_indexing', ('--analyse-indexing', ))])
  56     def test_analyse_indexing(self, mock_func_factory, func, params):
  57         mock = mock_func_factory(nominatim_db.tools.admin, func)
  58
  59         assert self.call_nominatim('admin', *params) == 0
  60         assert mock.called == 1