Skip to content

Commit 4b217b4

Browse files
committed
Update MySQL tests
1 parent c43f8ee commit 4b217b4

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

mysql_tests/test_engine.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,8 @@ async def test_issue_79():
3838
async with e.acquire():
3939
pass # pragma: no cover
4040
# noinspection PyProtectedMember
41-
assert len(e._ctx.get([])) == 0
41+
ctx = e._ctx.get()
42+
assert ctx and len(ctx.stack) == 0
4243

4344

4445
async def test_reuse(engine):
@@ -198,15 +199,15 @@ async def test_lazy(mocker):
198199
init_size = qsize(engine)
199200
async with engine.acquire(lazy=True):
200201
assert qsize(engine) == init_size
201-
assert len(engine._ctx.get()) == 1
202+
assert len(engine._ctx.get().stack) == 1
202203
assert engine._ctx.get() is None
203204
assert qsize(engine) == init_size
204205
async with engine.acquire(lazy=True):
205206
assert qsize(engine) == init_size
206-
assert len(engine._ctx.get()) == 1
207+
assert len(engine._ctx.get().stack) == 1
207208
assert await engine.scalar("select 1")
208209
assert qsize(engine) == init_size - 1
209-
assert len(engine._ctx.get()) == 1
210+
assert len(engine._ctx.get().stack) == 1
210211
assert engine._ctx.get() is None
211212
assert qsize(engine) == init_size
212213

0 commit comments

Comments
 (0)