Fix 'test_write_error_but_response_sent' test case

tests/_async/test_connection.py

@@ -12,7 +12,7 @@
     WriteError,
 )
 from httpcore.backends.base import AsyncNetworkStream
-from httpcore.backends.mock import AsyncMockBackend
+from httpcore.backends.mock import AsyncMockBackend, AsyncMockStream
 
 
 @pytest.mark.anyio
@@ -89,7 +89,7 @@ async def test_write_error_but_response_sent():
     will raise a `ConnectionNotAvailable` exception.
     """
 
-    class ErrorOnRequestTooLarge(AsyncMockBackend):
+    class ErrorOnRequestTooLargeStream(AsyncMockStream):
         def __init__(self, buffer: List[bytes], http2: bool = False) -> None:
             super().__init__(buffer, http2)
             self.count = 0
@@ -100,6 +100,16 @@ async def write(self, buffer: bytes, timeout: Optional[float] = None) -> None:
             if self.count > 1_000_000:
                 raise WriteError()
 
+    class ErrorOnRequestTooLarge(AsyncMockBackend):
+        async def connect_tcp(
+            self,
+            host: str,
+            port: int,
+            timeout: Optional[float] = None,
+            local_address: Optional[str] = None,
+        ) -> AsyncMockStream:
+            return ErrorOnRequestTooLargeStream(list(self._buffer), http2=self._http2)
+
     origin = Origin(b"https", b"example.com", 443)
     network_backend = ErrorOnRequestTooLarge(
         [

tests/_sync/test_connection.py

@@ -12,7 +12,7 @@
     WriteError,
 )
 from httpcore.backends.base import NetworkStream
-from httpcore.backends.mock import MockBackend
+from httpcore.backends.mock import MockBackend, MockStream
 
 
 
@@ -89,7 +89,7 @@ def test_write_error_but_response_sent():
     will raise a `ConnectionNotAvailable` exception.
     """
 
-    class ErrorOnRequestTooLarge(MockBackend):
+    class ErrorOnRequestTooLargeStream(MockStream):
         def __init__(self, buffer: List[bytes], http2: bool = False) -> None:
             super().__init__(buffer, http2)
             self.count = 0
@@ -100,6 +100,16 @@ def write(self, buffer: bytes, timeout: Optional[float] = None) -> None:
             if self.count > 1_000_000:
                 raise WriteError()
 
+    class ErrorOnRequestTooLarge(MockBackend):
+        def connect_tcp(
+            self,
+            host: str,
+            port: int,
+            timeout: Optional[float] = None,
+            local_address: Optional[str] = None,
+        ) -> MockStream:
+            return ErrorOnRequestTooLargeStream(list(self._buffer), http2=self._http2)
+
     origin = Origin(b"https", b"example.com", 443)
     network_backend = ErrorOnRequestTooLarge(
         [