Как написать модульный тест для конечной точки Django REST API, которая имеет permission_classes = (IsAuthenticated,).

Привет всем Я хочу написать модульные тесты для моего приложения django, чтобы проверить различные конечные точки API, но кажется, что я не могу разобраться в проблеме Вот примерный код того, что я сделал до сих пор :

urls.py :

path('translate/display/', DisplayTranslation.as_view(), name='display_translation'),

это соответствующий DRF view.py :

class DisplayTranslation(generics.ListAPIView):
    queryset = Translation.objects.all()
    serializer_class = TranslationSerializers
    permission_classes = (IsAuthenticated,)

и вот что я сделал на данный момент в моем модуле test.py :

apiclient = APIClient()

class TranslationTestCases(APITestCase):
    def setUp(self):
        self.role = baker.make(Roles)
        self.user = baker.make(Users, user_role=self.role)
        self.token = reverse('token_obtain_pair', kwargs={'email': self.user.email, 'password': self.user.password})
        self.translation = baker.make(Translation, _quantity=10)
    
    def test_get_all_translations(self):
        header = {'HTTP_AUTHORIZATION': 'Token {}'.format(self.token)} 
        response = self.client.get(reverse('display_translation'), {}, **header)
        self.assertEqual(response.status_code, 200)
        self.assertEqual(len(response.data), 10)

Вот ошибка, которую я получаю при запуске теста : in _reverse_with_prefix raise NoReverseMatch(msg) django.urls.exceptions.NoReverseMatch: Reverse for 'token_obtain_pair' with keyword arguments '{'email': 'dZbQNWCjif@example.com', 'password': 'PfQzPqqVVLAdLZtJyVUxVjkGJEgRABYdHxMRhCGZJWxnZxpxEgUkgUKklENrWlBiYOCxhaDtJXdtXCtNdOJYtSWTzIrdvPnrmezXBNjEYYTpyZWjOLMnMIBMAfYnLwcC'}' not found. 1 pattern(s) tried: ['token/$']

Дополнительная информация, в аутентификации на моем Django приложении, я работал с DRF, rest_auth и библиотекой SimpleJWT.

Что я могу сделать, чтобы улучшить свой код? или альтернативное решение? Я не смог найти проблему, похожую на мою.

В принципе, вы можете использовать

@patch.object(DisplayTranslation, "permission_classes", [])
def test_get_all_translations(self):
    ...

https://docs.python.org/3/library/unittest.mock.html#patch-object

С аутентификацией:

urlpatterns = [
    ...
    path('api/token/', TokenObtainPairView.as_view(), name='token_obtain_pair'),
    path('api/token/refresh/', TokenRefreshView.as_view(), name='token_refresh'),
    ...
]

class TranslationTestCases(APITestCase):
    def setUp(self):
        self.api_client = APIClient()
        self.role = baker.make(Roles)
        self.user = baker.make(Users, user_role=self.role)
        self.token_url = reverse('token_obtain_pair')
        self.translation = baker.make(Translation, _quantity=10)
        response = self.api_client.post(self.token_url, {"username": self.user.username, "password": self.user.password})
        self.token = response.json()["access"]
    
    def test_get_all_translations(self):
        header = {'HTTP_AUTHORIZATION': 'Bearer {}'.format(self.token)} 
        ...

https://django-rest-framework-simplejwt.readthedocs.io/en/latest/getting_started.html

Вернуться на верх