diff --git a/tests/test_database.py b/tests/test_database.py index 4e3c81a..c3aed6c 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -7,10 +7,13 @@ import sqlalchemy.orm import data.database as db -@fixture +@fixture(scope='session') def db_session() -> db.dal.Session: db.dal.connect() - return db.dal.Session() + yield db.dal.Session() + if hasattr(db.dal, 'engine') and db.dal.engine: + db.dal.engine.dispose() + db.dal.engine.pool.recreate() def test_base() -> None: @@ -49,10 +52,10 @@ def test_powerball_draw_integrity(db_session) -> None: try: session.add(draw) session.commit() - assert False except IntegrityError: assert True finally: + session.rollback() session.close()