def get_mocked_step_with_poll_question(self, question, script_slug='registration_script'): mocked_poll = Poll() mocked_poll.question = question poll_step = ScriptStep() poll_step.poll = mocked_poll poll_step.script = Script(slug=script_slug) return poll_step