gh-106368: Argument clinic: Fix minor bug in state_modulename_name (#107387)

AlexWaygood · web-flow · commit ecc05e23a1f4 · 2023-07-28T19:10:45.000+01:00
diff --git a/Lib/test/test_clinic.py b/Lib/test/test_clinic.py
@@ -84,6 +84,7 @@ def __init__(self):
             ('parser_definition', d('block')),
             ('impl_definition', d('block')),
         ))
+        self.functions = []
 
     def get_destination(self, name):
         d = self.destinations.get(name)
@@ -104,6 +105,9 @@ def directive(self, name, args):
 
     _module_and_class = clinic.Clinic._module_and_class
 
+    def __repr__(self):
+        return "<FakeClinic object>"
+
 
 class ClinicWholeFileTest(_ParserBase):
     def setUp(self):
@@ -672,6 +676,19 @@ def test_c_name(self):
         """)
         self.assertEqual("os_stat_fn", function.c_basename)
 
+    def test_cloning_nonexistent_function_correctly_fails(self):
+        stdout = self.parse_function_should_fail("""
+            cloned = fooooooooooooooooooooooo
+            This is trying to clone a nonexistent function!!
+        """)
+        expected_error = """\
+cls=None, module=<FakeClinic object>, existing='fooooooooooooooooooooooo'
+(cls or module).functions=[]
+Error on line 0:
+Couldn't find existing function 'fooooooooooooooooooooooo'!
+"""
+        self.assertEqual(expected_error, stdout)
+
     def test_return_converter(self):
         function = self.parse_function("""
             module os
diff --git a/Tools/clinic/clinic.py b/Tools/clinic/clinic.py
@@ -4707,11 +4707,9 @@ def state_modulename_name(self, line: str | None) -> None:
                     if existing_function.name == function_name:
                         break
                 else:
-                    existing_function = None
-                if not existing_function:
-                    print("class", cls, "module", module, "existing", existing)
-                    print("cls. functions", cls.functions)
-                    fail("Couldn't find existing function " + repr(existing) + "!")
+                    print(f"{cls=}, {module=}, {existing=}")
+                    print(f"{(cls or module).functions=}")
+                    fail(f"Couldn't find existing function {existing!r}!")
 
                 fields = [x.strip() for x in full_name.split('.')]
                 function_name = fields.pop()
