diff --git a/stubs/mock/mock/mock.pyi b/stubs/mock/mock/mock.pyi index 03e2ac23d..19dbf0ae1 100644 --- a/stubs/mock/mock/mock.pyi +++ b/stubs/mock/mock/mock.pyi @@ -1,5 +1,5 @@ from _typeshed import Incomplete -from collections.abc import Callable, Coroutine, Mapping, Sequence +from collections.abc import Callable, Coroutine, Iterable, Mapping, Sequence from contextlib import AbstractContextManager from types import TracebackType from typing import Any, Generic, TypeVar, overload @@ -316,7 +316,7 @@ class AsyncMockMixin(Base): def assert_awaited_with(_mock_self, *args: Any, **kwargs: Any) -> None: ... def assert_awaited_once_with(_mock_self, *args: Any, **kwargs: Any) -> None: ... def assert_any_await(_mock_self, *args: Any, **kwargs: Any) -> None: ... - def assert_has_awaits(_mock_self, calls: _CallList, any_order: bool = False) -> None: ... + def assert_has_awaits(_mock_self, calls: Iterable[_Call], any_order: bool = False) -> None: ... def assert_not_awaited(_mock_self) -> None: ... def reset_mock(self, *args: Any, **kwargs: Any) -> None: ... await_count: int