first commit

backend/tests/test_database.py

@@ -0,0 +1,78 @@
+import sqlite3
+from pathlib import Path
+
+from app.database import get_connection, init_db
+
+
+def test_init_db_creates_tables(tmp_data_dir: Path) -> None:
+    db_path = tmp_data_dir / "handin.db"
+    init_db(db_path)
+
+    conn = sqlite3.connect(db_path)
+    cursor = conn.execute(
+        "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name"
+    )
+    tables = {row[0] for row in cursor.fetchall()}
+    conn.close()
+
+    assert "site" in tables
+    assert "segments" in tables
+
+
+def test_init_db_idempotent(tmp_data_dir: Path) -> None:
+    db_path = tmp_data_dir / "handin.db"
+    init_db(db_path)
+    init_db(db_path)
+
+    conn = sqlite3.connect(db_path)
+    cursor = conn.execute(
+        "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name"
+    )
+    tables = {row[0] for row in cursor.fetchall()}
+    conn.close()
+
+    assert "site" in tables
+    assert "segments" in tables
+
+
+def test_wal_mode_enabled(tmp_data_dir: Path) -> None:
+    db_path = tmp_data_dir / "handin.db"
+    init_db(db_path)
+
+    conn = sqlite3.connect(db_path)
+    cursor = conn.execute("PRAGMA journal_mode")
+    mode = cursor.fetchone()[0]
+    conn.close()
+
+    assert mode == "wal"
+
+
+def test_get_connection_busy_timeout(db: Path) -> None:
+    conn = get_connection(db)
+    cursor = conn.execute("PRAGMA busy_timeout")
+    timeout = cursor.fetchone()[0]
+    conn.close()
+
+    assert timeout == 5000
+
+
+def test_segments_table_schema(db: Path) -> None:
+    conn = sqlite3.connect(db)
+    cursor = conn.execute("PRAGMA table_info(segments)")
+    columns = {row[1] for row in cursor.fetchall()}
+    conn.close()
+
+    expected = {
+        "id", "type", "sort_order", "title",
+        "content", "metadata", "created_at", "updated_at",
+    }
+    assert columns == expected
+
+
+def test_site_table_schema(db: Path) -> None:
+    conn = sqlite3.connect(db)
+    cursor = conn.execute("PRAGMA table_info(site)")
+    columns = {row[1] for row in cursor.fetchall()}
+    conn.close()
+
+    assert columns == {"key", "value"}