下面是一个使用APIClient和RequestFactory的示例:
# views.py
from rest_framework.test import APIClient
from rest_framework.test import APIRequestFactory
from rest_framework.views import APIView
from rest_framework.response import Response
class MyView(APIView):
def get(self, request):
data = {'message': 'Hello, World!'}
return Response(data)
# tests.py
from django.test import TestCase
from django.urls import reverse
from rest_framework import status
class MyViewTest(TestCase):
def setUp(self):
self.client = APIClient()
self.factory = APIRequestFactory()
def test_get(self):
url = reverse('my-view')
request = self.factory.get(url)
response = MyView.as_view()(request)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data, {'message': 'Hello, World!'})
def test_get_with_client(self):
url = reverse('my-view')
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data, {'message': 'Hello, World!'})
在这个示例中,MyView
是一个简单的APIView,它返回一个包含"Hello, World!"消息的JSON响应。在测试中,我们使用APIClient
和APIRequestFactory
来模拟HTTP请求,并断言期望的响应。
test_get
方法使用APIRequestFactory
来创建一个GET请求,并将其传递给MyView.as_view()
来获取响应。我们使用断言来验证响应的状态码和数据。
test_get_with_client
方法使用APIClient
来发送GET请求,这样我们就不需要手动创建请求对象。我们同样使用断言来验证响应的状态码和数据。
注意,为了能够在测试中使用基于类的视图(如APIView
),我们需要将视图注册为URLconf,并使用reverse
函数来获取URL。