From 06ee8f37d792a145b87351065ce1fc456a2e644e Mon Sep 17 00:00:00 2001
From: Chris Smith <ole088@gmail.com>
Date: Fri, 20 Feb 2026 13:38:28 -0500
Subject: [PATCH] Add additional db fixture and tests

---
 tests/test_database.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/tests/test_database.py b/tests/test_database.py
index bf1c664..4a7299c 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -1,7 +1,31 @@
+from datetime import datetime
+from pytest import fixture
+
 import sqlalchemy.orm
 
 import data.database as db
 
 
+@fixture
+def db_session() -> db.dal.Session:
+    db.dal.connect()
+    return db.dal.Session()
+
+
 def test_base() -> None:
     assert isinstance(db.Base(), sqlalchemy.orm.DeclarativeBase)
+
+
+def test_powerball_draw_by_date_blank(db_session) -> None:
+    session = db_session
+
+    blank_results = session.query(db.PowerballDraw).filter_by(draw_date='').all()
+    assert blank_results == []
+
+
+def test_powerball_draw_by_date(db_session) -> None:
+    session = db_session
+    date = datetime(2010, 2, 3).strftime('%Y-%m-%d')
+
+    record = session.query(db.PowerballDraw).filter_by(draw_date=date).first()
+    assert record.draw_date.strftime('%Y-%m-%d') == date