fix tests

Author: sfc-gh-pczajka

test/unit/aio/test_cursor_async_unit.py

@@ -60,7 +60,7 @@ def test_get_filetransfer_type(sql, _type):
     assert SnowflakeCursor.get_file_transfer_type(sql) == _type
 
 
-def test_cursor_attribute():
+async def test_cursor_attribute():
     fake_conn = FakeConnection()
     cursor = SnowflakeCursor(fake_conn)
     assert cursor.lastrowid is None

test/unit/aio/test_proxies_async.py

@@ -171,7 +171,7 @@ async def _execute_large_query(connect_kwargs, row_count: int):
         await cur.execute(
             f"select seq4() as n from table(generator(rowcount => {row_count}));"
         )
-        assert len(cur._result_set.batches) > 1
+        assert len(cur._execution_state.result_set.batches) > 1
         _ = [r async for r in cur]
     finally:
         await conn.close()