[async] Applied #2513 to async code

Author: sfc-gh-fpawlowski

src/snowflake/connector/aio/_cursor.py

@@ -24,7 +24,6 @@
 )
 from snowflake.connector._sql_util import get_file_transfer_type
 from snowflake.connector.aio._bind_upload_agent import BindUploadAgent
-from snowflake.connector.aio._file_transfer_agent import SnowflakeFileTransferAgent
 from snowflake.connector.aio._result_batch import (
     ResultBatch,
     create_batches_from_response,
@@ -1310,6 +1309,9 @@ def _create_file_transfer_agent(
         /,
         **kwargs,
     ) -> SnowflakeFileTransferAgent:
+        from snowflake.connector.aio._file_transfer_agent import (
+            SnowflakeFileTransferAgent,
+        )
 
         return SnowflakeFileTransferAgent(
             self,

test/unit/aio/test_cursor_async_unit.py

@@ -111,7 +111,7 @@ async def mock_cmd_query(*args, **kwargs):
 class TestUploadDownloadMethods(IsolatedAsyncioTestCase):
     """Test the _upload/_download/_upload_stream/_download_stream methods."""
 
-    @patch("snowflake.connector.aio._cursor.SnowflakeFileTransferAgent")
+    @patch("snowflake.connector.aio._file_transfer_agent.SnowflakeFileTransferAgent")
     async def test_download(self, MockFileTransferAgent):
         cursor, fake_conn, mock_file_transfer_agent_instance = self._setup_mocks(
             MockFileTransferAgent
@@ -131,7 +131,7 @@ async def test_download(self, MockFileTransferAgent):
         assert MockFileTransferAgent.call_args.kwargs.get("use_s3_regional_url", False)
         mock_file_transfer_agent_instance.execute.assert_called_once()
 
-    @patch("snowflake.connector.aio._cursor.SnowflakeFileTransferAgent")
+    @patch("snowflake.connector.aio._file_transfer_agent.SnowflakeFileTransferAgent")
     async def test_upload(self, MockFileTransferAgent):
         cursor, fake_conn, mock_file_transfer_agent_instance = self._setup_mocks(
             MockFileTransferAgent
@@ -151,7 +151,7 @@ async def test_upload(self, MockFileTransferAgent):
         assert MockFileTransferAgent.call_args.kwargs.get("use_s3_regional_url", False)
         mock_file_transfer_agent_instance.execute.assert_called_once()
 
-    @patch("snowflake.connector.aio._cursor.SnowflakeFileTransferAgent")
+    @patch("snowflake.connector.aio._file_transfer_agent.SnowflakeFileTransferAgent")
     async def test_download_stream(self, MockFileTransferAgent):
         cursor, fake_conn, mock_file_transfer_agent_instance = self._setup_mocks(
             MockFileTransferAgent
@@ -170,7 +170,7 @@ async def test_download_stream(self, MockFileTransferAgent):
         MockFileTransferAgent.assert_not_called()
         mock_file_transfer_agent_instance.execute.assert_not_called()
 
-    @patch("snowflake.connector.aio._cursor.SnowflakeFileTransferAgent")
+    @patch("snowflake.connector.aio._file_transfer_agent.SnowflakeFileTransferAgent")
     async def test_upload_stream(self, MockFileTransferAgent):
         cursor, fake_conn, mock_file_transfer_agent_instance = self._setup_mocks(
             MockFileTransferAgent