32 lines
766 B
Python
32 lines
766 B
Python
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
|