more robust import script

tests/test_import_excel.py

@@ -0,0 +1,65 @@
+import sqlite3
+import tempfile
+from pathlib import Path
+
+import pytest
+
+from scripts.import_excel import DEFAULT_DATABASE, import_data, inspect_database, verify_import
+
+
+def database_file():
+    directory = tempfile.TemporaryDirectory()
+    path = Path(directory.name) / "calendar.sqlite"
+    with sqlite3.connect(path) as database:
+        database.execute(
+            "CREATE TABLE date_choices (choice_date TEXT PRIMARY KEY, amount_cents INTEGER, updated_at TEXT DEFAULT CURRENT_TIMESTAMP)"
+        )
+        database.execute(
+            "CREATE TABLE date_statuses (status_date TEXT PRIMARY KEY, day_status TEXT, updated_at TEXT DEFAULT CURRENT_TIMESTAMP)"
+        )
+    return directory, path
+
+
+def test_default_database_is_anchored_to_project_directory():
+    assert DEFAULT_DATABASE.is_absolute()
+    assert DEFAULT_DATABASE.name == "calendar.sqlite"
+    assert DEFAULT_DATABASE.parent.name == "instance"
+
+
+def test_database_inspection_rejects_missing_or_wrong_database():
+    with tempfile.TemporaryDirectory() as directory:
+        missing = Path(directory) / "missing.sqlite"
+        with pytest.raises(RuntimeError, match="does not exist"):
+            inspect_database(missing)
+
+        wrong = Path(directory) / "wrong.sqlite"
+        with sqlite3.connect(wrong) as database:
+            database.execute("CREATE TABLE something_else (value TEXT)")
+        with pytest.raises(RuntimeError, match="wrong schema"):
+            inspect_database(wrong)
+
+
+def test_import_creates_backup_and_verifies_committed_values():
+    directory, path = database_file()
+    try:
+        backup = import_data(
+            path,
+            {"2026-07-15": 2750},
+            {"2026-07-16": "vacation"},
+        )
+
+        assert backup.is_file()
+        assert inspect_database(path) == (1, 1, "2026-07-15", "2026-07-16")
+        verify_import(path, {"2026-07-15": 2750}, {"2026-07-16": "vacation"})
+        assert inspect_database(backup) == (0, 0, None, None)
+    finally:
+        directory.cleanup()
+
+
+def test_verification_detects_values_that_do_not_match():
+    directory, path = database_file()
+    try:
+        with pytest.raises(RuntimeError, match="verification failed"):
+            verify_import(path, {"2026-07-15": 5500}, {})
+    finally:
+        directory.cleanup()