Add additional db fixture and tests

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