Avoid having to fix tests

This commit is contained in:
Ian Cordasco
2014-01-28 20:24:41 -06:00
parent 27eb54a20a
commit c2fab5b4ca
+3 -2
View File
@@ -18,9 +18,10 @@ class RequestException(IOError):
"""
Initialize RequestException with `request` and `response` objects.
"""
self.response = kwargs.pop('response', None)
response = kwargs.pop('response', None)
self.response = response
self.request = kwargs.pop('request', None)
if self.response and not self.request:
if response and not self.request and hasattr(response, 'request'):
self.request = self.response.request
super(RequestException, self).__init__(*args, **kwargs)