python - 使用凭据和 djoser 测试 Django API 登录

Question

我目前正在尝试使用 Django 测试我的 API 端点。到目前为止，我已经使用了 APITestCase，但在登录和使用我的 token 时遇到了问题。

我有一个密码为“1”的 Clark 用户，我让他登录，检索他的 token 并将其添加到 header ，然后再发送我的请求。但我仍然收到“Bad Credential Error”。

    connection_data = {'email':self.clark_user.email, 'password': '1'}
    self.client.login(email=self.clark_user.email, password='1')     

    token = self.client.post(settings.SERVER_HOST+'/api/v1/auth/login', connection_data)
    self.client.credentials(HTTP_AUTHORIZATION='Token ' + token.data['auth_token'])

    response = self.client.patch(self.url_people, serializer.data, format='json')
    self.assertEqual(response.status_code, status.HTTP_200_OK)
    self.client.logout()

我的堆栈如下:

• Django 休息框架 ( http://www.django-rest-framework.org/api-guide/testing/ )
• 左塞尔 ( https://github.com/sunscrapers/djoser )

和

• 从 rest_framework.test 导入 APITestCase
• from core.test_utils import BaseTest 用于我的测试。

是我用于测试的库。

Answer 1

Django REST 框架有 APIClient 类来满足这个目的。我认为与其通过 http 获取 Token，不如直接从 DB 中提取。代码看起来像

from rest_framework.authtoken.models import Token
from rest_framework.test import APIClient

# Include an appropriate `Authorization:` header on all requests.
token = Token.objects.get(user__username=self.clark_user.username)
client = APIClient()
client.credentials(HTTP_AUTHORIZATION='Token ' + token.key)

response = client.patch(self.url_people, serializer.data, format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
client.logout()

文档可以在 Testing Page 找到.