From b0b7c0c3fb58f52e3e51132a67dafd61840a5f10 Mon Sep 17 00:00:00 2001
From: Chris Smith <ole088@gmail.com>
Date: Sat, 28 Feb 2026 17:11:27 -0500
Subject: [PATCH] Remove unneeded assert, rollback session, and include fixture
 teardown

---
 tests/test_database.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

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()