Mock and MagickMock subclassing NonCallableMock (#3871)

This commit is contained in:
Rodrigo Castro
2020-03-24 11:32:57 -03:00
committed by GitHub
parent efe23f8cc1
commit 4b14e245d4
2 changed files with 14 additions and 6 deletions

View File

@@ -104,7 +104,7 @@ class CallableMixin(Base):
def __init__(self, spec: Optional[Any] = ..., side_effect: Optional[Any] = ..., return_value: Any = ..., wraps: Optional[Any] = ..., name: Optional[Any] = ..., spec_set: Optional[Any] = ..., parent: Optional[Any] = ..., _spec_state: Optional[Any] = ..., _new_name: Any = ..., _new_parent: Optional[Any] = ..., **kwargs: Any) -> None: ...
def __call__(_mock_self, *args: Any, **kwargs: Any) -> Any: ...
Mock = Any
class Mock(NonCallableMock, CallableMixin): ...
class _patch:
attribute_name: Any
@@ -158,8 +158,12 @@ patch: _patcher
class MagicMixin:
def __init__(self, *args: Any, **kw: Any) -> None: ...
NonCallableMagicMock = Any
MagicMock = Any
class NonCallableMagicMock(MagicMixin, NonCallableMock):
def mock_add_spec(self, spec: Any, spec_set: bool = ...) -> None: ...
class MagicMock(MagicMixin, Mock):
def mock_add_spec(self, spec: Any, spec_set: bool = ...) -> None: ...
if sys.version_info >= (3, 8):
AsyncMock = Any

View File

@@ -104,7 +104,7 @@ class CallableMixin(Base):
def __init__(self, spec: Optional[Any] = ..., side_effect: Optional[Any] = ..., return_value: Any = ..., wraps: Optional[Any] = ..., name: Optional[Any] = ..., spec_set: Optional[Any] = ..., parent: Optional[Any] = ..., _spec_state: Optional[Any] = ..., _new_name: Any = ..., _new_parent: Optional[Any] = ..., **kwargs: Any) -> None: ...
def __call__(_mock_self, *args: Any, **kwargs: Any) -> Any: ...
Mock = Any
class Mock(NonCallableMock, CallableMixin): ...
class _patch:
attribute_name: Any
@@ -158,8 +158,12 @@ patch: _patcher
class MagicMixin:
def __init__(self, *args: Any, **kw: Any) -> None: ...
NonCallableMagicMock = Any
MagicMock = Any
class NonCallableMagicMock(MagicMixin, NonCallableMock):
def mock_add_spec(self, spec: Any, spec_set: bool = ...) -> None: ...
class MagicMock(MagicMixin, Mock):
def mock_add_spec(self, spec: Any, spec_set: bool = ...) -> None: ...
if sys.version_info >= (3, 8):
AsyncMock = Any