class TweepyResultSetTests(unittest.TestCase):
def setUp(self):
self.results = ResultSet()
for i in ids_fixture:
self.results.append(IdItem(i))
self.results.append(NoIdItem())
def testids(self):
ids = self.results.ids()
self.assertListEqual(ids, ids_fixture)
def testmaxid(self):
self.assertEqual(self.results.max_id, 100)
def testsinceid(self):
self.assertEqual(self.results.since_id, 1)
def extract_users_tweets(handle, number):
"""
Argument Order: handle, number of tweets to extract
Extract's a user's tweets
"""
final = ResultSet() #can change to resultset later if I want
try:
for status in tweepy.Cursor(api.user_timeline,
screen_name=handle,
count=200,
include_rts=True).items(number):
final.append(status)
except:
print("{} is a protected user!")
return []
return final
def pages(cls, _: int) -> Iterator[ResultSet]:
json_name = 'timeline'
(_, kwargs) = mock_cursor.call_args
if 'since_id' not in kwargs:
json_name = 'fav'
json_path = f'{JSON_DIR}/twitter/tweets/{json_name}.json'
tweets: list = load_json(json_path)
tweet_list: ResultSet = ResultSet(max_id=1, since_id=1)
for tweet_dict in tweets:
tweet_list.append(tweepy.Status.parse(tweepy.api, tweet_dict))
return iter([tweet_list])
def setUp(self):
self.results = ResultSet()
for i in ids_fixture:
self.results.append(IdItem(i))
self.results.append(NoIdItem())
