Fix linting error and linting config
This commit is contained in:
@@ -46,9 +46,8 @@ def database_ctx(db: Database) -> Generator[Database, None, None]:
|
||||
|
||||
def test_select_one_executes() -> None:
|
||||
db = create_database(None) # in-memory by default
|
||||
with database_ctx(db):
|
||||
with session_ctx(db) as session:
|
||||
val = session.exec(text("SELECT 1")).scalar_one()
|
||||
with database_ctx(db), session_ctx(db) as session:
|
||||
val = session.exec(text("SELECT 1")).scalar_one()
|
||||
assert int(val) == 1
|
||||
|
||||
|
||||
@@ -56,9 +55,7 @@ def test_in_memory_persists_across_sessions_when_using_staticpool() -> None:
|
||||
db = create_database(None) # in-memory with StaticPool
|
||||
with database_ctx(db):
|
||||
with session_ctx(db) as s1:
|
||||
s1.exec(
|
||||
text("CREATE TABLE IF NOT EXISTS t (id INTEGER PRIMARY KEY, val TEXT);")
|
||||
)
|
||||
s1.exec(text("CREATE TABLE IF NOT EXISTS t (id INTEGER PRIMARY KEY, val TEXT);"))
|
||||
s1.exec(text("INSERT INTO t (val) VALUES (:v)").bindparams(v="hello"))
|
||||
with session_ctx(db) as s2:
|
||||
got = s2.exec(text("SELECT val FROM t")).scalar_one()
|
||||
@@ -67,10 +64,9 @@ def test_in_memory_persists_across_sessions_when_using_staticpool() -> None:
|
||||
|
||||
def test_sqlite_pragmas_applied() -> None:
|
||||
db = create_database(None)
|
||||
with database_ctx(db):
|
||||
with database_ctx(db), session_ctx(db) as session:
|
||||
# PRAGMA returns integer 1 when foreign_keys ON
|
||||
with session_ctx(db) as session:
|
||||
fk = session.exec(text("PRAGMA foreign_keys")).scalar_one()
|
||||
fk = session.exec(text("PRAGMA foreign_keys")).scalar_one()
|
||||
assert int(fk) == 1
|
||||
|
||||
|
||||
@@ -82,16 +78,8 @@ def test_rollback_on_exception() -> None:
|
||||
# Create table then insert and raise inside the same session to force rollback
|
||||
with pytest.raises(RuntimeError): # noqa: PT012, SIM117
|
||||
with session_ctx(db) as s:
|
||||
s.exec(
|
||||
text(
|
||||
"CREATE TABLE IF NOT EXISTS t_rb (id INTEGER PRIMARY KEY, val TEXT);"
|
||||
)
|
||||
)
|
||||
s.exec(
|
||||
text("INSERT INTO t_rb (val) VALUES (:v)").bindparams(
|
||||
v="will_rollback"
|
||||
)
|
||||
)
|
||||
s.exec(text("CREATE TABLE IF NOT EXISTS t_rb (id INTEGER PRIMARY KEY, val TEXT);"))
|
||||
s.exec(text("INSERT INTO t_rb (val) VALUES (:v)").bindparams(v="will_rollback"))
|
||||
# simulate handler error -> should trigger rollback in get_session
|
||||
raise RuntimeError("simulated failure")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user