Как добавить пользовательские заголовки в django тестовых случаях? - PullRequest
Новая
       22

Как добавить пользовательские заголовки в django тестовых случаях?

0 голосов
/ 23 марта 2020

Я добавил пользовательский класс аутентификации в django rest framework, который требует идентификатор клиента и секрет клиента при регистрации пользователя в заголовках. Я пишу тестовые случаи для регистрации пользователя следующим образом: - 

User = get_user_model()
client = Client()


class TestUserRegister(TestCase):
    def setUp(self):
        # pass
        self.test_users = {
            'test_user': {
                'email': 'testuser@gmail.com',
                'password': 'Test@1234',
                'username': 'test',
                'company': 'test',
                'provider': 'email'
            }
        }

        response = client.post(
            reverse('user_register'),
            headers={
                "CLIENTID": <client id>,
                "CLIENTSECRET": <client secret>
            },
            data={
                'email': self.test_users['test_user']['email'],
                'username': self.test_users['test_user']['username'],
                'password': self.test_users['test_user']['password'],
                'company': self.test_users['test_user']['company'],
                'provider': self.test_users['test_user']['provider'],
            },
            content_type='application/json',
        )
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

    def test_register(self):
        response = client.post(
            reverse('user_register'),
            headers={
                "CLIENTID": <client id>,
                "CLIENTSECRET": <client secret>
            },
            data={
                "first_name": "john",
                "last_name": "williams",
                "email": "john@gmail.com",
                "password": "John@1234",
                "username": "john",
                "company": "john and co.",
                "provider": "email",
            },
            content_type="application/json"
        )

        self.assertEqual(response.status_code, status.HTTP_201_CREATED)

Вот мой пользовательский класс аутентификации: - 

from oauth2_provider.models import Application
from rest_framework import authentication
from rest_framework.exceptions import APIException


class ClientAuthentication(authentication.BaseAuthentication):
    @staticmethod
    def get_client_credentials(request):
        try:
            client_id = request.headers.get('CLIENTID')
            client_secret = request.headers.get('CLIENTSECRET')
        except:
            raise APIException(detail="Missing Client Credentials", code=400)
        return {
            'client_id': client_id,
            'client_secret': client_secret
        }

    def authenticate(self, request):
        credentials = self.get_client_credentials(request)

        client_instance = Application.objects.filter(
            client_id=credentials['client_id'],
            client_secret=credentials['client_secret'],
        ).first()

        if not client_instance:
            raise APIException("Invalid client credentials", code=401)
        return client_instance, None

Но он все еще дает мне 500 внутренних ошибок сервера, все работает нормально, если я не добавляю пользовательскую аутентификацию в мое представление реестра. Как я могу это исправить?

1 Ответ

0 голосов
/ 23 марта 2020

Согласно документам способ передачи заголовков заключается в дополнительных аргументах ключевых слов 

client.post(url, ..., CLIENTID=<client id>, CLIENTSECRET=<client secret>)
...