python - 使用 APIRequestFactory 对 Django 的 rest api 进行单元测试时无法获取 url 参数

Question

我使用 Django 的 rest_framework 创建了一个 rest api .它是使用基于类的 View 创建的。

View .py

import logging
from rest_framework.views import APIView
from rest_framework.authentication import BasicAuthentication, SessionAuthentication, TokenAuthentication
from rest_framework.response import Response

from handlers.product import get_data

log = logging.getLogger(__name__)

class ProductView(APIView):
    authentication_classes = (TokenAuthentication, BasicAuthentication, SessionAuthentication)

    def get(self, request, **kwargs):
        try:
            product_id = kwargs['product_id']
            response = get_data(product_id)
            return Response(response)
        except KeyError as key:
            log.exception(key)
            return Response({'error_msg': '{} is required'.format(key)}, status=400)

urls.py

url(r'^product/(?P<product_id>[0-9]+)/data/$', ProductView.as_view(), name='product'),

这就是我尝试过的，

import pytest
from django.contrib.auth.models import User
from mixer.backend.django import mixer
from rest_framework.authtoken.models import Token
from rest_framework.test import APIRequestFactory

from views import *

pytestmark = pytest.mark.django_db


@pytest.fixture()
def token():
    user = mixer.blend(User)
    return Token.objects.create(user=user)


class TestProductView:

    def test_get_data(self, token):
        factory = APIRequestFactory()
        request = factory.get('/product/100230/data/', HTTP_AUTHORIZATION='Token {}'.format(token))
        response = ProductView.as_view()(request)
        assert response.status_code == 200

我的测试失败，因为我得到了一个异常(exception)，

KeyError: 'product_id'

但是当我运行我的应用程序并从浏览器请求 api 时，它从 url 中获取 product_id。

请注意，我真的必须使用 kwargs在 get api 方法，因为稍后我想让 get_data 通用并将所有 url 参数和查询参数传递给它。

更新 (29/01/2018)

Django 测试客户端 ( django.test.Client ) 来救援了!

我已经开始使用它而不是 RequestFactory每当我有 url 参数时。我不确定我是否应该把它作为这篇文章的答案。

Answer 1

RequestFactory 提供了一种生成请求实例的方法，该实例可用作任何 View 的第一个参数。下面 View 中的“请求”:

def view(request, url_parameter):
    ...

django.test.Client 将模拟 URL 上的 GET 和 POST 请求并观察响应——从低级 HTTP(结果 header 和状态代码)到页面内容的所有内容。
这是直接来自文档。但我会为任何展望 future 的人提供答案。

django.test.Client #Simulate 'GET' and 'POST' sides.