twitter_tests.py

# -*- coding: utf-8 -*-

from datetime import datetime
from pytest import fixture
from threading import Timer
from time import sleep

from twitter import Twitter
from twitter import TWITTER_CONSUMER_KEY
from twitter import TWITTER_CONSUMER_SECRET
from twitter import TWITTER_ACCESS_TOKEN
from twitter import TWITTER_ACCESS_TOKEN_SECRET


@fixture
def twitter():
    return Twitter(logs_to_cloud=False)


def test_environment_variables():
    assert TWITTER_CONSUMER_KEY
    assert TWITTER_CONSUMER_SECRET
    assert TWITTER_ACCESS_TOKEN
    assert TWITTER_ACCESS_TOKEN_SECRET


def callback(tweet):
    # TODO: Test whether the callback was called.
    assert tweet


def test_streaming(twitter):
    # Let the stream run for two seconds and run it again after a pause.
    Timer(2, twitter.stop_streaming).start()
    twitter.start_streaming(callback)
    sleep(2)
    Timer(2, twitter.stop_streaming).start()
    twitter.start_streaming(callback)


def test_make_tweet_text(twitter):
    assert twitter.make_tweet_text([{
        "name": "Boeing",
        "sentiment": -0.1,
        "ticker": "BA"}],
        "https://twitter.com/realDonaldTrump/status/806134244384899072") == (
        u"Boeing $BA \U0001f44e\n"
        u"https://twitter.com/realDonaldTrump/status/806134244384899072")
    assert twitter.make_tweet_text([{
        "name": "Ford",
        "sentiment": 0.3,
        "ticker": "F"}, {
        "name": "Fiat",
        "root": "Fiat Chrysler Automobiles",
        "sentiment": 0.3,
        "ticker": "FCAU"}],
        "https://twitter.com/realDonaldTrump/status/818461467766824961") == (
        u"Ford $F \U0001f44d\n"
        u"Fiat (Fiat Chrysler Automobiles) $FCAU \U0001f44d\n"
        u"https://twitter.com/realDonaldTrump/status/818461467766824961")
    assert twitter.make_tweet_text([{
        "name": "Lockheed Martin",
        "sentiment": -0.1,
        "ticker": "LMT"}, {
        "name": "Boeing",
        "sentiment": 0.1,
        "ticker": "BA"}],
        "https://twitter.com/realDonaldTrump/status/812061677160202240") == (
        u"Lockheed Martin $LMT \U0001f44e\n"
        u"Boeing $BA \U0001f44d\n"
        u"https://twitter.com/realDonaldTrump/status/812061677160202240")
    assert twitter.make_tweet_text([{
        "name": "General Motors",
        "sentiment": 0,
        "ticker": "GM"}],
        "https://twitter.com/realDonaldTrump/status/821697182235496450") == (
        u"General Motors $GM ¯\_(\u30c4)_/¯\n"
        u"https://twitter.com/realDonaldTrump/status/821697182235496450")


def test_get_tweets(twitter):
    statuses = twitter.get_tweets(["806134244384899072", "812061677160202240"])
    assert len(statuses) == 2
    assert statuses[0]["text"] == (
        "Boeing is building a brand new 747 Air Force One for future president"
        "s, but costs are out of control, more than $4 billion. Cancel order!")
    assert statuses[0]["id_str"] == "806134244384899072"
    assert statuses[0]["user"]["id_str"] == "25073877"
    assert statuses[0]["user"]["screen_name"] == "realDonaldTrump"
    assert statuses[0]["created_at"] == "Tue Dec 06 13:52:35 +0000 2016"
    assert statuses[1]["text"] == (
        "Based on the tremendous cost and cost overruns of the Lockheed Martin"
        " F-35, I have asked Boeing to price-out a comparable F-18 Super Horne"
        "t!")
    assert statuses[1]["id_str"] == "812061677160202240"
    assert statuses[1]["user"]["id_str"] == "25073877"
    assert statuses[1]["user"]["screen_name"] == "realDonaldTrump"
    assert statuses[1]["created_at"] == "Thu Dec 22 22:26:05 +0000 2016"


def test_get_tweet_link(twitter):
    tweet = twitter.get_tweets(["828574430800539648"])[0]
    assert twitter.get_tweet_link(tweet) == (
        "https://twitter.com/realDonaldTrump/status/828574430800539648")