Merge pull request #74 from ydb-platform/fix_async_dialect

Fix async cursor fetch methods

Author: vgvoleg

.github/workflows/tests.yml

@@ -19,8 +19,7 @@ jobs:
       max-parallel: 4
       matrix:
         python-version: [3.9]
-        environment: [test]
-        folder: [ydb_sqlalchemy, test]
+        environment: [test-unit, test-dialect]
 
     steps:
     - uses: actions/checkout@v1
@@ -34,4 +33,4 @@ jobs:
         python -m pip install --upgrade pip
         pip install tox==4.2.6
     - name: Run unit tests
-      run: tox -e ${{ matrix.environment }} -- ${{ matrix.folder }}
+      run: tox -e ${{ matrix.environment }}

requirements.txt

@@ -1,3 +1,3 @@
 sqlalchemy >= 1.4.0, < 3.0.0
 ydb >= 3.18.8
-ydb-dbapi >= 0.1.7
+ydb-dbapi >= 0.1.8

tox.ini

@@ -1,5 +1,5 @@
 [tox]
-envlist = test,test-all,test-dbapi,test-unit,black,black-format,style,coverage
+envlist = test,test-all,test-dialect,test-unit,black,black-format,style,coverage
 minversion = 4.2.6
 skipsdist = True
 ignore_basepython_conflict = true
@@ -29,6 +29,13 @@ commands =
     pytest -v ydb_sqlalchemy
     docker-compose down -v
 
+[testenv:test-dialect]
+commands =
+    docker-compose up -d
+    python {toxinidir}/wait_container_ready.py
+    pytest -v test --dbdriver ydb --dbdriver ydb_async
+    docker-compose down -v
+
 [testenv:test-unit]
 commands =
     pytest -v {toxinidir}/ydb_sqlalchemy

ydb_sqlalchemy/sqlalchemy/dbapi_adapter.py

@@ -87,13 +87,13 @@ def rowcount(self):
         return self._cursor.rowcount
 
     def fetchone(self):
-        return await_only(self._cursor.fetchone())
+        return self._cursor.fetchone()
 
     def fetchmany(self, size=None):
-        return await_only(self._cursor.fetchmany(size=size))
+        return self._cursor.fetchmany(size=size)
 
     def fetchall(self):
-        return await_only(self._cursor.fetchall())
+        return self._cursor.fetchall()
 
     def execute_scheme(self, sql, parameters=None):
         return await_only(self._cursor.execute_scheme(sql, parameters))