diff options
Diffstat (limited to 'test_requests.py')
| -rw-r--r-- | test_requests.py | 161 |
1 files changed, 148 insertions, 13 deletions
diff --git a/test_requests.py b/test_requests.py index 4a70982..93b8123 100644 --- a/test_requests.py +++ b/test_requests.py @@ -3,19 +3,29 @@ """Tests for Requests.""" +from __future__ import division import json import os import unittest +import pickle import requests from requests.auth import HTTPDigestAuth +from requests.compat import str + +try: + import StringIO +except ImportError: + import io as StringIO HTTPBIN = os.environ.get('HTTPBIN_URL', 'http://httpbin.org/') + def httpbin(*suffix): """Returns url for HTTPBIN resource.""" return HTTPBIN + '/'.join(suffix) + class RequestsTestCase(unittest.TestCase): _multiprocess_can_split_ = True @@ -28,9 +38,6 @@ class RequestsTestCase(unittest.TestCase): """Teardown.""" pass - def test_assertion(self): - assert 1 - def test_entry_points(self): requests.session @@ -54,6 +61,11 @@ class RequestsTestCase(unittest.TestCase): assert pr.url == req.url assert pr.body == 'life=42' + def test_no_content_length(self): + get_req = requests.Request('GET', httpbin('get')).prepare() + self.assertTrue('Content-Length' not in get_req.headers) + head_req = requests.Request('HEAD', httpbin('head')).prepare() + self.assertTrue('Content-Length' not in head_req.headers) def test_path_is_not_double_encoded(self): request = requests.Request('GET', "http://0.0.0.0/get/test case").prepare() @@ -70,10 +82,6 @@ class RequestsTestCase(unittest.TestCase): self.assertEqual(request.url, "http://example.com/path?key=value&a=b#fragment") - def test_HTTP_200_OK_GET(self): - r = requests.get(httpbin('get')) - self.assertEqual(r.status_code, 200) - def test_HTTP_200_OK_GET_ALTERNATIVE(self): r = requests.Request('GET', httpbin('get')) s = requests.Session() @@ -104,19 +112,29 @@ class RequestsTestCase(unittest.TestCase): r = requests.get(httpbin('get') + '?test=true', params={'q': 'test'}, headers=heads) self.assertEqual(r.status_code, 200) + def test_set_cookie_on_301(self): + s = requests.session() + url = httpbin('cookies/set?foo=bar') + r = s.get(url) + self.assertTrue(s.cookies['foo'] == 'bar') + + def test_cookie_sent_on_redirect(self): + s = requests.session() + s.get(httpbin('cookies/set?foo=bar')) + r = s.get(httpbin('redirect/1')) # redirects to httpbin('get') + self.assertTrue("Cookie" in r.json()["headers"]) + def test_user_agent_transfers(self): heads = { - 'User-agent': - 'Mozilla/5.0 (github.com/kennethreitz/requests)' + 'User-agent': 'Mozilla/5.0 (github.com/kennethreitz/requests)' } r = requests.get(httpbin('user-agent'), headers=heads) self.assertTrue(heads['User-agent'] in r.text) heads = { - 'user-agent': - 'Mozilla/5.0 (github.com/kennethreitz/requests)' + 'user-agent': 'Mozilla/5.0 (github.com/kennethreitz/requests)' } r = requests.get(httpbin('user-agent'), headers=heads) @@ -131,8 +149,6 @@ class RequestsTestCase(unittest.TestCase): self.assertEqual(r.status_code, 200) def test_BASICAUTH_TUPLE_HTTP_200_OK_GET(self): - - auth = ('user', 'pass') url = httpbin('basic-auth', 'user', 'pass') @@ -163,6 +179,18 @@ class RequestsTestCase(unittest.TestCase): r = s.get(url) self.assertEqual(r.status_code, 200) + def test_DIGEST_STREAM(self): + + auth = HTTPDigestAuth('user', 'pass') + url = httpbin('digest-auth', 'auth', 'user', 'pass') + + r = requests.get(url, auth=auth, stream=True) + self.assertNotEqual(r.raw.read(), b'') + + r = requests.get(url, auth=auth, stream=False) + self.assertEqual(r.raw.read(), b'') + + def test_DIGESTAUTH_WRONG_HTTP_401_GET(self): auth = HTTPDigestAuth('user', 'wrongpass') @@ -242,6 +270,9 @@ class RequestsTestCase(unittest.TestCase): requests.get(url, params={'foo': 'foo'}) requests.get(httpbin('ΓΈ'), params={'foo': 'foo'}) + def test_unicode_header_name(self): + requests.put(httpbin('put'), headers={str('Content-Type'): 'application/octet-stream'}, data='\xff') # compat.str is unicode. + def test_urlencoded_get_query_multivalued_param(self): r = requests.get(httpbin('get'), params=dict(test=['foo', 'baz'])) @@ -264,6 +295,110 @@ class RequestsTestCase(unittest.TestCase): self.assertEqual(r.status_code, 200) self.assertTrue(b"text/py-content-type" in r.request.body) + def test_hook_receives_request_arguments(self): + def hook(resp, **kwargs): + assert resp is not None + assert kwargs != {} + + requests.Request('GET', HTTPBIN, hooks={'response': hook}) + + def test_prepared_request_hook(self): + def hook(resp, **kwargs): + resp.hook_working = True + return resp + + req = requests.Request('GET', HTTPBIN, hooks={'response': hook}) + prep = req.prepare() + + s = requests.Session() + resp = s.send(prep) + + self.assertTrue(hasattr(resp, 'hook_working')) + + def test_links(self): + r = requests.Response() + r.headers = { + 'cache-control': 'public, max-age=60, s-maxage=60', + 'connection': 'keep-alive', + 'content-encoding': 'gzip', + 'content-type': 'application/json; charset=utf-8', + 'date': 'Sat, 26 Jan 2013 16:47:56 GMT', + 'etag': '"6ff6a73c0e446c1f61614769e3ceb778"', + 'last-modified': 'Sat, 26 Jan 2013 16:22:39 GMT', + 'link': ('<https://api.github.com/users/kennethreitz/repos?' + 'page=2&per_page=10>; rel="next", <https://api.github.' + 'com/users/kennethreitz/repos?page=7&per_page=10>; ' + ' rel="last"'), + 'server': 'GitHub.com', + 'status': '200 OK', + 'vary': 'Accept', + 'x-content-type-options': 'nosniff', + 'x-github-media-type': 'github.beta', + 'x-ratelimit-limit': '60', + 'x-ratelimit-remaining': '57' + } + self.assertEqual(r.links['next']['rel'], 'next') + + def test_cookie_parameters(self): + key = 'some_cookie' + value = 'some_value' + secure = True + domain = 'test.com' + rest = {'HttpOnly': True} + + jar = requests.cookies.RequestsCookieJar() + jar.set(key, value, secure=secure, domain=domain, rest=rest) + + self.assertEqual(len(jar), 1) + self.assertTrue('some_cookie' in jar) + + cookie = list(jar)[0] + self.assertEqual(cookie.secure, secure) + self.assertEqual(cookie.domain, domain) + self.assertEqual(cookie._rest['HttpOnly'], rest['HttpOnly']) + + def test_time_elapsed_blank(self): + r = requests.get(httpbin('get')) + td = r.elapsed + total_seconds = ((td.microseconds + (td.seconds + td.days * 24 * 3600) + * 10**6) / 10**6) + self.assertTrue(total_seconds > 0.0) + + def test_response_is_iterable(self): + r = requests.Response() + io = StringIO.StringIO('abc') + r.raw = io + self.assertTrue(next(iter(r))) + io.close() + + def test_get_auth_from_url(self): + url = 'http://user:pass@complex.url.com/path?query=yes' + self.assertEqual(('user', 'pass'), + requests.utils.get_auth_from_url(url)) + + def test_cannot_send_unprepared_requests(self): + r = requests.Request(url=HTTPBIN) + self.assertRaises(ValueError, requests.Session().send, r) + + def test_http_error(self): + error = requests.exceptions.HTTPError() + self.assertEqual(error.response, None) + response = requests.Response() + error = requests.exceptions.HTTPError(response=response) + self.assertEqual(error.response, response) + error = requests.exceptions.HTTPError('message', response=response) + self.assertEqual(str(error), 'message') + self.assertEqual(error.response, response) + + def test_session_pickling(self): + r = requests.Request('GET', httpbin('get')) + s = requests.Session() + + s = pickle.loads(pickle.dumps(s)) + + r = s.send(r.prepare()) + self.assertEqual(r.status_code, 200) + if __name__ == '__main__': unittest.main() |