encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-04 12:30:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed response header and status handling in DjangoTestAdapter

Browse Source
This commit is contained in:
Brian Rosner 2016-11-04 17:12:41 -06:00
parent 3cb9d036e8
commit 94a3d6ae80
No known key found for this signature in database
GPG Key ID: A520B7186E677E82
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
rest_framework/test.py
View File

@ -86,13 +86,18 @@ if requests is not None:
# Make the outgoing request via WSGI. # Make the outgoing request via WSGI.
raw_response = self.get_raw_response(request) raw_response = self.get_raw_response(request)
# Gather all response headers
response_headers = [(str(k), str(v)) for k, v in raw_response.items()]
for c in raw_response.cookies.values():
response_headers.append((str('Set-Cookie'), str(c.output(header=''))))
# Build the underlying urllib3.HTTPResponse # Build the underlying urllib3.HTTPResponse
raw_kwargs['status'] = raw_response.status_code raw_kwargs['status'] = raw_response.status_code
raw_kwargs['reason'] = raw_response.status_text raw_kwargs['reason'] = raw_response.reason_phrase
raw_kwargs['headers'] = raw_response.items() raw_kwargs['headers'] = response_headers
raw_kwargs['version'] = 11 raw_kwargs['version'] = 11
raw_kwargs['preload_content'] = False raw_kwargs['preload_content'] = False
raw_kwargs['original_response'] = MockOriginalResponse(raw_response.items()) raw_kwargs['original_response'] = MockOriginalResponse(response_headers)
raw_kwargs['body'] = io.BytesIO(raw_response.content) raw_kwargs['body'] = io.BytesIO(raw_response.content)
raw = requests.packages.urllib3.HTTPResponse(**raw_kwargs) raw = requests.packages.urllib3.HTTPResponse(**raw_kwargs)