mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-13 05:06:53 +03:00
Fix request body/POST access (#5590)
* Modernize middleware tests * Added a failing test for #5582 * Set data ref on underlying django request
This commit is contained in:
parent
15024f3f07
commit
9f66e8badd
|
@ -250,9 +250,10 @@ class Request(object):
|
||||||
else:
|
else:
|
||||||
self._full_data = self._data
|
self._full_data = self._data
|
||||||
|
|
||||||
# copy files refs to the underlying request so that closable
|
# copy data & files refs to the underlying request so that closable
|
||||||
# objects are handled appropriately.
|
# objects are handled appropriately.
|
||||||
self._request._files = self._files
|
self._request._post = self.POST
|
||||||
|
self._request._files = self.FILES
|
||||||
|
|
||||||
def _load_stream(self):
|
def _load_stream(self):
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -1,34 +1,76 @@
|
||||||
from django.conf.urls import url
|
from django.conf.urls import url
|
||||||
from django.contrib.auth.models import User
|
from django.contrib.auth.models import User
|
||||||
|
from django.http import HttpRequest
|
||||||
from django.test import override_settings
|
from django.test import override_settings
|
||||||
|
|
||||||
from rest_framework.authentication import TokenAuthentication
|
from rest_framework.authentication import TokenAuthentication
|
||||||
from rest_framework.authtoken.models import Token
|
from rest_framework.authtoken.models import Token
|
||||||
|
from rest_framework.request import is_form_media_type
|
||||||
|
from rest_framework.response import Response
|
||||||
from rest_framework.test import APITestCase
|
from rest_framework.test import APITestCase
|
||||||
from rest_framework.views import APIView
|
from rest_framework.views import APIView
|
||||||
|
|
||||||
|
|
||||||
|
class PostView(APIView):
|
||||||
|
def post(self, request):
|
||||||
|
return Response(data=request.data, status=200)
|
||||||
|
|
||||||
|
|
||||||
urlpatterns = [
|
urlpatterns = [
|
||||||
url(r'^$', APIView.as_view(authentication_classes=(TokenAuthentication,))),
|
url(r'^auth$', APIView.as_view(authentication_classes=(TokenAuthentication,))),
|
||||||
|
url(r'^post$', PostView.as_view()),
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
class MyMiddleware(object):
|
class RequestUserMiddleware(object):
|
||||||
|
def __init__(self, get_response):
|
||||||
|
self.get_response = get_response
|
||||||
|
|
||||||
def process_response(self, request, response):
|
def __call__(self, request):
|
||||||
|
response = self.get_response(request)
|
||||||
assert hasattr(request, 'user'), '`user` is not set on request'
|
assert hasattr(request, 'user'), '`user` is not set on request'
|
||||||
assert request.user.is_authenticated(), '`user` is not authenticated'
|
assert request.user.is_authenticated, '`user` is not authenticated'
|
||||||
|
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
class RequestPOSTMiddleware(object):
|
||||||
|
def __init__(self, get_response):
|
||||||
|
self.get_response = get_response
|
||||||
|
|
||||||
|
def __call__(self, request):
|
||||||
|
assert isinstance(request, HttpRequest)
|
||||||
|
|
||||||
|
# Parse body with underlying Django request
|
||||||
|
request.body
|
||||||
|
|
||||||
|
# Process request with DRF view
|
||||||
|
response = self.get_response(request)
|
||||||
|
|
||||||
|
# Ensure request.POST is set as appropriate
|
||||||
|
if is_form_media_type(request.content_type):
|
||||||
|
assert request.POST == {'foo': ['bar']}
|
||||||
|
else:
|
||||||
|
assert request.POST == {}
|
||||||
|
|
||||||
return response
|
return response
|
||||||
|
|
||||||
|
|
||||||
@override_settings(ROOT_URLCONF='tests.test_middleware')
|
@override_settings(ROOT_URLCONF='tests.test_middleware')
|
||||||
class TestMiddleware(APITestCase):
|
class TestMiddleware(APITestCase):
|
||||||
|
|
||||||
|
@override_settings(MIDDLEWARE=('tests.test_middleware.RequestUserMiddleware',))
|
||||||
def test_middleware_can_access_user_when_processing_response(self):
|
def test_middleware_can_access_user_when_processing_response(self):
|
||||||
user = User.objects.create_user('john', 'john@example.com', 'password')
|
user = User.objects.create_user('john', 'john@example.com', 'password')
|
||||||
key = 'abcd1234'
|
key = 'abcd1234'
|
||||||
Token.objects.create(key=key, user=user)
|
Token.objects.create(key=key, user=user)
|
||||||
|
|
||||||
with self.settings(
|
self.client.get('/auth', HTTP_AUTHORIZATION='Token %s' % key)
|
||||||
MIDDLEWARE_CLASSES=('tests.test_middleware.MyMiddleware',)
|
|
||||||
):
|
@override_settings(MIDDLEWARE=('tests.test_middleware.RequestPOSTMiddleware',))
|
||||||
auth = 'Token ' + key
|
def test_middleware_can_access_request_post_when_processing_response(self):
|
||||||
self.client.get('/', HTTP_AUTHORIZATION=auth)
|
response = self.client.post('/post', {'foo': 'bar'})
|
||||||
|
assert response.status_code == 200
|
||||||
|
|
||||||
|
response = self.client.post('/post', {'foo': 'bar'}, format='json')
|
||||||
|
assert response.status_code == 200
|
||||||
|
|
Loading…
Reference in New Issue
Block a user