Skip to content

Commit 4d576d0

Browse files
committed
fix existing tests
1 parent b7ef95c commit 4d576d0

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/aio/query/test_query_session_pool.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,15 +10,14 @@ async def test_checkout_provides_created_session(self, pool: QuerySessionPoolAsy
1010
async with pool.checkout() as session:
1111
assert session._state._state == QuerySessionStateEnum.CREATED
1212

13-
assert session._state._state == QuerySessionStateEnum.CLOSED
14-
1513
@pytest.mark.asyncio
1614
async def test_oneshot_query_normal(self, pool: QuerySessionPoolAsync):
1715
res = await pool.execute_with_retries("select 1;")
1816
assert len(res) == 1
1917

2018
@pytest.mark.asyncio
2119
async def test_oneshot_ddl_query(self, pool: QuerySessionPoolAsync):
20+
await pool.execute_with_retries("drop table if exists Queen;")
2221
await pool.execute_with_retries("create table Queen(key UInt64, PRIMARY KEY (key));")
2322
await pool.execute_with_retries("drop table Queen;")
2423

0 commit comments

Comments
 (0)