05/23 Phase 1

tests/test_database.py

@@ -0,0 +1,144 @@
+"""
+tests/test_database.py
+----------------------
+Tests for db/database.py:
+  - init_db() creates all required tables
+  - Default games are seeded (Powerball, Mega Millions)
+  - init_db() is idempotent (safe to call multiple times)
+  - get_db_stats() returns correct draw counts
+"""
+
+import pytest
+from db.database import init_db, get_db_stats, get_connection
+
+
+def test_tables_created(tmp_db):
+    """All 4 tables must exist after init_db()."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' ORDER BY name")
+    tables = {row["name"] for row in cursor.fetchall()}
+    conn.close()
+
+    assert "games" in tables, "Missing table: games"
+    assert "draws" in tables, "Missing table: draws"
+    assert "predictions" in tables, "Missing table: predictions"
+    assert "fetch_log" in tables, "Missing table: fetch_log"
+
+
+def test_default_games_seeded(tmp_db):
+    """Powerball and Mega Millions must be seeded after init_db()."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT name FROM games ORDER BY name")
+    names = [row["name"] for row in cursor.fetchall()]
+    conn.close()
+
+    assert "Powerball" in names, "Powerball not seeded"
+    assert "Mega Millions" in names, "Mega Millions not seeded"
+
+
+def test_powerball_config(tmp_db):
+    """Powerball config must match spec: 5 balls 1-69, bonus 1-26."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM games WHERE name = 'Powerball'")
+    row = cursor.fetchone()
+    conn.close()
+
+    assert row is not None
+    assert row["main_count"] == 5
+    assert row["main_max"] == 69
+    assert row["bonus_count"] == 1
+    assert row["bonus_max"] == 26
+    assert row["active"] == 1
+
+
+def test_megamillions_config(tmp_db):
+    """Mega Millions config must match spec: 5 balls 1-70, bonus 1-25."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM games WHERE name = 'Mega Millions'")
+    row = cursor.fetchone()
+    conn.close()
+
+    assert row is not None
+    assert row["main_count"] == 5
+    assert row["main_max"] == 70
+    assert row["bonus_count"] == 1
+    assert row["bonus_max"] == 25
+    assert row["active"] == 1
+
+
+def test_init_db_idempotent(tmp_db):
+    """Calling init_db() multiple times must not raise or duplicate games."""
+    init_db()
+    init_db()
+
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT COUNT(*) as cnt FROM games")
+    count = cursor.fetchone()["cnt"]
+    conn.close()
+
+    assert count == 2, f"Expected 2 games, got {count} — possible duplicate seed"
+
+
+def test_unique_index_on_draws(tmp_db):
+    """Unique index on (game_id, draw_date) must prevent duplicate inserts."""
+    conn = get_connection()
+    cursor = conn.cursor()
+
+    # Get Powerball id
+    cursor.execute("SELECT id FROM games WHERE name = 'Powerball'")
+    pb_id = cursor.fetchone()["id"]
+
+    # First insert — should succeed
+    cursor.execute("""
+        INSERT INTO draws (game_id, draw_date, numbers, bonus, source)
+        VALUES (?, ?, ?, ?, ?)
+    """, (pb_id, "2024-03-01", "5,12,33,47,65", "8", "test"))
+    conn.commit()
+
+    # Duplicate insert — must raise IntegrityError
+    import sqlite3
+    with pytest.raises(sqlite3.IntegrityError):
+        cursor.execute("""
+            INSERT INTO draws (game_id, draw_date, numbers, bonus, source)
+            VALUES (?, ?, ?, ?, ?)
+        """, (pb_id, "2024-03-01", "1,2,3,4,5", "9", "test"))
+        conn.commit()
+
+    conn.close()
+
+
+def test_get_db_stats_empty(tmp_db):
+    """get_db_stats() returns 0 draw count for all games when DB is empty."""
+    stats = get_db_stats()
+    assert "Powerball" in stats
+    assert "Mega Millions" in stats
+    assert stats["Powerball"] == 0
+    assert stats["Mega Millions"] == 0
+
+
+def test_get_db_stats_with_draws(tmp_db):
+    """get_db_stats() returns correct count after inserting draws."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT id FROM games WHERE name = 'Powerball'")
+    pb_id = cursor.fetchone()["id"]
+
+    cursor.execute("""
+        INSERT INTO draws (game_id, draw_date, numbers, bonus)
+        VALUES (?, ?, ?, ?)
+    """, (pb_id, "2024-01-06", "5,12,33,47,65", "8"))
+    cursor.execute("""
+        INSERT INTO draws (game_id, draw_date, numbers, bonus)
+        VALUES (?, ?, ?, ?)
+    """, (pb_id, "2024-01-10", "2,19,30,44,58", "14"))
+    conn.commit()
+    conn.close()
+
+    stats = get_db_stats()
+    assert stats["Powerball"] == 2
+    assert stats["Mega Millions"] == 0