feat: add async context manager support to ClientTransport - Apply suggestion from reviewer

Author: cchinchilla-dev

tests/client/test_base_client.py

@@ -1,4 +1,4 @@
-from unittest.mock import AsyncMock, MagicMock
+from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
 
@@ -63,39 +63,28 @@ def base_client(
 
 @pytest.mark.asyncio
 async def test_transport_async_context_manager() -> None:
-    class TestTransport(ClientTransport):
-        def __init__(self) -> None:
-            self.closed = False
-
-        async def close(self) -> None:
-            self.closed = True
-
-    TestTransport.__abstractmethods__ = set()  # type: ignore[attr-defined]
-
-    transport = TestTransport()
-    async with transport as t:
-        assert t is transport
-
-    assert transport.closed
+    with (
+        patch.object(ClientTransport, '__abstractmethods__', set()),
+        patch.object(ClientTransport, 'close', new_callable=AsyncMock),
+    ):
+        transport = ClientTransport()
+        async with transport as t:
+            assert t is transport
+            transport.close.assert_not_awaited()
+        transport.close.assert_awaited_once()
 
 
 @pytest.mark.asyncio
 async def test_transport_async_context_manager_on_exception() -> None:
-    class TestTransport(ClientTransport):
-        def __init__(self) -> None:
-            self.closed = False
-
-        async def close(self) -> None:
-            self.closed = True
-
-    TestTransport.__abstractmethods__ = set()  # type: ignore[attr-defined]
-
-    transport = TestTransport()
-    with pytest.raises(RuntimeError, match='boom'):
-        async with transport:
-            raise RuntimeError('boom')
-
-    assert transport.closed
+    with (
+        patch.object(ClientTransport, '__abstractmethods__', set()),
+        patch.object(ClientTransport, 'close', new_callable=AsyncMock),
+    ):
+        transport = ClientTransport()
+        with pytest.raises(RuntimeError, match='boom'):
+            async with transport:
+                raise RuntimeError('boom')
+        transport.close.assert_awaited_once()
 
 
 @pytest.mark.asyncio