# Attach Cookie header to request.
self.headers['Cookie'] = cookie_header
- # If the request fails but exceptions are suppressed,
- # we'll build a blank response.
- r = None
-
try:
# Send the request.
r = conn.urlopen(
except MaxRetryError, e:
if not self.config.get('safe_mode', False):
raise ConnectionError(e)
+ else:
+ r = None
except (_SSLError, _HTTPError), e:
if not self.config.get('safe_mode', False):
def generate():
chunk = []
- while True:
+ while 1:
c = self.raw.read(1)
if not c:
break
lines = '\n'.join(r.iter_lines())
self.assertEqual(lines, quote)
- def test_timeout(self):
-
- # When not in safe mode, should raise Timeout exception
- with self.assertRaises(requests.exceptions.Timeout):
- r = requests.get(httpbin('stream', '1000'), timeout=0.0001)
-
- # In safe mode, should return a blank response
- r = requests.get(httpbin('stream', '1000'), timeout=0.0001,
- config=dict(safe_mode=True))
- self.assertIsNone(r.content)
-
if __name__ == '__main__':
unittest.main()