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')
