port check-database function to python

[nominatim.git] / test / python / test_db_status.py

diff --git a/test/python/test_db_status.py b/test/python/test_db_status.py
index 399a0036ba96186597deb1038dd14f9e60da0082..c659147148d4b2d970a5008e984e8fce02cc7242 100644 (file)
--- a/test/python/test_db_status.py
+++ b/test/python/test_db_status.py
@@ -65,7 +65,6 @@ def test_set_status_filled_table(status_table, temp_db_conn, temp_db_cursor):
 
     assert 1 == temp_db_cursor.scalar("SELECT count(*) FROM import_status")
 
 
     assert 1 == temp_db_cursor.scalar("SELECT count(*) FROM import_status")
 

-

     date = dt.datetime.fromordinal(1000100).replace(tzinfo=dt.timezone.utc)
     nominatim.db.status.set_status(temp_db_conn, date=date, seq=456, indexed=False)
 
     date = dt.datetime.fromordinal(1000100).replace(tzinfo=dt.timezone.utc)
     nominatim.db.status.set_status(temp_db_conn, date=date, seq=456, indexed=False)
 


@@ -75,6 +74,20 @@ def test_set_status_filled_table(status_table, temp_db_conn, temp_db_cursor):
     assert temp_db_cursor.fetchone() == [date, 456, False]
 
 
     assert temp_db_cursor.fetchone() == [date, 456, False]
 
 

+def test_set_status_missing_date(status_table, temp_db_conn, temp_db_cursor):
+    date = dt.datetime.fromordinal(1000000).replace(tzinfo=dt.timezone.utc)
+    nominatim.db.status.set_status(temp_db_conn, date=date)
+
+    assert 1 == temp_db_cursor.scalar("SELECT count(*) FROM import_status")
+
+    nominatim.db.status.set_status(temp_db_conn, date=None, seq=456, indexed=False)
+
+    temp_db_cursor.execute("SELECT * FROM import_status")
+
+    assert temp_db_cursor.rowcount == 1
+    assert temp_db_cursor.fetchone() == [date, 456, False]
+
+

 def test_get_status_empty_table(status_table, temp_db_conn):
     assert nominatim.db.status.get_status(temp_db_conn) == (None, None, None)
 
 def test_get_status_empty_table(status_table, temp_db_conn):
     assert nominatim.db.status.get_status(temp_db_conn) == (None, None, None)