Fix typing

lundberg · Oct 10, 2020 · 3848e36 · 3848e36
1 parent b9ff034
commit 3848e36
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/respx/models.py b/respx/models.py
@@ -32,7 +32,7 @@
 Headers = List[Tuple[bytes, bytes]]
 Request = Tuple[
     bytes,  # http method
-    URL,
+    Union[str, URL],
     Headers,
     Union[Iterable[bytes], AsyncIterable[bytes]],  # body
 ]

diff --git a/respx/transports.py b/respx/transports.py
@@ -21,7 +21,6 @@
     RequestPattern,
     ResponseTemplate,
     SyncResponse,
-    URLPatternTypes,
     decode_request,
     decode_response,
 )
@@ -329,7 +328,7 @@ def assert_all_called(self) -> None:
     def match(
         self,
         method: bytes,
-        url: URLPatternTypes,
+        url: Union[str, URL],
         headers: Headers = None,
         stream: Union[SyncByteStream, AsyncByteStream] = None,
     ) -> Tuple[Optional[RequestPattern], Request, Optional[ResponseTemplate]]:
@@ -378,7 +377,7 @@ def match(
     def request(
         self,
         method: bytes,
-        url: URLPatternTypes,
+        url: Union[str, URL],
         headers: Headers = None,
         stream: SyncByteStream = None,
         ext: dict = None,