Fixed KeyError when encountering mocked annotations

Fixes #116.

Author: agronholm

CHANGELOG.rst

@@ -1,3 +1,9 @@
+1.10.1
+======
+
+* Fixed ``KeyError`` when encountering mocked annotations (``autodoc_mock_imports``)
+
+
 1.10.0
 ======
 

sphinx_autodoc_typehints.py

@@ -58,7 +58,7 @@ def get_annotation_class_name(annotation) -> str:
 def get_annotation_args(annotation, module: str, class_name: str) -> Tuple:
     try:
         original = getattr(sys.modules[module], class_name)
-    except AttributeError:
+    except (KeyError, AttributeError):
         pass
     else:
         if annotation is original:

tests/roots/test-dummy/dummy_module.py

@@ -1,4 +1,5 @@
 import typing
+from mailbox import Mailbox
 from typing import Callable, Union
 
 try:
@@ -248,3 +249,11 @@ class Decorator:
 
     def __init__(self, func: Callable[[int, str], str]):
         pass
+
+
+def mocked_import(x: Mailbox):
+    """
+    A docstring.
+
+    :param x: function
+    """

tests/roots/test-dummy/index.rst

@@ -34,3 +34,5 @@ Dummy Module
     :special-members: __init__
 
 .. autodecorator:: dummy_module.Decorator
+
+.. autofunction:: dummy_module.mocked_import

tests/test_sphinx_autodoc_typehints.py

@@ -207,6 +207,7 @@ def test_sphinx_output(app, status, warning, always_document_param_types):
         sys.path.insert(0, str(test_path))
 
     app.config.always_document_param_types = always_document_param_types
+    app.config.autodoc_mock_imports = ['mailbox']
     app.build()
 
     assert 'build succeeded' in status.getvalue()  # Build succeeded
@@ -487,6 +488,13 @@ class dummy_module.DataClass
 
            Parameters:
               **func** ("Callable"[["int", "str"], "str"]) -- function
+
+        dummy_module.mocked_import(x)
+
+           A docstring.
+
+           Parameters:
+              **x** ("Mailbox") -- function
         ''')
         expected_contents = expected_contents.format(**format_args).replace('–', '--')
         assert text_contents == expected_contents