raise exception if aquire session from stopped pool

Author: rekby

tests/aio/test_table.py

@@ -0,0 +1,12 @@
+import pytest
+import ydb.aio
+
+
+@pytest.mark.asyncio
+class TestSessionPool:
+    async def test_checkout_from_stopped_pool(self, driver):
+        pool = ydb.aio.SessionPool(driver, 1)
+        await pool.stop()
+
+        with pytest.raises(ValueError):
+            await pool.acquire()

tests/table/table_test.py

@@ -1,6 +1,16 @@
+import pytest
 import ydb
 
 
+class TestSessionPool:
+    def test_checkout_from_stopped_pool(self, driver_sync):
+        pool = ydb.SessionPool(driver_sync, 1)
+        pool.stop()
+
+        with pytest.raises(ValueError):
+            pool.acquire()
+
+
 class TestTable:
     def test_create_table_with_not_null_primary_key_by_api(self, driver_sync, database):
         table_path = database + "/test_table"

ydb/_sp_impl.py

@@ -335,6 +335,10 @@ def _on_keep_alive(self, session, f):
             self._destroy(session, "keep-alive-error")
 
     def acquire(self, blocking=True, timeout=None):
+        if self._should_stop.is_set():
+            self._logger.debug("Acquired not inited session")
+            raise ValueError("Take session from closed session pool.")
+
         waiter = self.subscribe()
         has_result = False
         if blocking:

ydb/aio/table.py

@@ -342,7 +342,7 @@ async def acquire(self, timeout: float = None, retry_timeout: float = None, retr
 
         if self._should_stop.is_set():
             self._logger.debug("Acquired not inited session")
-            return self._create()
+            raise ValueError("Take session from closed session pool.")
 
         if retry_timeout is None:
             retry_timeout = timeout