fix(main): Add support for Python 3.9

Author: Uziel Silva

google/cloud/sql/connector/connector.py

@@ -491,7 +491,7 @@ async def close_async(self) -> None:
             proxy_task = asyncio.gather(self._proxy)
             try:
                 await asyncio.wait_for(proxy_task, timeout=0.1)
-            except TimeoutError:
+            except (asyncio.CancelledError, asyncio.TimeoutError, TimeoutError):
                 pass # This task runs forever so it is expected to throw this exception
 
 

tests/unit/test_connector.py

@@ -317,6 +317,12 @@ async def test_Connector_connect_local_proxy(
                 mock_connect.assert_called_once()
                 mock_proxy.assert_called_once()
                 assert connection is True
+    
+                proxy_task = asyncio.gather(task)
+                try:
+                    await asyncio.wait_for(proxy_task, timeout=0.1)
+                except (asyncio.CancelledError, asyncio.TimeoutError, TimeoutError):
+                    pass # This task runs forever so it is expected to throw this exception
 
 
 @pytest.mark.asyncio

tests/unit/test_proxy.py

@@ -44,7 +44,7 @@ async def test_proxy_creates_folder(context: ssl.SSLContext, kwargs: Any) -> Non
     proxy_task = asyncio.gather(task)
     try:
         await asyncio.wait_for(proxy_task, timeout=0.1)
-    except TimeoutError:
+    except (asyncio.CancelledError, asyncio.TimeoutError, TimeoutError):
         pass # This task runs forever so it is expected to throw this exception
 
 @pytest.mark.usefixtures("proxy_server")