python之DRF 响应内容类型在测试期间设置为无

我正在使用 Django Rest Framework(版本 3.6.2)创建 REST API。我已经定义了我的 View 集，它继承自 GenericViewSet 并覆盖了 retrieve 方法来实现自定义行为。

class FooViewSet(viewsets.GenericViewSet): 
    serializer_class = FooSerializer 
 
    def retrieve(self, request, *args, **kwargs): 
        ...  
        serializer = self.get_serializer(data) 
        return Response(serializer.data) 

我想在从浏览器访问此端点时使用 BrowsableAPI，并在访问此端点时接收 json 响应，例如从代码。我已使用以下设置配置 DRF:

    REST_FRAMEWORK = { 
    'DEFAULT_RENDERER_CLASSES': ( 
        'rest_framework.renderers.JSONRenderer', 
        'rest_framework.renderers.BrowsableAPIRenderer', 
    ), 
    'TEST_REQUEST_RENDERER_CLASSES': ( 
        'rest_framework.renderers.JSONRenderer', 
    ), 
    'TEST_REQUEST_DEFAULT_FORMAT':'json' 
} 

一切都按预期工作，我可以从浏览器访问可浏览的 API，当使用 Postman 工具发出请求时，我得到 json 响应。不幸的是，我无法在测试期间获得相同的结果。

class GetFooDetailViewTest(APITestCase): 
 
    def test_get_should_return_json(self): 
        response = self.client.get(self.VIEW_URL) 
        self.assertEqual(response.content_type, "application/json") 

我希望响应将 content_type 设置为 application/json(这是我可以在浏览器和 Postman 的响应中看到的 header )。但此测试失败 - response.content_type 设置为 None。在调试这个测试时，我发现 response._headers 字典看起来像这样

{ 
    'vary': ('Vary', 'Cookie'), 
    'x-frame-options': ('X-Frame-Options', 'SAMEORIGIN'), 
    'content-type': ('Content-Type', 'application/json'), 
    'allow': ('Allow', 'GET, PUT, DELETE, OPTIONS') 
} 

因此似乎设置了正确的 header ，但它没有填充到 content_type 属性中。我错过了什么吗？

请您参考如下方法:

这就是我测试内容类型的方式。在极少数情况下，我的代码自己决定内容类型，所以我个人检查我没有做错什么。 DRF 代码已经过测试。

self.assertEqual("application/json", resp['Content-Type']) 

您只需要依靠 DRF 做正确的事，它不是您可以或不需要测试的东西。例如，您没有测试 DRF 是否正确解析了您的 json 主体。测试服务器与真实服务器不完全相同，但非常接近。例如，您将从 response.data 中获取真实对象，而不是 json 编码/解码的对象。

查看 LiveServerTestCase如果你需要它，但它会更慢。