assert_has_awaits takes Iterable[_Call] instead of _CallList (#5052)

JonathanMeans · web-flow · commit f4c35f1059c2 · 2021-02-22T22:55:51.000+01:00
diff --git a/stdlib/unittest/mock.pyi b/stdlib/unittest/mock.pyi
@@ -1,5 +1,5 @@
 import sys
-from typing import Any, Callable, Generic, List, Mapping, Optional, Sequence, Tuple, Type, TypeVar, Union, overload
+from typing import Any, Callable, Generic, Iterable, List, Mapping, Optional, Sequence, Tuple, Type, TypeVar, Union, overload
 
 _F = TypeVar("_F", bound=Callable[..., Any])
 _T = TypeVar("_T")
@@ -342,7 +342,7 @@ if sys.version_info >= (3, 8):
         def assert_awaited_with(self, *args: Any, **kwargs: Any) -> None: ...
         def assert_awaited_once_with(self, *args: Any, **kwargs: Any) -> None: ...
         def assert_any_await(self, *args: Any, **kwargs: Any) -> None: ...
-        def assert_has_awaits(self, calls: _CallList, any_order: bool = ...) -> None: ...
+        def assert_has_awaits(self, calls: Iterable[_Call], any_order: bool = ...) -> None: ...
         def assert_not_awaited(self) -> None: ...
         def reset_mock(self, *args: Any, **kwargs: Any) -> None: ...
         await_count: int
