encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-06-20 21:43:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Attempt to fix the tests

Browse Source
This commit is contained in:
Bruno Alla 2024-08-30 18:41:09 +01:00
parent 48616f6f28
commit 6b55ccc22a
No known key found for this signature in database
1 changed files with 32 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
tests/test_middleware.py
View File

@ -13,6 +13,7 @@ from rest_framework.authentication import (
BasicAuthentication, TokenAuthentication BasicAuthentication, TokenAuthentication
) )
from rest_framework.authtoken.models import Token from rest_framework.authtoken.models import Token
from rest_framework.decorators import api_view
from rest_framework.request import is_form_media_type from rest_framework.request import is_form_media_type
from rest_framework.response import Response from rest_framework.response import Response
from rest_framework.test import APITestCase from rest_framework.test import APITestCase
@ -24,24 +25,34 @@ class PostAPIView(APIView):
return Response(data=request.data, status=200) return Response(data=request.data, status=200)
class GetAPIView(APIView): with override_settings(
def get(self, request): REST_FRAMEWORK={
return Response(data={"status": "ok"}, status=200) 'DEFAULT_PERMISSION_CLASSES': [
'rest_framework.permissions.IsAuthenticated',
],
}
):
class GetAPIView(APIView):
def get(self, request):
return Response(data={"status": "ok"}, status=200)
class GetView(View):
def get(self, request):
return HttpResponse("OK", status=200)
class GetView(View): @api_view(['GET'])
def get(self, request): def get_func_view(request):
return HttpResponse("OK", status=200) return HttpResponse("OK", status=200)
urlpatterns = [
urlpatterns = [ path('api/auth', APIView.as_view(authentication_classes=(TokenAuthentication,))),
path('api/auth', APIView.as_view(authentication_classes=(TokenAuthentication,))), path('api/post', PostAPIView.as_view()),
path('api/post', PostAPIView.as_view()), path('api/get', GetAPIView.as_view()),
path('api/get', GetAPIView.as_view()), path('api/get-func', get_func_view),
path('api/basic', GetAPIView.as_view(authentication_classes=(BasicAuthentication,))), path('api/basic', GetAPIView.as_view(authentication_classes=(BasicAuthentication,))),
path('api/token', GetAPIView.as_view(authentication_classes=(TokenAuthentication,))), path('api/token', GetAPIView.as_view(authentication_classes=(TokenAuthentication,))),
path('get', GetView.as_view()), path('get', GetView.as_view()),
] ]
class RequestUserMiddleware: class RequestUserMiddleware:
@ -134,6 +145,13 @@ class TestLoginRequiredMiddleware(APITestCase):
response = self.client.get('/api/get') response = self.client.get('/api/get')
assert response.status_code == status.HTTP_200_OK assert response.status_code == status.HTTP_200_OK
def test_allows_request_when_authenticated_function_view(self):
user = User.objects.create_user('john', 'john@example.com', 'password')
self.client.force_login(user)
response = self.client.get('/api/get-func')
assert response.status_code == status.HTTP_200_OK
def test_allows_request_when_token_authenticated(self): def test_allows_request_when_token_authenticated(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'