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