encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-07 13:54:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Py3 compat

Browse Source
This commit is contained in:
Tom Christie 2016-08-17 18:09:19 +01:00
parent 64e19c738f
commit da47c345c0
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
rest_framework/test.py
View File

@ -26,6 +26,21 @@ def force_authenticate(request, user=None, token=None):
if requests is not None: if requests is not None:
class HeaderDict(requests.packages.urllib3._collections.HTTPHeaderDict):
def get_all(self, key):
return self.getheaders(self, key)
class MockOriginalResponse(object):
def __init__(self, headers):
self.msg = HeaderDict(headers)
self.closed = False
def isclosed(self):
return self.closed
def close(self):
self.closed = True
class DjangoTestAdapter(requests.adapters.HTTPAdapter): class DjangoTestAdapter(requests.adapters.HTTPAdapter):
""" """
A transport adapter for `requests`, that makes requests via the A transport adapter for `requests`, that makes requests via the
@ -65,17 +80,6 @@ if requests is not None:
raw_kwargs = {} raw_kwargs = {}
def start_response(wsgi_status, wsgi_headers): def start_response(wsgi_status, wsgi_headers):
class MockOriginalResponse(object):
def __init__(self, headers):
self.msg = requests.packages.urllib3._collections.HTTPHeaderDict(headers)
self.closed = False
def isclosed(self):
return self.closed
def close(self):
self.closed = True
status, _, reason = wsgi_status.partition(' ') status, _, reason = wsgi_status.partition(' ')
raw_kwargs['status'] = int(status) raw_kwargs['status'] = int(status)
raw_kwargs['reason'] = reason raw_kwargs['reason'] = reason