如何从 rest_framework.test.Client 指定接受标头?

cak*_*ins 5 python testing django django-rest-framework

我正在尝试设置 API 端点以根据传入请求的 Accept 标头使用 HTML 或 JSON 进行回复。我已经开始工作了,通过 curl 测试:

> curl --no-proxy localhost -H "Accept: application/json" -X GET http://localhost:8000/feedback/
{"message":"feedback Hello, world!"}

> curl --no-proxy localhost -H "Accept: text/html" -X GET http://localhost:8000/feedback/
<html><body>
<h1>Root</h1>
<h2>feedback Hello, world!</h2>
</body></html>
Run Code Online (Sandbox Code Playgroud)

不过,我不知道如何使用 APITestCase().self.client 来指定应该接受哪些内容。

我的观点看起来像

class Root(APIView):
    renderer_classes = (TemplateHTMLRenderer,JSONRenderer)
    template_name="feedback/root.html"
    def get(self,request,format=None):
        data={"message": "feedback Hello, world!"}
        return Response(data)
Run Code Online (Sandbox Code Playgroud)

我的测试代码看起来像

class RootTests(APITestCase):
    def test_can_get_json(self):
        response = self.client.get('/feedback/',format='json',Accept='application/json')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.accepted_media_type,'application/json')
        js=response.json()
        self.assertIn('message', js)
        self.assertEqual(js['message'],'feedback Hello, world!')
Run Code Online (Sandbox Code Playgroud)

它在 response.accepted_media_type 的测试中死亡。这样做的正确方法是什么?我能找到的所有内容都表明格式参数应该足够了。

Ken*_*ars 8

正如此处正确说明的那样,文档似乎没有过多说明如何使用测试客户端向请求添加标头。但是,该extra参数可以用于此目的,但诀窍是您必须按照 http 标头的确切方式编写它。所以你应该这样做:

self.client.get('/feedback/', HTTP_ACCEPT='application/json') 
Run Code Online (Sandbox Code Playgroud)