如何在 DRF 测试中使用 JWT 令牌？

Posted 2023-03-10

【中文标题】如何在 DRF 测试中使用 JWT 令牌？

【英文标题】：How can I use JWT token in DRF tests?

【发布时间】：2020-10-23 19:52:31

【问题描述】：

我需要测试我的 API。例如，我有图像列表页面。我需要对此页面进行测试，但如果没有身份验证，我无法做到这一点。我使用智威汤逊。我不知道怎么做。 请帮帮我。

tests.py

class ImagesListTestCase(APITestCase):
    def test_images_list(self):
        response = self.client.get('/api/', HTTP_AUTHORIZATION="JWT ".format("eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ0b2tlbl90eXBlIjoiYWNjZXNzIiwiZXhwIjoxNTkzNzUxMzMxLCJqdGkiOiI3NWE3NDNkMGU3MDQ0MGNiYjQ3NDExNjQ3MTI5NWVjNSIsInVzZXJfaWQiOjF9.7HkhZ1hRV8OtQJMMLEAVwUnJ0yDt8agFadAsJztFb6A"))
        self.assertEqual(response.status_code, status.HTTP_200_OK)

我试过了

response = self.client.get('/api/', Authorization="Bearer <token>")

还有

response = self.client.get('/api/', Authorization="JWT <token>")

response = self.client.get('/api/', HTTP_AUTHORIZATION="Bearer <token>")

【问题讨论】：

先获取JWT令牌。

那我该怎么办？

阅读您的 api 文档以了解附加令牌的方式，并阅读您的 http 客户端文档以了解如何执行此操作。

你有什么错误吗？

【参考方案1】：

django 创建一个临时数据库进行测试；所以最好像这样从username 和password 获取令牌：

class ImagesListTestCase(APITestCase):
    def setUp(self) :
    self.register_url = reverse("your:register:view")  # for example : "users:register"
    self.user_data = 
        "username": "test_user",
        "email": "[e-mail]",
        "password": "123456"
    
    self.client.post(self.register_url,self.user_data)      # user created

    auth_url = reverse("your:login:view") #for example :"users:token_obtain_pair"
    self.access_token = self.client.post(auth_url,
        "username" : self.user_data.get("username") ,
        "password" : self.user_data.get("password")
    ).data.get("access")       # get access_token for authorization
    self.client.credentials(HTTP_AUTHORIZATION=f'Bearer self.access_token')
    
    def test_images_list(self):
        response = self.client.get('/api/')
        self.assertEqual(response.status_code, status.HTTP_200_OK)

【参考方案2】：

您可以创建需要 JWT 身份验证的用户，例如：

    def setUp(self) -> None:
    self.client = APIClient()

    if os.environ.get('GITHUB_WORKFLOW'):
        local_token_url = 'https://testing/app/token/jwt'
    else:
        local_token_url = 'http://127.0.0.1:8000/app/token/jwt'

    response = requests.post(local_token_url, 'email': '[e-mail]', 'password': '123pass321')
    self.test_user1 = json.loads(response.text)
    response = requests.post(local_token_url, 'email': '[e-mail]', 'password': '123pass321')
    self.test_user2 = json.loads(response.text)

    self.contact_person = apm.AppUser.objects.create(email="[e-mail]", first_name="John",
                                                     last_name="Doe",
                                                     company_name="test_company_1", phone_number="08077745673",
                                                     is_active=True)

然后您可以像下面那样使用它们解析数据 =

self.client.credentials(HTTP_AUTHORIZATION="Bearer ".format(self.test_user1.get('access')))
    response = self.client.post(reverse('create-list'), data=data, format='json')
    print(response.data)
    self.assertEqual(response.status_code, status.HTTP_201_CREATE