def test_one_emoji_in_text_when_get_tweet_length_then_length_plus_1_returned(
self):
twitter = Twitter()
sentence = "this is a test 🔺"
self.assertEqual(
len(sentence) + 1, twitter._get_tweet_length(sentence))
def test_given_one_short_sentence_without_header_when_split_tweets_then_one_tweet_returned(
self):
sentence = "this is a shot sentence"
sentences = [sentence]
twitter = Twitter()
result = twitter._split_tweets(sentences, None)
self.assertEqual(sentences, result)
def test_when_send_dm_error_then_send_dm_called(self):
with patch.object(Twitter, 'client'):
twitter = Twitter()
twitter.client = MagicMock()
dm = "example"
twitter.send_dm(dm)
twitter.client.get_user.assert_called_once_with("aitormagan")
twitter.client.send_direct_message(
twitter.client.get_user.return_value.id, dm)
def test_given_file_cannot_be_downloaded_when_download_file_then_exception_risen(
self, requests_mock):
requests_mock.get.return_value.status_code = 500
url = MagicMock()
with self.assertRaises(MediaNotAccessibleError) as context:
Twitter._download_file(url, MagicMock())
self.assertEqual("File could not be downloaded",
str(context.exception))
requests_mock.get.assert_called_once_with(url)
def test_given_one_short_sentence_with_header_when_split_tweets_then_sentence_with_header_returned(
self):
sentence = "this is a shot sentence"
header = "this is a header"
sentences = [sentence]
twitter = Twitter()
result = twitter._split_tweets(sentences, header)
self.assertEqual([header + " (1/1):\n\n" + sentence], result)
def test_given_sentences_with_length_above_280_with_header_when_split_tweets_then_two_tweet_returned(
self):
sentence = "this is a shot sentence"
header = "this is a header"
sentences = 20 * [sentence]
twitter = Twitter()
result = twitter._split_tweets(sentences, header)
self.assertEqual(2, len(result))
self.assertTrue(result[0].startswith(header + " (1/2):\n\n"))
self.assertTrue(result[1].startswith(header + " (2/2):\n\n"))
def test_given_sentences_with_length_below_280_with_header_when_split_tweets_then_one_tweet_returned(
self):
sentence1 = "this is a shot sentence"
sentence2 = "this is another sentence"
header = "this is a header"
sentences = [sentence1, sentence2]
twitter = Twitter()
result = twitter._split_tweets(sentences, header)
self.assertEqual(
[header + " (1/1):\n\n" + sentence1 + "\n" + sentence2], result)
def test_given_tweet_when_publish_tweet_then_client_called(self):
with patch.object(Twitter, 'client'):
twitter = Twitter()
twitter.client = MagicMock()
tweet = MagicMock()
in_response_to = MagicMock()
tweet_id = twitter.publish_tweet(tweet, in_response_to)
self.assertEqual(twitter.client.update_status.return_value.id,
tweet_id)
twitter.client.update_status.assert_called_once_with(
tweet, in_response_to)
def test_when_publish_sentences_in_tweets_then_split_and_publish(self):
twitter = Twitter()
twitter._split_tweets = MagicMock()
twitter.publish_tweets = MagicMock()
sentences = MagicMock()
header = MagicMock()
last_tweet = MagicMock()
result = twitter.publish_sentences_in_tweets(sentences, header,
last_tweet)
self.assertEqual(twitter.publish_tweets.return_value, result)
twitter._split_tweets.assert_called_once_with(sentences, header)
twitter.publish_tweets.assert_called_once_with(
twitter._split_tweets.return_value, last_tweet=last_tweet)
def test_given_file_can_be_downloaded_when_download_file_then_file_is_written(
self, requests_mock):
chunk1 = MagicMock()
chunk2 = MagicMock()
requests_mock.get.return_value.status_code = 200
requests_mock.get.return_value.__iter__ = lambda x: iter(
[chunk1, chunk2])
url = MagicMock()
file = MagicMock()
Twitter._download_file(url, file)
requests_mock.get.assert_called_once_with(url)
file.write.assert_has_calls([call(chunk1), call(chunk2)])
file.flush.assert_called_once_with()
def test_when_publish_tweets_then_split_and_publish(self):
twitter = Twitter()
tweet1_id = 123
tweet2_id = 456
previous_id = 789
tweet1 = "test1"
tweet2 = "test2"
twitter.publish_tweet = MagicMock(side_effect=[tweet1_id, tweet2_id])
tweets = [tweet1, tweet2]
last_tweet_id = twitter.publish_tweets(tweets, previous_id)
self.assertEqual(tweet2_id, last_tweet_id)
twitter.publish_tweet.assert_has_calls(
[call(tweet1, previous_id),
call(tweet2, tweet1_id)])
def test_given_url_and_text_when_publish_with_media_then_file_downloaded_and_tweet_published(
self, temp_file_mock):
with patch.object(Twitter, 'client'):
twitter = Twitter()
twitter.client = MagicMock()
twitter._download_file = MagicMock()
url = "http://example.com/file.jpg"
text = "this is an example"
in_response_to = MagicMock()
tweet_id = twitter.publish_tweet_with_media(
text, url, in_response_to)
self.assertEqual(twitter.client.update_with_media.return_value.id,
tweet_id)
temp_file_mock.assert_called_once_with(suffix=".png")
with temp_file_mock.return_value as temp_file:
twitter._download_file.assert_called_once_with(url, temp_file)
twitter.client.update_with_media.assert_called_once_with(
temp_file.name, text, in_reply_to_status_id=in_response_to)
def test_given_no_client_when_get_client_then_client_is_built(
self, os_mock, tweepy_mock):
twitter = Twitter()
client = twitter.client
self.assertEqual(client, tweepy_mock.API.return_value)
tweepy_mock.API.assert_called_once_with(
tweepy_mock.OAuthHandler.return_value)
tweepy_mock.OAuthHandler.assert_called_once_with(
os_mock.environ.get.return_value, os_mock.environ.get.return_value)
tweepy_mock.OAuthHandler.return_value.set_access_token.assert_called_once_with(
os_mock.environ.get.return_value, os_mock.environ.get.return_value)
os_mock.environ.get.assert_has_calls([
call("API_SECRET", ""),
call("API_SECRET_KEY", ""),
call("ACCESS_TOKEN", ""),
call("ACCESS_TOKEN_SECRET", "")
])
def test_given_exception_downloading_file_when_publish_with_media_then_tweet_without_media_published(
self, temp_file_mock):
with patch.object(Twitter, 'client'):
twitter = Twitter()
twitter.client = MagicMock()
twitter._download_file = MagicMock(
side_effect=MediaNotAccessibleError())
twitter.publish_tweet = MagicMock()
url = "http://example.com/file.jpg"
text = "this is an example"
in_response_to = MagicMock()
tweet_id = twitter.publish_tweet_with_media(
text, url, in_response_to)
self.assertEqual(twitter.publish_tweet.return_value, tweet_id)
temp_file_mock.assert_called_once_with(suffix=".png")
with temp_file_mock.return_value as temp_file:
twitter._download_file.assert_called_once_with(url, temp_file)
twitter.client.update_with_media.assert_not_called()
twitter.publish_tweet.assert_called_once_with(
text, in_response_to)
def test_no_emoji_in_text_when_get_tweet_length_then_normal_length_returned(
self):
twitter = Twitter()
sentence = "this is a test"
self.assertEqual(len(sentence), twitter._get_tweet_length(sentence))
import sys
import logging
from datetime import datetime, timedelta, date
from urllib.error import HTTPError
from helpers.twitter import Twitter
from helpers.db import Influx, Measurement
from helpers.ministry_report import SpainCovid19MinistryReport
from helpers.reports import get_report_by_ccaa, get_graph_url, get_global_report
from constants import DAYS_WITHOUT_REPORT
twitter = Twitter()
influx = Influx()
def main():
today = datetime.now()
yesterday = subtract_days_ignoring_weekends(today, 1)
data = influx.get_stat_group_by_day(Measurement.PCRS, today)
if not data:
try:
update_database(today)
publish_report(today, yesterday)
except HTTPError:
logging.info("PDF is not available yet...")
except Exception as e:
logging.exception(
"Unhandled exception while trying to publish tweets")
from utils.network import get_network_speed
from utils.system import get_system_uptime
from helpers.status import Status
from helpers.twitter import Twitter
# Get statuss
nwspeed = get_network_speed()
uptime = get_system_uptime()
status = Status(
network_speed=nwspeed,
uptime=uptime,
)
status_msg = status.get_msg()
# Tweet
twitter = Twitter()
twitter.tweet(msg=status_msg)