Remove unneeded assert, rollback session, and include fixture teardown

This commit is contained in:
chris committed 2026-02-28 17:11:27 -05:00
1 parent 909d1ad55b
commit b0b7c0c3fb
1 file changed
+6 -3
+6 -3
View File
@@ -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()