def test_gets_extra_data_with_helper_with_no_data(self): message = RunnerMessage(self.success_message_sample, None) data = message.get_extra_data() self.assertEqual(data, None)
def test_gets_extra_data_with_helper_method(self): message = RunnerMessage(self.success_message_sample, 'test') data = message.get_extra_data() self.assertEqual(data, 'test')