def test_get_tweets(self, mock_oauthhandler): mock_tweets = [Mock(text="@Tweet")] with patch.object( API, 'user_timeline', return_value=mock_tweets) as mock_api: tweets = get_tweets(10) assert not mock_api.called self.assertEquals([], tweets)
def test_get_tweets(self, mock_oauthhandler): mock_tweets = [Mock(text="@Tweet")] with patch.object( API, 'user_timeline', return_value=mock_tweets) as mock_api: tweets = get_tweets(10) mock_api.assert_called_once_with('pythonglasgow', count=10) formatted_tweet = ('@<a href="http://twitter.com/Tweet" ' 'title="#Tweet on Twitter">Tweet</a>') self.assertEquals([(mock_tweets[0], formatted_tweet), ], tweets)
def test_get_tweets_error(self, mock_oauthhandler): with patch.object( API, 'user_timeline', side_effect=TweepError("")): get_tweets(10)