如何使用 Django-oauth-toolkit 进行身份验证，通过 Django-rest-framework 测试 API 端点

我有一个 Django-rest-framework 视图集/路由器来定义 API 端点。视图集定义如下：

class DocumentViewSet(viewsets.ModelViewSet):
    permission_classes = [permissions.IsAuthenticated, TokenHasReadWriteScope]
    model = Document

路由器定义为

router = DefaultRouter()
router.register(r'documents', viewsets.DocumentViewSet)

带有 url 模式url(r'^api/', include(router.urls))

通过获取正确的访问令牌并使用它进行授权，我可以在浏览器中/通过curl 很好地访问此端点。但是，尚不清楚如何针对此端点编写测试。

这是我尝试过的：

class DocumentAPITests(APITestCase):
    def test_get_all_documents(self):
        user = User.objects.create_user('test', '[email protected] /cdn-cgi/l/email-protection', 'test')
        client = APIClient()
        client.credentials(username="test", password="test")
        response = client.get("/api/documents/")
        self.assertEqual(response.status_code, 200) 

此操作失败，并显示来自以下地址的 HTTP 401 响应：client.get()称呼。使用 django-oauth-toolkit 进行 oauth2 身份验证来测试 DRF 中的 API 端点的正确方法是什么？

当您编写测试时，您应该致力于从测试本身中提取未测试的任何内容，通常将任何设置代码放在setUp测试方法。对于使用 OAuth 的 API 测试，这通常包括测试用户、OAuth 应用程序和活动访问令牌。

For django-oauth-toolkit以及其他 Django 应用程序，我总是推荐查看测试，看看他们是如何做到的 https://github.com/evonove/django-oauth-toolkit/blob/dea3d7206148ad89a624301ff97280b94bf209ce/oauth2_provider/tests/test_token_revocation.py。这使您可以避免进行不必要的 API 调用，尤其是对于 OAuth 等多部分流程，并且仅创建所需的少数模型对象。

def setUp(self):
    self.test_user = UserModel.objects.create_user("test_user", "[email protected] /cdn-cgi/l/email-protection", "123456")

    self.application = Application(
        name="Test Application",
        redirect_uris="http://localhost",
        user=self.test_user,
        client_type=Application.CLIENT_CONFIDENTIAL,
        authorization_grant_type=Application.GRANT_AUTHORIZATION_CODE,
    )
    self.application.save()

def test_revoke_access_token(self):
    from datetime import datetime
    from django.utils import timezone

    tok = AccessToken.objects.create(
        user=self.test_user, token='1234567890',
        application=self.application, scope='read write',
        expires=timezone.now() + datetime.timedelta(days=1)
    )

从那里您只需使用已生成的令牌进行身份验证。你可以通过以下方式做到这一点注入Authorization header https://github.com/jpadilla/django-rest-framework-oauth/blob/f26dcc30b499198318e9049401bb8ce126f77855/tests/test_authentication.py#L334-L344，或者你可以use the force_authenticate method http://www.django-rest-framework.org/api-guide/testing/#forcing-authentication由 Django REST 框架提供。