Django - Как добавить маркер доступа к Client.post в Django test?

Ниже приведен код. Каждая конечная точка имеет процесс аутентификации, который также приведен ниже. Я хочу иметь возможность прикрепить маркер доступа, который находится в cls.user, к Client.post, чтобы я мог протестировать все конечные точки и убедиться, что они также правильно аутентифицируются. Как я могу это сделать? В идеале я должен прикрепить <bearer> <access token> к request.Meta['HTTP_AUTHORIZATION']

test.py

import json
from cheers.models import *
from warrant import Cognito
from django.urls import reverse
from django.test import TestCase
from rest_framework import status
from cheers.models import GoalCategory, Post
from dummy_factory.Factories import UserFactory, GoalFactory


class PostTest(TestCase):
    @classmethod
    # Generates Test DB data to persist throughout all tests
    def setUpTestData(cls) -> None:
        cls.goal_category = 'health'
        GoalCategory.objects.create(category=cls.goal_category, emoji_url='url')
        cls.user = UserFactory()
        cls.goal = GoalFactory()
        user_obj = User.objects.get(pk=cls.user.phone_number)
        goal_obj = Goal.objects.get(pk=cls.goal.uuid)
        Post.objects.create(creator_id=user_obj, goal_id=goal_obj, body='Some text')
        cls.user = Cognito(<Some login credentials>)
        cls.user.authenticate(password=<password>)

    def test_create(self):
        response = self.client.post(reverse('post'),
                                    data=json.dumps({'creator_id': str(self.user.uuid),
                                                     'goal_id': str(self.goal.uuid),
                                                     'body': 'Some text #Test'}),
                                    content_type='application/json')
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

Тестирование функции аутентификатора

def cognito_authenticator(view_func):
    def wrapped_view(request, *args, **kwargs):
        # Check the cognito token from the request.
        token = request.META['HTTP_AUTHORIZATION'].split(' ')[1]

        try:
            jwt.decode_cognito_jwt(token)
        except Exception:
            # Fail if invalid
            return Response("Invalid JWT", status=status.HTTP_401_UNAUTHORIZED)  # Or HttpResponseForbidden()
        else:
            # Proceed with the view if valid
            return view_func(request, *args, **kwargs)

    return wrapped_view

Вы можете установить заголовок следующим образом:

token = 'sometoken'
response = self.client.post(
    reverse('post'),
    data=json.dumps({
        'creator_id': str(self.user.uuid),
        'goal_id': str(self.goal.uuid),
        'body': 'Some text #Test'
    }),
    content_type='application/json',
    **{'HTTP_AUTHORIZATION': f'Bearer {token}'}
)

И затем получить доступ к заголовку, используя:

request.META['HTTP_AUTHORIZATION']
Вернуться на верх