test: add test for rest_framework_simplejwt.token_blacklist support

Author: bartvanandel

tests/contrib/test_rest_auth.py

@@ -6,7 +6,7 @@
 from django.urls import include, path
 from rest_framework import viewsets
 
-from tests import assert_schema, generate_schema
+from tests import assert_schema, generate_schema, get_request_schema
 from tests.models import SimpleModel, SimpleSerializer
 
 try:
@@ -79,3 +79,35 @@ class XViewset(viewsets.ModelViewSet):
         'jwtCookieAuth': {'type': 'apiKey', 'in': 'cookie', 'name': 'jwt-session'},
         'jwtHeaderAuth': {'type': 'http', 'scheme': 'bearer', 'bearerFormat': 'JWT'}
     }
+
+
+@pytest.mark.contrib('dj_rest_auth', 'rest_framework_simplejwt')
+def test_rest_auth_token_blacklist(no_warnings, settings):
+    # flush module import cache to re-evaluate conditional import
+    import dj_rest_auth.urls
+    reload(dj_rest_auth.urls)
+
+    settings.INSTALLED_APPS += (
+        'rest_framework_simplejwt',
+        'rest_framework_simplejwt.token_blacklist',
+    )
+    settings.SIMPLE_JWT = {
+        'BLACKLIST_AFTER_ROTATION': True
+    }
+
+    urlpatterns = [
+        path('rest-auth/', include('dj_rest_auth.urls')),
+    ]
+    schema = generate_schema(None, patterns=urlpatterns)
+    assert get_request_schema(schema['paths']['/rest-auth/logout/']['post'])['$ref'] == '#/components/schemas/Logout'
+    assert schema['components']['schemas']['Logout'] == {
+        'type': 'object',
+        'properties': {
+            'refresh': {
+                'type': 'string',
+            },
+        },
+        'required': [
+            'refresh',
+        ],
+    }