Как написать модульный тест для конечной точки 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