def test_patching_event_firing(self):
class A(object):
def f(self):
pass;
patcher.add_start_end_reporting("event1",A,"f")
patcher.add_start_end_reporting("event2",A,"f")
events=[]
with EventCatcher(events):
a= A()
a.f()
self.assertEqual(events,
[
("event2","start",None),
("event1","start",None),
("event1","end",None),
("event2","end",None),
]
)
def test_patching_dedup(self):
class A(object):
def f(self):
pass;
patcher.add_start_end_reporting("event1",A,"f")
patcher.add_start_end_reporting("event2",A,"f")
patcher.add_start_end_reporting("event1",A,"f")
self.assertEqual(patcher._get_existing_wrapped_events(A.f), ["event1","event2"])