Remove unneeded assert, rollback session, and include fixture teardown
This commit is contained in:
1 parent
909d1ad55b
commit
b0b7c0c3fb
1 file changed
+6
-3
@@ -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()
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user