def test_get_retweeted_tweet_caches(self, get_method): "Should only fetch retweeted Tweet from DB once." retweeted_tweet = TweetFactory(text='Retweeted tweet!', twitter_id=123) tweet = TweetFactory(retweeted_status_id=123) tweet.get_retweeted_tweet() tweet.get_retweeted_tweet() self.assertEqual(get_method.call_count, 1)
def test_get_retweeted_tweet_none(self): tweet = TweetFactory(retweeted_status_id=None) self.assertIsNone(tweet.get_retweeted_tweet())
def test_get_retweeted_tweet(self): retweeted_tweet = TweetFactory(text='Retweeted tweet!', twitter_id=123) tweet = TweetFactory(retweeted_status_id=123) self.assertEqual(tweet.get_retweeted_tweet().text, 'Retweeted tweet!')