e91232f722
Allows performing mocked HTTP requests. Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import requests
|
|
|
|
from django.utils.six import BytesIO
|
|
|
|
|
|
class TestClientAdapter(requests.adapters.BaseAdapter):
|
|
def __init__(self, test_case):
|
|
self.test_case = test_case
|
|
|
|
def build_response(self, request, django_response):
|
|
"""
|
|
Build a requests response from a Django response
|
|
"""
|
|
response = requests.Response()
|
|
|
|
# Fallback to None if there's no status_code, for whatever reason.
|
|
response.status_code = getattr(
|
|
django_response, 'status_code', None
|
|
)
|
|
|
|
# Make headers case-insensitive.
|
|
response.headers = requests.structures.CaseInsensitiveDict(
|
|
getattr(django_response, 'headers', {})
|
|
)
|
|
|
|
# Set encoding.
|
|
response.encoding = requests.utils.get_encoding_from_headers(
|
|
headers=response.headers
|
|
)
|
|
response.raw = BytesIO(django_response.getvalue())
|
|
|
|
response.reason = django_response.reason_phrase
|
|
|
|
if isinstance(request.url, bytes):
|
|
response.url = request.url.decode('utf-8')
|
|
else:
|
|
response.url = request.url
|
|
|
|
# Add new cookies from the server.
|
|
requests.cookies.extract_cookies_to_jar(
|
|
jar=response.cookies, request=request,
|
|
response=django_response
|
|
)
|
|
|
|
# Give the Response some context.
|
|
response.request = django_response
|
|
response.connection = self
|
|
|
|
return response
|
|
|
|
def close(self):
|
|
"""
|
|
No connection needs to be closed, but the method must exists
|
|
or a NotImplementedError is raised.
|
|
"""
|
|
|
|
def send(
|
|
self, request=None, stream=False, timeout=None, verify=True,
|
|
cert=None, proxies=None
|
|
):
|
|
"""Craft a Django request based on the attribute of
|
|
requests' request object.
|
|
"""
|
|
return self.build_response(
|
|
request=request, django_response=self.test_case.generic(
|
|
data=request.body,
|
|
headers=request.headers,
|
|
method=request.method,
|
|
viewname=self.test_case.test_view_name,
|
|
)
|
|
)
|