tests/test_sanity.py

   1 # encoding: utf-8
   2 from __future__ import unicode_literals
   3
   4 import os
   5 from random import choice
   6 import time
   7 import pickle
   8 import json
   9
  10 from twitter import Twitter, NoAuth, OAuth, read_token_file, TwitterHTTPError
  11 from twitter.api import TwitterDictResponse, TwitterListResponse
  12 from twitter.cmdline import CONSUMER_KEY, CONSUMER_SECRET
  13
  14 noauth = NoAuth()
  15 oauth = OAuth(*read_token_file('tests/oauth_creds')
  16               + (CONSUMER_KEY, CONSUMER_SECRET))
  17
  18 twitter11 = Twitter(domain='api.twitter.com',
  19                     auth=oauth,
  20                     api_version='1.1')
  21
  22 twitter11_na = Twitter(domain='api.twitter.com',
  23                        auth=noauth,
  24                        api_version='1.1')
  25
  26 AZaz = "abcdefghijklmnopqrstuvwxyz1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  27
  28
  29 def get_random_str():
  30     return ''.join(choice(AZaz) for _ in range(10))
  31
  32
  33 def test_API_set_tweet():
  34     random_tweet = "A random tweet " + get_random_str()
  35     twitter11.statuses.update(status=random_tweet)
  36     time.sleep(5)
  37     recent = twitter11.statuses.home_timeline()
  38     assert recent
  39     assert isinstance(recent.rate_limit_remaining, int)
  40     assert isinstance(recent.rate_limit_reset, int)
  41     texts = [tweet['text'] for tweet in recent]
  42     assert random_tweet in texts
  43
  44
  45 def test_API_set_unicode_tweet():
  46     random_tweet = "A random tweet with unicode üøπ" + get_random_str()
  47     twitter11.statuses.update(status=random_tweet)
  48     time.sleep(5)
  49     recent = twitter11.statuses.home_timeline()
  50     assert recent
  51     texts = [tweet['text'] for tweet in recent]
  52     assert random_tweet in texts
  53
  54
  55 def clean_link(text):
  56     pos = text.find(" http://t.co")
  57     if pos != -1:
  58         return text[:pos]
  59     return text
  60
  61 __location__ = os.path.realpath(
  62     os.path.join(os.getcwd(), os.path.dirname(__file__)))
  63
  64 def test_API_set_unicode_twitpic(base64=False):
  65     random_tweet = "A random twitpic from %s with unicode üøπ" % \
  66                     ("base64" if base64 else "file") + get_random_str()
  67     if base64:
  68         img = b"iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAIAAACQkWg2AAAAAXNSR0IArs4c6QAAAAlwSFlzAAALEwAACxMBAJqcGAAAAAd0SU1FB94JFhMBAJv5kaUAAAAZdEVYdENvbW1lbnQAQ3JlYXRlZCB3aXRoIEdJTVBXgQ4XAAAA4UlEQVQoz7WSIZLGIAxG6c5OFZjianBcIOfgPkju1DsEBWfAUEcNGGpY8Xe7dDoVFRvHfO8NJGRorZE39UVe1nd/WNfVObcsi3OOEAIASikAmOf5D2q/FWPUWgshKKWfiFIqhNBaxxhPjPQ05/z+Bs557xw9hBC89ymlu5BS8t6HEC5NW2sR8alRRLTWXoRSSinlSejT12M9BAAAgCeoTw9BSimlfBIu6WdYtVZEVErdaaUUItZaL/9wOsaY83YAMMb0dGtt6Jdv3/ec87ZtOWdCCGNsmibG2DiOJzP8+7b+AAOmsiPxyHWCAAAAAElFTkSuQmCC"
  69     else:
  70         with open(os.path.join(__location__, "test.png"), "rb") as f:
  71             img = f.read()
  72     params = {"status": random_tweet, "media[]": img}
  73     if base64:
  74         params["_base64"] = True
  75     twitter11.statuses.update_with_media(**params)
  76     time.sleep(5)
  77     recent = twitter11.statuses.home_timeline()
  78     assert recent
  79     texts = [clean_link(tweet['text']) for tweet in recent]
  80     assert random_tweet in texts
  81
  82 def test_API_set_unicode_twitpic_base64():
  83     test_API_set_unicode_twitpic(base64=True)
  84
  85
  86 def test_search():
  87     # In 1.1, search works on api.twitter.com not search.twitter.com
  88     # and requires authorisation
  89     results = twitter11.search.tweets(q='foo')
  90     assert results
  91
  92
  93 def test_get_trends():
  94     # This is one method of inserting parameters, using named
  95     # underscore params.
  96     world_trends = twitter11.trends.available(_woeid=1)
  97     assert world_trends
  98
  99
 100 def test_get_trends_2():
 101     # This is a nicer variation of the same call as above.
 102     world_trends = twitter11.trends._(1)
 103     assert world_trends
 104
 105
 106 def test_get_trends_3():
 107     # Of course they broke it all again in 1.1...
 108     assert twitter11.trends.place(_id=1)
 109
 110
 111 def test_TwitterHTTPError_raised_for_invalid_oauth():
 112     test_passed = False
 113     try:
 114         twitter11_na.statuses.mentions_timeline()
 115     except TwitterHTTPError:
 116         # this is the error we are looking for :)
 117         test_passed = True
 118     assert test_passed
 119
 120
 121 def test_picklability():
 122     res = TwitterDictResponse({'a': 'b'})
 123     p = pickle.dumps(res)
 124     res2 = pickle.loads(p)
 125     assert res == res2
 126     assert res2['a'] == 'b'
 127
 128     res = TwitterListResponse([1, 2, 3])
 129     p = pickle.dumps(res)
 130     res2 = pickle.loads(p)
 131     assert res == res2
 132     assert res2[2] == 3
 133
 134
 135 def test_jsonifability():
 136     res = TwitterDictResponse({'a': 'b'})
 137     p = json.dumps(res)
 138     res2 = json.loads(p)
 139     assert res == res2
 140     assert res2['a'] == 'b'
 141
 142     res = TwitterListResponse([1, 2, 3])
 143     p = json.dumps(res)
 144     res2 = json.loads(p)
 145     assert res == res2
 146     assert res2[2] == 3