def test_process_tweet_previous_older(self, put_mock, find_mock,
checked_mock):
tweet = RawTweetHelper.common_raw_tweet()
self.target.process_tweet(tweet)
assert find_mock.call_count == 3
assert put_mock.call_count == 3
assert checked_mock.call_count == 1
def test_store(self):
tweet = RawTweetHelper.common_raw_tweet()
self.target.store(tweet, ['Emperor', 'Caniggia'])
cooccurrence = self.target.get_first({})
assert cooccurrence is not None
assert cooccurrence['user_id'] == tweet['user_id']
assert cooccurrence['created_at'] == tweet['created_at']
assert cooccurrence['pair'] == ['Emperor', 'Caniggia']
def test_put(self):
tweet = RawTweetHelper.common_raw_tweet()
self.target.put('emperor', tweet, 'Emperor')
hashtag = self.target.find('emperor')
assert hashtag is not None
assert hashtag['tweet_id'] == tweet['_id']
assert hashtag['user_id'] == tweet['user_id']
assert hashtag['created_at'] == tweet['created_at']
assert hashtag['original'] == 'Emperor'
assert hashtag['appearances'] == 1
def test_put_twice(self):
tweet = RawTweetHelper.common_raw_tweet()
self.target.put('emperor', tweet, 'Emperor')
tweet['created_at'] = datetime.strptime('2019-01-22', '%Y-%m-%d')
self.target.put('emperor', tweet, 'Emperor')
hashtag = self.target.find('emperor')
assert hashtag is not None
assert hashtag['tweet_id'] == tweet['_id']
assert hashtag['user_id'] == tweet['user_id']
assert hashtag['created_at'] == datetime.strptime(
'2019-01-22', '%Y-%m-%d')
assert hashtag['original'] == 'Emperor'
assert hashtag['appearances'] == 2
def test_process_retweet_ten_hashtags(self, store_mock, checked_mock):
tweet = RawTweetHelper.common_raw_tweet_ten_hashtags()
self.target.process_tweet(tweet)
assert store_mock.call_count == 0
assert checked_mock.call_count == 1