Merge branch 'master' into fix/nox-session-chaining

lundberg · Nov 12, 2021 · 93e649c · 93e649c
2 parents c345425 + 8cc098f
commit 93e649c
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/respx/router.py b/respx/router.py
@@ -339,6 +339,30 @@ def __init__(
         )
         self._using = using
 
+    @overload
+    def __call__(
+        self,
+        func: None = None,
+        *,
+        assert_all_called: Optional[bool] = None,
+        assert_all_mocked: Optional[bool] = None,
+        base_url: Optional[str] = None,
+        using: Optional[Union[str, Default]] = DEFAULT,
+    ) -> "MockRouter":
+        ...  # pragma: nocover
+
+    @overload
+    def __call__(
+        self,
+        func: Callable = ...,
+        *,
+        assert_all_called: Optional[bool] = None,
+        assert_all_mocked: Optional[bool] = None,
+        base_url: Optional[str] = None,
+        using: Optional[Union[str, Default]] = DEFAULT,
+    ) -> Callable:
+        ...  # pragma: nocover
+
     def __call__(
         self,
         func: Optional[Callable] = None,