[Async] Apply 2241 to async code

Author: sfc-gh-fpawlowski

src/snowflake/connector/aio/_file_transfer_agent.py

@@ -301,6 +301,7 @@ async def _create_file_transfer_client(
                 self._cursor._connection,
                 self._command,
                 unsafe_file_write=self._unsafe_file_write,
+                use_virtual_endpoints=self._gcs_use_virtual_endpoints,
             )
             if client.security_token:
                 logger.debug(f"len(GCS_ACCESS_TOKEN): {len(client.security_token)}")

src/snowflake/connector/aio/_gcs_storage_client.py

@@ -38,6 +38,7 @@ def __init__(
         cnx: SnowflakeConnection,
         command: str,
         unsafe_file_write: bool = False,
+        use_virtual_endpoints: bool = False,
     ) -> None:
         """Creates a client object with given stage credentials.
 
@@ -72,6 +73,7 @@ def __init__(
         self.endpoint: str | None = (
             None if "endPoint" not in stage_info else stage_info["endPoint"]
         )
+        self.use_virtual_endpoints: bool = use_virtual_endpoints
 
     async def _has_expired_token(self, response: aiohttp.ClientResponse) -> bool:
         return self.security_token and response.status == 401
@@ -147,6 +149,8 @@ def generate_url_and_rest_args() -> (
                         if "region" not in self.stage_info
                         else self.stage_info["region"]
                     ),
+                    self.endpoint,
+                    self.use_virtual_endpoints,
                 )
                 access_token = self.security_token
             else:
@@ -185,6 +189,7 @@ def generate_url_and_rest_args() -> (
                         else self.stage_info["region"]
                     ),
                     self.endpoint,
+                    self.use_virtual_endpoints,
                 )
                 access_token = self.security_token
                 gcs_headers["Authorization"] = f"Bearer {access_token}"
@@ -309,6 +314,8 @@ def generate_url_and_authenticated_headers():
                         if "region" not in self.stage_info
                         else self.stage_info["region"]
                     ),
+                    self.endpoint,
+                    self.use_virtual_endpoints,
                 )
                 gcs_headers = {"Authorization": f"Bearer {self.security_token}"}
                 rest_args = {"headers": gcs_headers}