def test_performance(self): EventBroker.get_instance().add(self.log) start = time.process_time() self.protocol() finish = time.process_time() self.assertTrue(finish - start < 1.0)
def test_add_listener(self): EventBroker.get_instance().add(self.log) expected_results = [] # Test positional args self.my_object.event_A(1, 2) expected_results.append({ 'arguments': { 'arg1': 1, 'arg2': 2, 'arg3': 'foo', 'self': self.my_object }, 'name': 'my-event-A', 'function': 'TraceTestCase.MyClass.event_A', 'result': 100 }) self.assertDictEqual(expected_results[-1], self.events[-1]) # Test named args self.my_object.event_A(arg1=1, arg2=2) expected_results.append({ 'arguments': { 'arg1': 1, 'arg2': 2, 'arg3': 'foo', 'self': self.my_object }, 'name': 'my-event-A', 'function': 'TraceTestCase.MyClass.event_A', 'result': 100 }) self.assertDictEqual(expected_results[-1], self.events[-1]) # Override defaults, and use mixed positonal/named args self.my_object.event_A(1, arg2=2, arg3='bar') expected_results.append({ 'arguments': { 'arg1': 1, 'arg2': 2, 'arg3': 'bar', 'self': self.my_object }, 'name': 'my-event-A', 'function': 'TraceTestCase.MyClass.event_A', 'result': 100 }) self.assertDictEqual(expected_results[-1], self.events[-1]) # Call event with default name self.my_object.event_B(1) expected_results.append({ 'arguments': { 'arg1': 1, 'self': self.my_object }, 'name': 'TraceTestCase.MyClass.event_B', 'function': 'TraceTestCase.MyClass.event_B', 'result': None }) self.assertDictEqual(expected_results[-1], self.events[-1])