Fix some api2 tests.

diff --git a/tests/test_api2.py b/tests/test_api2.py
index 5de0468dccbbb341dd0a0e2216b48ed066bf55fe..0ca8f98a62af64135681a558e7543f9727af5913 100644 (file)
--- a/tests/test_api2.py
+++ b/tests/test_api2.py
@@ -58,8 +58,8 @@ def test_post_a_new_tweet():
 
 
 def test_can_get_raw_response():
-    raw_api = TwitterAPI(return_raw_response=True)
-    res = raw_api.get("statuses/public_timeline")
+    raw_api = TwitterAPI(return_raw_response=True, auth=oauth)
+    res = raw_api.get("statuses/home_timeline")
     assert res
     assert res.headers
     assert res.json
@@ -72,4 +72,5 @@ def test_can_stream_some_tweets():
         assert tweet
         tweets += 1
         if tweets > 2:
-            break
+            return
+

diff --git a/twitter/api2.py b/twitter/api2.py
index 7dc0bc2e264e7cb78bc1670e53a6d97d5ea7468b..4d9d093b5f6b498812ea833d97ff287e3bf9ce4d 100644 (file)
--- a/twitter/api2.py
+++ b/twitter/api2.py
@@ -102,7 +102,7 @@ class TwitterAPI(object):
                  secure=True, stream=False, return_raw_response=False):
         """
         `host`        host to connect to (api.twitter.com)
-        `api_ver`     API version 
+        `api_ver`     API version
         """
         self.host = host
         self.api_ver = api_ver
@@ -117,7 +117,7 @@ class TwitterAPI(object):
                                          path, kwargs)
         data = self.auth.encode_params(url, 'GET', remaining_params)
         headers = self.auth.generate_headers()
-        res = requests.get(url + '?' + data, headers=headers)
+        res = requests.get(url + '?' + data, headers=headers, prefetch=(not self.stream))
         return handle_res(res, self.return_raw_response, self.stream)
 
 
@@ -126,7 +126,7 @@ class TwitterAPI(object):
                                          path, kwargs)
         data = self.auth.encode_params(url, 'POST', remaining_params)
         headers = self.auth.generate_headers()
-        res = requests.post(url, data=data, headers=headers)
+        res = requests.post(url, data=data, headers=headers, prefetch=(not self.stream))
         return handle_res(res, self.return_raw_response, self.stream)