Move recv_chunk() into a stand alone function. Further minimize memory use by using...

[z_archive/twitter.git] / tests / test_sanity.py

diff --git a/tests/test_sanity.py b/tests/test_sanity.py
index 742bd7a99f6af8d07859ed77b6cc1377a7e40ca0..06cefafa82e9891235477b2db5480744cdaa4d75 100644 (file)
--- a/tests/test_sanity.py
+++ b/tests/test_sanity.py
@@ -1,17 +1,19 @@
+# encoding: utf8
 
 from random import choice
+import time
 
-from twitter import Twitter, NoAuth, OAuth, read_token_file
+from twitter import Twitter, NoAuth, OAuth, read_token_file, TwitterHTTPError
 from twitter.cmdline import CONSUMER_KEY, CONSUMER_SECRET
 
 noauth = NoAuth()
 oauth = OAuth(*read_token_file('tests/oauth_creds')
                + (CONSUMER_KEY, CONSUMER_SECRET))
 
-twitter = Twitter(domain='api.twitter.com',
-                  auth=oauth,
-                  api_version='1')
-twitter_na = Twitter(domain='api.twitter.com', auth=noauth, api_version='1')
+twitter11 = Twitter(domain='api.twitter.com',
+                    auth=oauth,
+                    api_version='1.1')
+twitter11_na = Twitter(domain='api.twitter.com', auth=noauth, api_version='1.1')
 
 
 AZaz = "abcdefghijklmnopqrstuvwxyz1234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ"
@@ -20,29 +22,40 @@ def get_random_str():
     return ''.join(choice(AZaz) for _ in range(10))
 
 
-def test_API_get_some_public_tweets():
-    updates = twitter_na.statuses.public_timeline()
-    assert updates
-    assert updates[0]['created_at']
-
-
 def test_API_set_tweet():
     random_tweet = "A random tweet " + get_random_str()
-    twitter.statuses.update(status=random_tweet)
-
-    recent = twitter.statuses.user_timeline()
+    twitter11.statuses.update(status=random_tweet)
+    time.sleep(2)
+    recent = twitter11.statuses.home_timeline()
     assert recent
+    assert isinstance(recent.rate_limit_remaining, int)
+    assert isinstance(recent.rate_limit_reset, int)
     assert random_tweet == recent[0]['text']
 
 
-def test_API_friendship_exists():
-    assert True == twitter.friendships.exists(
-        user_a='ptttest0001', user_b='sixohsix')
-    assert False == twitter.friendships.exists(
-        user_a='gruber', user_b='ptttest0001')
+def test_API_set_unicode_tweet():
+    random_tweet = u"A random tweet with unicode üøπ" + get_random_str()
+    twitter11.statuses.update(status=random_tweet)
+
+    recent = twitter11.statuses.home_timeline()
+    assert recent
+    assert random_tweet == recent[0]['text']
 
 
 def test_search():
-    t_search = Twitter(domain='search.twitter.com')
-    results = t_search.search(q='foo')
+    results = twitter11.search.tweets(q='foo')
     assert results
+
+
+def test_get_trends_3():
+    # Of course they broke it all again in 1.1...
+    assert twitter11.trends.place(_id=1)
+
+def test_TwitterHTTPError_raised_for_invalid_oauth():
+    test_passed = False
+    try:
+        twitter11_na.statuses.mentions_timeline()
+    except TwitterHTTPError:
+        # this is the error we are looking for :)
+        test_passed = True
+    assert test_passed