]> git.openstreetmap.org Git - nominatim.git/blob - test/python/cli/test_cmd_admin.py
adjusted tests for --clean-deleted-relations command
[nominatim.git] / 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