def get_mocked_step_with_message(self, message): mocked_step = ScriptStep() mocked_step.message = message return mocked_step