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