fixup! Fix get_results_from_sfqid with DictCursor + multi statements (#2531)

Author: sfc-gh-turbaszek

src/snowflake/connector/aio/_cursor.py

@@ -76,7 +76,7 @@ def __init__(
         super().__init__(connection, use_dict_result)
         # the following fixes type hint
         self._connection = typing.cast("SnowflakeConnection", self._connection)
-        self._inner_cursor = typing.cast(SnowflakeCursor, self._inner_cursor)
+        self._inner_cursor: SnowflakeCursor | None = None
         self._lock_canceling = asyncio.Lock()
         self._timebomb: asyncio.Task | None = None
         self._prefetch_hook: typing.Callable[[], typing.Awaitable] | None = None
@@ -958,7 +958,7 @@ async def fetchall(self) -> list[tuple] | list[dict]:
             await self._prefetch_hook()
         if self._result is None and self._result_set is not None:
             self._result: ResultSetIterator = await self._result_set._create_iter(
-                is_fetch_all=True
+                is_fetch_all=True,
             )
             self._result_state = ResultState.VALID
 

test/integ/aio_it/test_multi_statement_async.py

@@ -154,15 +154,14 @@ async def test_async_exec_multi(conn_cnx, cursor_class, skip_to_last_set: bool):
             assert con.is_still_running(await con.get_query_status(q_id))
         await _wait_while_query_running_async(con, q_id, sleep_time=1)
     async with conn_cnx() as con:
-        async with con.cursor() as cur:
+        async with con.cursor(cursor_class) as cur:
             await _wait_until_query_success_async(
                 con, q_id, num_checks=3, sleep_per_check=1
             )
             assert (
                 await con.get_query_status_throw_if_error(q_id) == QueryStatus.SUCCESS
             )
 
-            await cur.get_results_from_sfqid(q_id)
             if cursor_class == SnowflakeCursor:
                 expected = [
                     [(1,)],
@@ -177,6 +176,9 @@ async def test_async_exec_multi(conn_cnx, cursor_class, skip_to_last_set: bool):
                     lambda x: len(x) == 1 and len(x[0]) == 1 and x[0]["COUNT(*)"] > 0,
                     [{"'B'": "b"}],
                 ]
+
+            await cur.get_results_from_sfqid(q_id)
+            assert isinstance(cur, cursor_class)
             await _check_multi_statement_results(
                 cur,
                 checks=expected,