Correctly initialize TestResponses

There's a pretty awful hack currently in place to add the reason
attribute to the TestResponse class.  This code correctly initializes
the class, including the reason attribute.

Change-Id: I73384db1f89add051547f4d5db36ab1e647e84ef
Closes-Bug: 1613740
This commit is contained in:
woodnt 2016-09-08 14:14:28 -06:00
parent cec7a02170
commit 2c859f1d6d
1 changed files with 1 additions and 2 deletions

View File

@ -35,11 +35,10 @@ class TestResponse(requests.Response):
It also provides some convenience to initialize with a dict.
"""
reason = ''
def __init__(self, data):
self._text = None
super(TestResponse, self)
super(TestResponse, self).__init__()
if isinstance(data, dict):
self.status_code = data.get('status_code', 200)
self.headers = data.get('headers', None)