From c2fd5686950f49128c9d282bf28d440a71225c6b Mon Sep 17 00:00:00 2001 From: Tom Hogans Date: Tue, 16 Aug 2011 01:38:39 -0400 Subject: [PATCH] Added tests for requests.session --- test_requests.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/test_requests.py b/test_requests.py index ff98c41..d6f948e 100755 --- a/test_requests.py +++ b/test_requests.py @@ -13,6 +13,7 @@ except ImportError: import requests +from requests.session import Session HTTPBIN_URL = 'http://httpbin.org/' @@ -455,5 +456,28 @@ class RequestsTestSuite(unittest.TestCase): self.assertEquals(len(r.history), 3) + def test_session_HTTP_200_OK_GET(self): + s = Session() + r = s.get(httpbin('/')) + self.assertEqual(r.status_code, 200) + + def test_session_HTTPS_200_OK_GET(self): + s = Session() + r = s.get(httpsbin('/')) + self.assertEqual(r.status_code, 200) + + def test_session_persistent_headers(self): + heads = {'User-agent': 'Mozilla/5.0'} + s = Session() + s.headers = heads + # Make 2 requests from Session object, should send header both times + r1 = s.get(httpbin('user-agent') + assert heads['User-agent'] in r1.content + r2 = s.get(httpbin('user-agent') + assert heads['User-agent'] in r2.content + self.assertEqual(r.status_code, 200) + + + if __name__ == '__main__': unittest.main() -- 2.7.4