Fix stub generation false reporting

mypy/stubdoc.py

@@ -179,6 +179,16 @@ def __init__(self, function_name: str) -> None:
 
     def add_token(self, token: tokenize.TokenInfo) -> None:
         """Process next token from the token stream."""
+
+        if (
+            self.state[-1] == STATE_ARGUMENT_TYPE
+            and token.type != tokenize.NAME
+            and len(self.accumulator) == 0
+        ):
+            # the next token after : must be a name
+            self.reset()
+            return
+
         if (
             token.type == tokenize.NAME
             and token.string == self.function_name
@@ -281,8 +291,13 @@ def add_token(self, token: tokenize.TokenInfo) -> None:
             self.accumulator = ""
 
         elif token.type == tokenize.OP and token.string == "->" and self.state[-1] == STATE_INIT:
+            if len(self.accumulator) == 0:
+                self.state.append(STATE_RETURN_VALUE)
+            else:
+                # ) is not directly followed by ->
+                self.reset()
+                return
             self.accumulator = ""
-            self.state.append(STATE_RETURN_VALUE)
 
         # ENDMAKER is necessary for python 3.4 and 3.5.
         elif token.type in (tokenize.NEWLINE, tokenize.ENDMARKER) and self.state[-1] in (
@@ -305,6 +320,17 @@ def add_token(self, token: tokenize.TokenInfo) -> None:
             self.args = []
             self.ret_type = "Any"
             # Leave state as INIT.
+        elif (
+            token.type == tokenize.NAME
+            and self.state[-1] == STATE_ARGUMENT_LIST
+            and len(self.accumulator) != 0
+            and self.accumulator != "*"
+            and self.accumulator != "**"
+        ):
+            # not the token after the argument list started
+            # special case for *args and **kwargs
+            self.reset()
+            return
         else:
             self.accumulator += token.string
 

mypy/test/teststubgen.py

@@ -380,6 +380,27 @@ def test_infer_sig_from_docstring(self) -> None:
             ],
         )
 
+        assert_equal(
+            infer_sig_from_docstring("\nfunc(*args)", "func"),
+            [FunctionSig(name="func", args=[ArgSig(name="*args")], ret_type="Any")],
+        )
+
+        assert_equal(
+            infer_sig_from_docstring("\nfunc(**kwargs)", "func"),
+            [FunctionSig(name="func", args=[ArgSig(name="**kwargs")], ret_type="Any")],
+        )
+
+        assert_equal(
+            infer_sig_from_docstring("\nfunc(*args, **kwargs)", "func"),
+            [
+                FunctionSig(
+                    name="func",
+                    args=[ArgSig(name="*args"), ArgSig(name="**kwargs")],
+                    ret_type="Any",
+                )
+            ],
+        )
+
     def test_infer_sig_from_docstring_duplicate_args(self) -> None:
         assert_equal(
             infer_sig_from_docstring("\nfunc(x, x) -> str\nfunc(x, y) -> int", "func"),
@@ -399,6 +420,77 @@ def test_infer_sig_from_docstring_bad_indentation(self) -> None:
             None,
         )
 
+    def test_infer_sig_from_docstring_invalid_signature(self) -> None:
+
+        assert_equal(infer_sig_from_docstring("\nfunc() --> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(name1 name2) -> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(name1, name2 name3) -> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(name2, name3) -> None, None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(invalid::name) -> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(invalid: [type]) -> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(invalid: (type)) -> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(invalid: -type) -> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(invalid<name) -> None", "func"), [])
+
+        assert_equal(infer_sig_from_docstring("\nfunc(cpp::type name) -> None", "func"), [])
+        assert_equal(infer_sig_from_docstring("\nfunc(name) -> cpp::type", "func"), [])
+        assert_equal(infer_sig_from_docstring("\nvoid func(int name) {", "func"), [])
+        assert_equal(infer_sig_from_docstring("\nvoid func(std::vector<int> name)", "func"), [])
+
+    def test_infer_sig_from_docstring_multiple_overloads(self) -> None:
+        input = """
+func(*args, **kwargs)
+Overloaded function.
+
+1. func(self: class_type) -> None
+
+2. func(self: class_type, a: float, b: int) -> None
+"""
+        assert_equal(
+            infer_sig_from_docstring(input, "func"),
+            [
+                FunctionSig(
+                    name="func", args=[ArgSig(name="self", type="class_type")], ret_type="None"
+                ),
+                FunctionSig(
+                    name="func",
+                    args=[
+                        ArgSig(name="self", type="class_type"),
+                        ArgSig(name="a", type="float"),
+                        ArgSig(name="b", type="int"),
+                    ],
+                    ret_type="None",
+                ),
+                FunctionSig(
+                    name="func",
+                    args=[ArgSig(name="*args"), ArgSig(name="**kwargs")],
+                    ret_type="Any",
+                ),
+            ],
+        )
+
+    def test_infer_sig_from_docstring_deeply_nested_types(self) -> None:
+        assert_equal(
+            infer_sig_from_docstring(
+                "\nfunc(name: dict[str, dict[str, list[tuple[int, float]]]]) -> None", "func"
+            ),
+            [
+                FunctionSig(
+                    name="func",
+                    args=[ArgSig(name="name", type="dict[str,dict[str,list[tuple[int,float]]]]")],
+                    ret_type="None",
+                )
+            ],
+        )
+
     def test_infer_arg_sig_from_anon_docstring(self) -> None:
         assert_equal(
             infer_arg_sig_from_anon_docstring("(*args, **kwargs)"),