mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-07-27 08:29:59 +03:00
Add unit test
This commit is contained in:
parent
0768339e57
commit
34c7c9a771
|
@ -10,6 +10,7 @@ from django.test import TestCase, override_settings
|
||||||
from django.urls import path
|
from django.urls import path
|
||||||
|
|
||||||
from rest_framework import fields, serializers
|
from rest_framework import fields, serializers
|
||||||
|
from rest_framework.authtoken.models import Token
|
||||||
from rest_framework.decorators import api_view
|
from rest_framework.decorators import api_view
|
||||||
from rest_framework.response import Response
|
from rest_framework.response import Response
|
||||||
from rest_framework.test import (
|
from rest_framework.test import (
|
||||||
|
@ -19,10 +20,12 @@ from rest_framework.test import (
|
||||||
|
|
||||||
@api_view(['GET', 'POST', 'PUT', 'PATCH', 'DELETE', 'OPTIONS'])
|
@api_view(['GET', 'POST', 'PUT', 'PATCH', 'DELETE', 'OPTIONS'])
|
||||||
def view(request):
|
def view(request):
|
||||||
return Response({
|
data = {'auth': request.META.get('HTTP_AUTHORIZATION', b'')}
|
||||||
'auth': request.META.get('HTTP_AUTHORIZATION', b''),
|
if request.user:
|
||||||
'user': request.user.username
|
data['user'] = request.user.username
|
||||||
})
|
if request.auth:
|
||||||
|
data['token'] = request.auth.key
|
||||||
|
return Response(data)
|
||||||
|
|
||||||
|
|
||||||
@api_view(['GET', 'POST'])
|
@api_view(['GET', 'POST'])
|
||||||
|
@ -82,10 +85,25 @@ class TestAPITestClient(TestCase):
|
||||||
"""
|
"""
|
||||||
Setting `.force_authenticate()` forcibly authenticates each request.
|
Setting `.force_authenticate()` forcibly authenticates each request.
|
||||||
"""
|
"""
|
||||||
|
# User only
|
||||||
user = User.objects.create_user('example', 'example@example.com')
|
user = User.objects.create_user('example', 'example@example.com')
|
||||||
self.client.force_authenticate(user)
|
self.client.force_authenticate(user=user)
|
||||||
response = self.client.get('/view/')
|
response = self.client.get('/view/')
|
||||||
assert response.data['user'] == 'example'
|
assert response.data['user'] == 'example'
|
||||||
|
assert 'token' not in response.data
|
||||||
|
|
||||||
|
# Token only
|
||||||
|
token = Token.objects.create(key='xyz', user=user)
|
||||||
|
self.client.force_authenticate(token=token)
|
||||||
|
response = self.client.get('/view/')
|
||||||
|
assert response.data['token'] == 'xyz'
|
||||||
|
assert 'user' not in response.data
|
||||||
|
|
||||||
|
# User and token
|
||||||
|
self.client.force_authenticate(user=user, token=token)
|
||||||
|
response = self.client.get('/view/')
|
||||||
|
assert response.data['user'] == 'example'
|
||||||
|
assert response.data['token'] == 'xyz'
|
||||||
|
|
||||||
def test_force_authenticate_with_sessions(self):
|
def test_force_authenticate_with_sessions(self):
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Reference in New Issue
Block a user