Add Session.prepare_request test case.
authorRobert Estelle <robert@btbapp.com>
Wed, 31 Jul 2013 06:02:13 +0000 (23:02 -0700)
committerRobert Estelle <robert@btbapp.com>
Wed, 31 Jul 2013 06:02:13 +0000 (23:02 -0700)
test_requests.py

index ee0883498d2089b1337d6e5a06ffbc240cad4c83..482881da520b24f038d829739c732aebd4eafec3 100755 (executable)
@@ -489,6 +489,23 @@ class RequestsTestCase(unittest.TestCase):
         a.params['_'] = '?'
         self.assertNotEqual(a.params, b.params)
 
+    def test_prepared_from_session(self):
+        class DummyAuth(requests.auth.AuthBase):
+            def __call__(self, r):
+                r.headers['Dummy-Auth-Test'] = 'dummy-auth-test-ok'
+                return r
+
+        req = requests.Request('GET', httpbin('headers'))
+        self.assertEqual(req.auth, None)
+
+        s = requests.Session()
+        s.auth = DummyAuth()
+
+        prep = s.prepare_request(req)
+        resp = s.send(prep)
+
+        self.assertTrue(resp.json()['headers']['Dummy-Auth-Test'], 'dummy-auth-test-ok')
+
     def test_links(self):
         r = requests.Response()
         r.headers = {