Add additional viewset to verify simpleJWT extension

tfranzel · Apr 6, 2024 · c0a27e2 · c0a27e2
1 parent b228b38
commit c0a27e2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/contrib/test_simplejwt.py b/tests/contrib/test_simplejwt.py
@@ -8,7 +8,7 @@
 
 try:
     from rest_framework_simplejwt.authentication import (
-        JWTAuthentication, JWTTokenUserAuthentication,
+        JWTAuthentication, JWTTokenUserAuthentication, JWTStatelessUserAuthentication
     )
     from rest_framework_simplejwt.views import (
         TokenObtainPairView, TokenObtainSlidingView, TokenRefreshView, TokenVerifyView,
@@ -33,9 +33,14 @@ class X2Viewset(mixins.ListModelMixin, viewsets.GenericViewSet):
     authentication_classes = [JWTTokenUserAuthentication]
     required_scopes = ['x:read', 'x:write']
 
+class X3Viewset(mixins.ListModelMixin, viewsets.GenericViewSet):
+    serializer_class = XSerializer
+    authentication_classes = [JWTStatelessUserAuthentication]
+    required_scopes = ['x:read', 'x:write']
+
 
 @pytest.mark.contrib('rest_framework_simplejwt')
-@pytest.mark.parametrize('view', [XViewset, X2Viewset])
+@pytest.mark.parametrize('view', [XViewset, X2Viewset, X3Viewset])
 def test_simplejwt(no_warnings, view):
     router = routers.SimpleRouter()
     router.register('x', view, basename="x")