如何在`django rest_framework test`的`APIClient`的标头中添加身份验证令牌

Posted 2023-02-27

【中文标题】如何在`django rest_framework test`的`APIClient`的标头中添加身份验证令牌

【英文标题】：How to add authentication token in header of `APIClient` in `django rest_framework test`

【发布时间】：2018-11-13 16:08:34

【问题描述】：

我将oauth2_provider 用于我的rest_framework。我正在尝试为我的 api 编写测试用例。我已获得访问令牌。但我无法在APIClient 中使用access token 对用户进行身份验证 我希望让这个 curl 命令与 APIClient 一起使用。

curl -H "Authorization: Bearer <your_access_token>" http://localhost:8000/api/v1/users/current/

我试过了

client.get('/api/v1/users/current/',  headers='Authorization': 'Bearer '.format(self.access_token))

和

client.credentials(HTTP_AUTHORIZATION='Token ' + self.access_token)

这里是sn-p的部分

from rest_framework.test import APIClient    
from rest_framework.test import APITestCase
....

class APITest(APITestCase):
    def setUp(self):
        ...
        self.client = APIClient()
        ...
        response = self.client.post('/api/v1/oauth2/token/', post_data)
        self.access_token = response.json()['access_token']

    def test_get_current_user(self):
        client.get('/api/v1/users/current/',  headers='Authorization': 'Bearer '.format(self.access_token))

收到回复

<HttpResponseForbidden status_code=403, "text/html; charset=utf-8">

【问题讨论】：

也许您应该检查这些以使用 DRF 进行测试 django-rest-framework.org/api-guide/testing/#authenticating django-rest-framework.org/api-guide/testing/…

【参考方案1】：

由于您在 curl 中使用 Authorization: Bearer，因此您还应该使用带有 Bearer 字词的 client.credentials 而不是 Token：

client.credentials(HTTP_AUTHORIZATION='Bearer ' + self.access_token)

【讨论】：

我在client.get() 中尝试了与headers= 的各种组合，但忘了尝试这个。我工作。谢谢

这条线放在哪里，之前：client.get()?

收到此错误：ValueError: invalid literal for int() with base 10: '