Merge pull request #3523 from mtmail/import-styles-country-code

[nominatim.git] / test / python / tools / test_freeze.py

diff --git a/test/python/tools/test_freeze.py b/test/python/tools/test_freeze.py
index 19a06a9f7a49041c81d603f5ede9c57869515fd6..f64850fb2e99312aade36ebbd404c4f23020a132 100644 (file)
--- a/test/python/tools/test_freeze.py
+++ b/test/python/tools/test_freeze.py
@@ -1,7 +1,13 @@
+# SPDX-License-Identifier: GPL-3.0-or-later
+#
+# This file is part of Nominatim. (https://nominatim.org)
+#
+# Copyright (C) 2024 by the Nominatim developer community.
+# For a full list of authors see the git log.

 """
 Tests for freeze functions (removing unused database parts).
 """
 """
 Tests for freeze functions (removing unused database parts).
 """

-from nominatim.tools import freeze
+from nominatim_db.tools import freeze

 
 NOMINATIM_RUNTIME_TABLES = [
     'country_name', 'country_osm_grid',
 
 NOMINATIM_RUNTIME_TABLES = [
     'country_name', 'country_osm_grid',


@@ -24,6 +30,8 @@ def test_drop_tables(temp_db_conn, temp_db_cursor, table_factory):
     for table in NOMINATIM_RUNTIME_TABLES + NOMINATIM_DROP_TABLES:
         table_factory(table)
 
     for table in NOMINATIM_RUNTIME_TABLES + NOMINATIM_DROP_TABLES:
         table_factory(table)
 

+    assert not freeze.is_frozen(temp_db_conn)
+

     freeze.drop_update_tables(temp_db_conn)
 
     for table in NOMINATIM_RUNTIME_TABLES:
     freeze.drop_update_tables(temp_db_conn)
 
     for table in NOMINATIM_RUNTIME_TABLES:


@@ -32,18 +40,20 @@ def test_drop_tables(temp_db_conn, temp_db_cursor, table_factory):
     for table in NOMINATIM_DROP_TABLES:
         assert not temp_db_cursor.table_exists(table)
 
     for table in NOMINATIM_DROP_TABLES:
         assert not temp_db_cursor.table_exists(table)
 

+    assert freeze.is_frozen(temp_db_conn)
+

 def test_drop_flatnode_file_no_file():
 def test_drop_flatnode_file_no_file():

-    freeze.drop_flatnode_file('')
+    freeze.drop_flatnode_file(None)

 
 
 def test_drop_flatnode_file_file_already_gone(tmp_path):
 
 
 def test_drop_flatnode_file_file_already_gone(tmp_path):

-    freeze.drop_flatnode_file(str(tmp_path / 'something.store'))
+    freeze.drop_flatnode_file(tmp_path / 'something.store')

 
 
 
 

-def test_drop_flatnode_file_delte(tmp_path):
+def test_drop_flatnode_file_delete(tmp_path):

     flatfile = tmp_path / 'flatnode.store'
     flatfile.write_text('Some content')
 
     flatfile = tmp_path / 'flatnode.store'
     flatfile.write_text('Some content')
 

-    freeze.drop_flatnode_file(str(flatfile))
+    freeze.drop_flatnode_file(flatfile)

 
     assert not flatfile.exists()
 
     assert not flatfile.exists()