def test_get_mentioned_for_reply(self): user = '******' mentioned = ('dialelo', 'mental_floss', '4n_4Wfu1_US3RN4M3') expected_output = list(mentioned) expected_output.append(user) expected_output = map(prepend_at, expected_output) text = "@%s, @%s and @%s" % mentioned status = create_status(user=user, text=text) expected = set(filter(prepend_at, expected_output)) mentioned_for_reply = get_mentioned_for_reply(status) self.assertEqual(expected, set(mentioned_for_reply))
def reply(self): status = self.timelines.get_active_status() if status is None: return if is_DM(status): self.direct_message() return author = get_authors_username(status) mentioned = get_mentioned_for_reply(status) try: mentioned.remove('@%s' % self.user.screen_name) except ValueError: pass self.ui.show_tweet_editor(prompt=_('Reply to %s' % author), content=' '.join(mentioned), done_signal_handler=self.tweet_handler)