Fix tests

Author: rekby

tests/aio/test_session_pool.py

@@ -62,14 +62,14 @@ async def test_close_basic_logic_case_1(driver):
     waiter = asyncio.ensure_future(pool.acquire())
 
     await pool.stop()
-    waiter_sess = waiter.result()
-    assert not waiter_sess.initialized()
-    after_stop = await pool.acquire()
-    assert not after_stop.initialized()
+
+    with pytest.raises(ValueError):
+        waiter.result()
+
+    with pytest.raises(ValueError):
+        await pool.acquire()
 
     await pool.release(s)
-    await pool.release(after_stop)
-    await pool.release(waiter_sess)
     assert pool._active_count == 0
 
 
@@ -106,9 +106,9 @@ async def test_close_basic_logic_case_2(driver):
 
     assert pool._active_count == 0
 
-    sess = await pool.acquire()
+    with pytest.raises(ValueError):
+        await pool.acquire()
 
-    assert not sess.initialized()
     await pool.stop()
 
 

tests/session_pool.py

@@ -0,0 +1,40 @@
+import pytest
+
+import ydb
+
+
+def test_close_basic_logic_case_1(driver_sync):
+    pool = ydb.SessionPool(driver_sync, 1)
+    s = pool.acquire()
+
+    pool.stop()
+
+    with pytest.raises(ValueError):
+        pool.acquire()
+
+    pool.release(s)
+    assert pool._pool_impl._active_count == 0
+
+
+def test_close_basic_logic_case_2(driver_sync):
+    pool = ydb.SessionPool(driver_sync, 10)
+    acquired = []
+
+    for _ in range(10):
+        acquired.append(pool.acquire())
+
+    for _ in range(3):
+        pool.release(acquired.pop(-1))
+
+    pool.stop()
+    assert pool._pool_impl._active_count == 7
+
+    while acquired:
+        pool.release(acquired.pop(-1))
+
+    assert pool._pool_impl._active_count == 0
+
+    with pytest.raises(ValueError):
+        pool.acquire()
+
+    pool.stop()