def test_get_tweet_api_limited(self, mock_get_user): ''' Ensure api limit is passed on to the caller ''' mock_get_user.side_effect = RateLimitError('Limit Exceeded') lookup = TwitterLookup() with self.assertRaises(RateLimitError): lookup.get_tweet(1)
def test_get_tweet_no_result(self, mock_get_status): mock_get_status.side_effect = TweepError('test error', api_code=144) lookup = TwitterLookup() self.assertFalse(lookup.get_tweet(1))
def test_get_tweet(self, mock_get_status): lookup = TwitterLookup() lookup.get_tweet(1) mock_get_status.assert_called_once_with(1)
def test_get_user_no_result(self, mock_get_user): ''' Check handling tweepy error when no result ''' mock_get_user.side_effect = TweepError('test error', api_code=50) lookup = TwitterLookup() self.assertFalse(lookup.get_user('does not exist'))
def test_get_user_id(self, mock_get_user): lookup = TwitterLookup() lookup.get_user(1) mock_get_user.assert_called_once_with(id=1)
def test_get_user_name(self, mock_get_user): lookup = TwitterLookup() lookup.get_user('test_user') mock_get_user.assert_called_once_with(id='test_user')