时间戳

首页 > 解决方案 > 如何在 django rest 框架中传递用户令牌以进行 API 测试？

django - 如何在 django rest 框架中传递用户令牌以进行 API 测试？

问题描述

我正在编写一些测试来检查与我的 API 的连接。

我已经通过令牌进行了识别，并且我成功地为特定测试用户检索了令牌：

token = Token.objects.get(user__username='testuser')

我正在努力的是使用该令牌来创建一个成功的 API 请求，如下所示：

client = APIClient(HTTP_AUTHORIZATION='Token ' + token.key)
response = client.get('/patientFull/1/',headers={'Authorization': 'Token ' + token.key})

我一直在寻找许多方法来完成这项工作，这些是我尝试做的一些方法：

response = requests.get('http://127.0.0.1:8000/patientFull/1/',headers={'Authorization': 'Token ' + token.key} )


client = APIClient()
client.credentials(HTTP_AUTHORIZATION='Token ' + token.key)
response = client.get('/patientFull/1/')

该测试是一个简单的断言来检查响应是否有来自服务器的 200 OK HTTP 响应。

以上所有这些方式都会返回 403 HTTP 响应。

这是我的测试的完整代码（我正在使用夹具来用测试数据填充我的测试数据库）：

import json
import requests
from rest_framework.authtoken.models import Token
from rest_framework.test import APIRequestFactory, APITestCase, APIClient


class CustomerAPITestBack(APITestCase):
    fixtures = ['new-fixtures.json']

    def testDE(self):
        token = Token.objects.get(user__username='jpmichel')
        client = APIClient(HTTP_AUTHORIZATION='Token ' + token.key)
        response = client.get('/patientFull/1/',headers={'Authorization': 'Token ' + token.key})
        self.assertEqual(200, response.status_code)

我已经为令牌配置了我的 settings.py 文件：

REST_FRAMEWORK = {
    'DEFAULT_AUTHENTICATION_CLASSES': (
        'rest_framework.authentication.BasicAuthentication',
        'rest_framework.authentication.SessionAuthentication',
        'PHCAdmin.authentication.tokenAuthentication.ExpiringTokenAuthentication',
    ),
    'EXCEPTION_HANDLER': 'PHCAdmin.functions.pxlth_exception_handler',
}

REST_FRAMEWORK_EXPIRY_TIME = 12 # in hours
REST_FRAMEWORK_PASSWORD_RENEWALS = 90 # in days

如果我禁用令牌身份验证，则此测试通过（GET 返回 200 OK）

我应该如何执行我的 GET 请求，以便它使用令牌识别为有效用户并返回 200 OK HTTP 响应？

标签： djangodjango-rest-frameworkdjango-testing

解决方案

只是为了结束这个问题：经过一番研究，我发现服务器上的令牌与本地机器上的不一样，我只需要更新双方的令牌。上面的代码工作正常。

推荐阅读