def test_it_notifies_a_function_call_in_the_current_thread(self): # Arrange call_handler = Mock() process_scoped_tracer = ProcessScopedTracer(call_handler) # Act process_scoped_tracer.start() function1() process_scoped_tracer.stop() # Assert assert_equal(call_handler.on_call.call_count, 1) assert_in('function1', [call[0][0].function_name for call in call_handler.on_call.call_args_list])
def test_it_notifies_a_function_call_in_a_new_thread(self): # Arrange call_handler = Mock() process_scoped_tracer = ProcessScopedTracer(call_handler) # Act process_scoped_tracer.start() new_thread = threading.Thread(target=function1) new_thread.start() new_thread.join() process_scoped_tracer.stop() # Assert assert_in('function1', [call[0][0].function_name for call in call_handler.on_call.call_args_list])