gh-122029: Log call events in sys.setprofile when it's a method with c function (GH-122072)

Log call events in sys.setprofile when it is a method with a C function.
This commit is contained in:
Tian Gao 2024-07-23 15:25:26 -07:00 committed by GitHub
parent 41a91bd67f
commit e91ef13861
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 28 additions and 0 deletions

View file

@ -479,6 +479,20 @@ class TestEdgeCases(unittest.TestCase):
sys.setprofile(lambda *args: None)
f()
def test_method_with_c_function(self):
# gh-122029
# When we have a PyMethodObject whose im_func is a C function, we
# should record both the call and the return. f = classmethod(repr)
# is just a way to create a PyMethodObject with a C function.
class A:
f = classmethod(repr)
events = []
sys.setprofile(lambda frame, event, args: events.append(event))
A().f()
sys.setprofile(None)
# The last c_call is the call to sys.setprofile
self.assertEqual(events, ['c_call', 'c_return', 'c_call'])
if __name__ == "__main__":
unittest.main()