def test_print_all_user_output():
captured_output = StringIO()
sys.stdout = captured_output
users_collector = st.CollectorUserOutput()
get_users_to_tweet_output_test([st.PrintUserOutput(), users_collector])
sys.stdout = sys.__stdout__
assert captured_output.getvalue().count('User(') == len(
users_collector.get_scrapped_users())
def test_user_csv_read_iterator():
file_name = get_temp_test_file_name('csv')
collector = st.CollectorUserOutput()
get_users_to_tweet_output_test([collector, st.CsvUserOutput(file_name)])
iterator = st.UserCsvFileIterator(file_name, 4)
list_from_iterator = []
iterator.open()
while True:
try:
list_from_iterator.extend(next(iterator))
except StopIteration:
break
two_lists_assert_equal(list_from_iterator, collector.get_scrapped_users())
def test_user_json_lines_read_iterator():
file_name = get_temp_test_file_name('jl')
collector = st.CollectorUserOutput()
get_users_to_tweet_output_test([collector, st.JsonLineFileUserOutput(file_name)])
iterator = st.UserJsonLineFileIterator(file_name, 2)
list_from_iterator = []
iterator.open()
while True:
try:
list_from_iterator.extend(next(iterator))
except StopIteration:
break
iterator.close()
two_lists_assert_equal(list_from_iterator, collector.get_scrapped_users())
def get_users() -> List[st.User]:
collect_user_output = st.CollectorUserOutput()
task = st.GetUsersTask(
list(set([tweet.user_name for tweet in get_tweets()]))[:10])
st.GetUsersRunner(task, [collect_user_output]).run()
return collect_user_output.get_scrapped_users()