def test_tweet_reader_retweet(self):
tweets = [{
"tweet": {
"id_str": "42",
"full_text": "RT @github \\o/"
}
}, {
"tweet": {
"id_str": "43",
"full_text": ""
}
}, {
"tweet": {
"id_str": "49",
"full_text": ""
}
}, {
"tweet": {
"id_str": "44",
"full_text": "RT @google OK, Google"
}
}]
expected = [{"tweet": {"id_str": "42"}}, {"tweet": {"id_str": "44"}}]
actual = []
for idx, val in enumerate(
TweetReader(FakeReader(tweets), filters=["retweets"]).read()):
self.assertEqual(expected[idx]["tweet"]["id_str"],
val["tweet"]["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
def test_tweet_reader_reply(self):
tweets = [{
"tweet": {
"id_str": "12",
"in_reply_to_user_id_str": ""
}
}, {
"tweet": {
"id_str": "14",
"in_reply_to_user_id_str": "200"
}
}, {
"tweet": {
"id_str": "16",
"in_reply_to_user_id_str": ""
}
}, {
"tweet": {
"id_str": "18",
"in_reply_to_user_id_str": "203"
}
}]
expected = [{"tweet": {"id_str": "14"}}, {"tweet": {"id_str": "18"}}]
actual = []
for idx, val in enumerate(
TweetReader(FakeReader(tweets), filters=["replies"]).read()):
self.assertEqual(expected[idx]["tweet"]["id_str"],
val["tweet"]["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
def test_tweet_destroyer_dry_run(self):
tweets = [{
"tweet": {
"id_str": "42",
"full_text": ""
}
}, {
"tweet": {
"id_str": "43",
"full_text": ""
}
}, {
"tweet": {
"id_str": "49",
"full_text": ""
}
}]
api = FakeTwitterApi()
destroyer = TweetDestroyer(api, dry_run=True)
for idx, val in enumerate(TweetReader(FakeReader(tweets)).read()):
destroyer.destroy(val["tweet"]["id_str"])
self.assertEqual(len(api.destroyed_tweets), 0)
def test_tweet_reader_retweets(self):
tweets = [{
"tweet": {
"id_str": "21",
"retweet_count": 0
}
}, {
"tweet": {
"id_str": "22",
"retweet_count": 1
}
}, {
"tweet": {
"id_str": "23",
"retweet_count": 2
}
}]
expected = [{"tweet": {"id_str": "21"}}]
actual = []
for idx, val in enumerate(
TweetReader(FakeReader(tweets), min_retweets=1).read()):
self.assertEqual(expected[idx]["tweet"]["id_str"],
val["tweet"]["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
def test_tweet_reader_spare(self):
tweets = [{
"tweet": {
"id_str": "21"
}
}, {
"tweet": {
"id_str": "22"
}
}, {
"tweet": {
"id_str": "23"
}
}]
expected = [{"tweet": {"id_str": "21"}}]
actual = []
for idx, val in enumerate(
TweetReader(FakeReader(tweets), spare=["22", "23"]).read()):
self.assertEqual(expected[idx]["tweet"]["id_str"],
val["tweet"]["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
def test_tweet_reader_none_date(self):
tweets = [{"tweet": {"id_str": "21", "created_at": "Wed March 06 20:22:06 +0000 2013"}}]
expected = [{"tweet": {"id_str": "21"}}]
actual = []
for idx, val in enumerate(TweetReader(FakeReader(tweets)).read()):
self.assertEqual(expected[idx]["tweet"]["id_str"], val["tweet"]["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
def test_tweet_reader_since_date(self):
tweets = [{"tweet": {"id_str": "23", "created_at": "Thu Apr 23 13:10:23 +0000 2020"}},
{"tweet": {"id_str": "24", "created_at": "Sat Apr 25 14:34:33 +0000 2020"}}]
expected = [{"tweet": {"id_str": "24"}}]
actual = []
for idx, val in enumerate(TweetReader(FakeReader(tweets),
since_date="2020-04-24").read()):
self.assertEqual(expected[idx]["tweet"]["id_str"], val["tweet"]["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
def test_tweet_reader_min_none_arg(self):
tweets = [{"id_str": "21", "retweet_count": 0}]
expected = [{"id_str": "21"}]
actual = []
for idx, val in enumerate(
TweetReader(FakeReader(tweets),
min_likes=None,
min_retweets=None).read()):
self.assertEqual(expected[idx]["id_str"], val["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
def test_tweet_reader_date(self):
tweets = [{
"id_str": "21",
"created_at": "Wed March 06 20:22:06 +0000 2013"
}, {
"id_str": "22",
"created_at": "Thu March 05 20:22:06 +0000 2014"
}]
expected = [{"id_str": "21"}]
actual = []
for idx, val in enumerate(
TweetReader(FakeReader(tweets),
until_date="2014-02-01").read()):
self.assertEqual(expected[idx]["id_str"], val["id_str"])
actual.append(val)
self.assertEqual(len(expected), len(actual))
