From: Florimond Manca <florimond.manca@gmail.com>
Date: Mon, 16 Sep 2019 06:34:26 +0000 (+0200)
Subject: Fix remaining references to BaseStream in tests (#347)
X-Git-Tag: 0.7.3~6
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=6c6148ec5c3fc4c32238664d4d8feb3f90c7d88a;p=thirdparty%2Fhttpx.git

Fix remaining references to BaseStream in tests (#347)
---

diff --git a/tests/dispatch/utils.py b/tests/dispatch/utils.py
index aee4c18a..b8bbb780 100644
--- a/tests/dispatch/utils.py
+++ b/tests/dispatch/utils.py
@@ -169,13 +169,13 @@ class MockRawSocketBackend:
         self.received_data = []
         self.stream = MockRawSocketStream(self)
 
-    async def connect(
+    async def open_tcp_stream(
         self,
         hostname: str,
         port: int,
         ssl_context: typing.Optional[ssl.SSLContext],
         timeout: TimeoutConfig,
-    ) -> BaseStream:
+    ) -> BaseTCPStream:
         self.received_data.append(
             b"--- CONNECT(%s, %d) ---" % (hostname.encode(), port)
         )
@@ -183,11 +183,11 @@ class MockRawSocketBackend:
 
     async def start_tls(
         self,
-        stream: BaseStream,
+        stream: BaseTCPStream,
         hostname: str,
         ssl_context: ssl.SSLContext,
         timeout: TimeoutConfig,
-    ) -> BaseStream:
+    ) -> BaseTCPStream:
         self.received_data.append(b"--- START_TLS(%s) ---" % hostname.encode())
         return self.stream
 
@@ -196,7 +196,7 @@ class MockRawSocketBackend:
         return getattr(self.backend, name)
 
 
-class MockRawSocketStream(BaseStream):
+class MockRawSocketStream(BaseTCPStream):
     def __init__(self, backend: MockRawSocketBackend):
         self.backend = backend