adjusted tests for --clean-deleted-relations command

[nominatim.git] / test / python / cli / test_cmd_admin.py

diff --git a/test/python/cli/test_cmd_admin.py b/test/python/cli/test_cmd_admin.py
index 75ae3cd2a3bc9921b56b3ee0f950382874c6ec61..0fe6e3270fa328acba9420e2bed3f386d2d7d3fb 100644 (file)
--- a/test/python/cli/test_cmd_admin.py
+++ b/test/python/cli/test_cmd_admin.py
@@ -33,6 +33,18 @@ def test_admin_migrate(cli_call, mock_func_factory):
     assert mock.called == 1
 
 
+def test_admin_clean_deleted_relations(cli_call, mock_func_factory):
+    mock = mock_func_factory(nominatim.tools.admin, 'clean_deleted_relations')
+
+    assert cli_call('admin', '--clean-deleted', '--age', '1 month') == 0
+    assert mock.called == 1
+
+def test_admin_clean_deleted_relations_no_age(cli_call, mock_func_factory):
+    mock = mock_func_factory(nominatim.tools.admin, 'clean_deleted_relations')
+
+    with pytest.raises(SystemExit):
+        cli_call('admin', '--clean-deleted')
+
 class TestCliAdminWithDb:
 
     @pytest.fixture(autouse=True)