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
|
import data.database as db
|
||||||
|
|
||||||
|
|
||||||
@fixture
|
@fixture(scope='session')
|
||||||
def db_session() -> db.dal.Session:
|
def db_session() -> db.dal.Session:
|
||||||
db.dal.connect()
|
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:
|
def test_base() -> None:
|
||||||
@@ -49,10 +52,10 @@ def test_powerball_draw_integrity(db_session) -> None:
|
|||||||
try:
|
try:
|
||||||
session.add(draw)
|
session.add(draw)
|
||||||
session.commit()
|
session.commit()
|
||||||
assert False
|
|
||||||
except IntegrityError:
|
except IntegrityError:
|
||||||
assert True
|
assert True
|
||||||
finally:
|
finally:
|
||||||
|
session.rollback()
|
||||||
session.close()
|
session.close()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user