235 lines
7.6 KiB
Python
235 lines
7.6 KiB
Python
"""
|
|
tests/test_history.py
|
|
---------------------
|
|
Tests for the History screen data layer — get_draws_with_game().
|
|
UI widget instantiation is tested with a hidden Tk root (skipped on headless).
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from db.models import (
|
|
get_draws_with_game,
|
|
get_game_by_name,
|
|
insert_draw,
|
|
)
|
|
|
|
|
|
# ── get_draws_with_game ───────────────────────────────────────────────────────
|
|
|
|
def test_returns_empty_for_fresh_db(tmp_db):
|
|
rows = get_draws_with_game()
|
|
assert rows == []
|
|
|
|
|
|
def test_returns_all_games_when_no_filter(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
mm = get_game_by_name("Mega Millions")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7, source="powerball_ny")
|
|
insert_draw(mm["id"], "2024-01-02", [7, 11, 22, 29, 38], bonus=4, source="megamillions_ny")
|
|
|
|
rows = get_draws_with_game()
|
|
assert len(rows) == 2
|
|
|
|
|
|
def test_game_name_is_included(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7, source="powerball_ny")
|
|
|
|
rows = get_draws_with_game()
|
|
assert rows[0]["game_name"] == "Powerball"
|
|
|
|
|
|
def test_filters_by_game_id(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
mm = get_game_by_name("Mega Millions")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7, source="powerball_ny")
|
|
insert_draw(mm["id"], "2024-01-02", [7, 11, 22, 29, 38], bonus=4, source="megamillions_ny")
|
|
|
|
rows = get_draws_with_game(game_id=pb["id"])
|
|
assert len(rows) == 1
|
|
assert rows[0]["game_name"] == "Powerball"
|
|
|
|
|
|
def test_filters_by_date_from(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
insert_draw(pb["id"], "2024-06-01", [5, 10, 20, 30, 40], bonus=3)
|
|
|
|
rows = get_draws_with_game(date_from="2024-06-01")
|
|
assert len(rows) == 1
|
|
assert rows[0]["draw_date"] == "2024-06-01"
|
|
|
|
|
|
def test_filters_by_date_to(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
insert_draw(pb["id"], "2024-06-01", [5, 10, 20, 30, 40], bonus=3)
|
|
|
|
rows = get_draws_with_game(date_to="2024-01-31")
|
|
assert len(rows) == 1
|
|
assert rows[0]["draw_date"] == "2024-01-01"
|
|
|
|
|
|
def test_filters_by_date_range(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
insert_draw(pb["id"], "2024-03-15", [5, 10, 20, 30, 40], bonus=3)
|
|
insert_draw(pb["id"], "2024-12-01", [9, 18, 27, 36, 45], bonus=15)
|
|
|
|
rows = get_draws_with_game(date_from="2024-02-01", date_to="2024-06-30")
|
|
assert len(rows) == 1
|
|
assert rows[0]["draw_date"] == "2024-03-15"
|
|
|
|
|
|
def test_default_order_is_desc(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
insert_draw(pb["id"], "2024-06-01", [5, 10, 20, 30, 40], bonus=3)
|
|
|
|
rows = get_draws_with_game()
|
|
assert rows[0]["draw_date"] == "2024-06-01"
|
|
assert rows[1]["draw_date"] == "2024-01-01"
|
|
|
|
|
|
def test_order_asc(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
insert_draw(pb["id"], "2024-06-01", [5, 10, 20, 30, 40], bonus=3)
|
|
|
|
rows = get_draws_with_game(order="ASC")
|
|
assert rows[0]["draw_date"] == "2024-01-01"
|
|
|
|
|
|
def test_limit(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
for day in range(1, 6):
|
|
insert_draw(pb["id"], f"2024-01-{day:02d}", [day, 2, 3, 4, 5], bonus=day)
|
|
|
|
rows = get_draws_with_game(limit=3)
|
|
assert len(rows) == 3
|
|
|
|
|
|
def test_combined_filters(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
mm = get_game_by_name("Mega Millions")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
insert_draw(pb["id"], "2024-03-01", [5, 10, 20, 30, 40], bonus=3)
|
|
insert_draw(mm["id"], "2024-03-01", [7, 11, 22, 29, 38], bonus=4)
|
|
|
|
rows = get_draws_with_game(game_id=pb["id"], date_from="2024-02-01")
|
|
assert len(rows) == 1
|
|
assert rows[0]["game_name"] == "Powerball"
|
|
assert rows[0]["draw_date"] == "2024-03-01"
|
|
|
|
|
|
|
|
# ── number filter ─────────────────────────────────────────────────────────────
|
|
|
|
def test_number_filter_returns_matching_draws(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 13, 36, 61, 69], bonus=7)
|
|
insert_draw(pb["id"], "2024-01-03", [2, 14, 37, 55, 68], bonus=10)
|
|
|
|
rows = get_draws_with_game(number=13)
|
|
assert len(rows) == 1
|
|
assert rows[0]["draw_date"] == "2024-01-01"
|
|
|
|
|
|
def test_number_filter_no_match_returns_empty(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
|
|
rows = get_draws_with_game(number=99)
|
|
assert rows == []
|
|
|
|
|
|
def test_number_filter_does_not_false_match_substring(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
# numbers 1 and 13 — searching for 1 must NOT return the draw containing 13
|
|
insert_draw(pb["id"], "2024-01-01", [13, 23, 33, 43, 53], bonus=7)
|
|
|
|
rows = get_draws_with_game(number=1)
|
|
assert rows == []
|
|
|
|
|
|
def test_number_filter_matches_first_position(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
|
|
rows = get_draws_with_game(number=1)
|
|
assert len(rows) == 1
|
|
|
|
|
|
def test_number_filter_matches_last_position(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 69], bonus=7)
|
|
|
|
rows = get_draws_with_game(number=69)
|
|
assert len(rows) == 1
|
|
|
|
|
|
def test_number_filter_combined_with_game(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
mm = get_game_by_name("Mega Millions")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 13, 36, 61, 69], bonus=7)
|
|
insert_draw(mm["id"], "2024-01-02", [1, 14, 37, 55, 68], bonus=4)
|
|
|
|
rows = get_draws_with_game(game_id=pb["id"], number=1)
|
|
assert len(rows) == 1
|
|
assert rows[0]["game_name"] == "Powerball"
|
|
|
|
|
|
def test_number_filter_none_returns_all(tmp_db):
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 2, 3, 4, 5], bonus=7)
|
|
insert_draw(pb["id"], "2024-01-03", [6, 7, 8, 9, 10], bonus=2)
|
|
|
|
rows = get_draws_with_game(number=None)
|
|
assert len(rows) == 2
|
|
|
|
|
|
# ── UI smoke test (skipped on headless) ───────────────────────────────────────
|
|
|
|
def _has_display():
|
|
try:
|
|
import tkinter as tk
|
|
root = tk.Tk()
|
|
root.withdraw()
|
|
root.destroy()
|
|
return True
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
@pytest.mark.skipif(not _has_display(), reason="no display available")
|
|
def test_history_screen_instantiates(tmp_db):
|
|
import tkinter as tk
|
|
from ui.history import HistoryScreen
|
|
|
|
root = tk.Tk()
|
|
root.withdraw()
|
|
try:
|
|
screen = HistoryScreen(root)
|
|
assert screen.winfo_exists()
|
|
finally:
|
|
root.destroy()
|
|
|
|
|
|
@pytest.mark.skipif(not _has_display(), reason="no display available")
|
|
def test_history_screen_shows_inserted_rows(tmp_db):
|
|
import tkinter as tk
|
|
from ui.history import HistoryScreen
|
|
|
|
pb = get_game_by_name("Powerball")
|
|
insert_draw(pb["id"], "2024-01-01", [1, 13, 36, 61, 69], bonus=7, source="powerball_ny")
|
|
insert_draw(pb["id"], "2024-03-01", [5, 20, 35, 50, 65], bonus=15, source="powerball_ny")
|
|
|
|
root = tk.Tk()
|
|
root.withdraw()
|
|
try:
|
|
screen = HistoryScreen(root)
|
|
children = screen._tree.get_children()
|
|
assert len(children) == 2
|
|
finally:
|
|
root.destroy()
|