Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions Lib/test/test_unittest/testmock/testmock.py
Original file line number Diff line number Diff line change
Expand Up @@ -1743,6 +1743,17 @@ def static_method(): pass
mock_method.assert_called_once_with()
self.assertRaises(TypeError, mock_method, 'extra_arg')

#Issue145754
def test_create_autospec_type_hints_typechecking(self):
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from typing import Tuple

def foo(x: Tuple[int, ...]) -> None:
pass

mock.create_autospec(foo)

#Issue21238
def test_mock_unsafe(self):
m = Mock()
Expand Down
3 changes: 2 additions & 1 deletion Lib/unittest/mock.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@
import pkgutil
from inspect import iscoroutinefunction
import threading
from annotationlib import Format
from dataclasses import fields, is_dataclass
from types import CodeType, ModuleType, MethodType
from unittest.util import safe_repr
Expand Down Expand Up @@ -119,7 +120,7 @@ def _get_signature_object(func, as_instance, eat_self):
else:
sig_func = func
try:
return func, inspect.signature(sig_func)
return func, inspect.signature(sig_func, annotation_format=Format.FORWARDREF)
except ValueError:
# Certain callable types are not supported by inspect.signature()
return None
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
Request signature during mock autospec with ``FORWARDREF`` annotation format.
This prevents runtime errors when a type referred in a type hint is
imported within a ``TYPE_CHECKING`` block.
Loading