def test_when_get_no_existed_event_count_should_return_zero(self): # Given p = Packet("receiver", " sender", []) # When event_a_log_count = p.get_event_count("A") event_b_log_count = p.get_event_count("B") # Then self.assertEqual(event_a_log_count, 0) self.assertEqual(event_b_log_count, 0)
def test_when_get_existed_event_count_should_return_log_count(self): # Given e1 = Event("LOGAA", 2, 1, 1, 1, 1) e2 = Event("LOGAB", 3, 1, 1, 1, 1) p = Packet("receiver", " sender", [e1, e2]) # When event_a_log_count = p.get_event_count("A") event_b_log_count = p.get_event_count("B") # Then self.assertEqual(event_a_log_count, 2) self.assertEqual(event_b_log_count, 3)