test/python/cli/test_cmd_admin.py

   1 # SPDX-License-Identifier: GPL-2.0-only
   2 #
   3 # This file is part of Nominatim. (https://nominatim.org)
   4 #
   5 # Copyright (C) 2022 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 functionionality. They use a lot of monkeypatching to avoid executing
  12 the actual functions.
  13 """
  14 import pytest
  15
  16 import nominatim.tools.admin
  17 import nominatim.tools.check_database
  18 import nominatim.tools.migration
  19 import nominatim.clicmd.admin
  20
  21
  22 def test_admin_command_check_database(cli_call, mock_func_factory):
  23     mock = mock_func_factory(nominatim.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.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.tools.admin, 'clean_deleted_relations')
  38
  39     assert cli_call('admin', '--clean-deleted', '--age', '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.tools.admin, 'clean_deleted_relations')
  44
  45     with pytest.raises(SystemExit):
  46         cli_call('admin', '--clean-deleted')
  47
  48 class TestCliAdminWithDb:
  49
  50     @pytest.fixture(autouse=True)
  51     def setup_cli_call(self, cli_call, temp_db, cli_tokenizer_mock):
  52         self.call_nominatim = cli_call
  53         self.tokenizer_mock = cli_tokenizer_mock
  54
  55
  56     @pytest.mark.parametrize("func, params", [('analyse_indexing', ('--analyse-indexing', ))])
  57     def test_analyse_indexing(self, mock_func_factory, func, params):
  58         mock = mock_func_factory(nominatim.tools.admin, func)
  59
  60         assert self.call_nominatim('admin', *params) == 0
  61         assert mock.called == 1