Move token backend selection into separate method (#388)

rest_framework_simplejwt/tokens.py

@@ -35,7 +35,7 @@ def __init__(self, token=None, verify=True):
         # Set up token
         if token is not None:
             # An encoded token was provided
-            from .state import token_backend
+            token_backend = self.get_token_backend()
 
             # Decode token
             try:
@@ -77,9 +77,7 @@ def __str__(self):
         """
         Signs and returns a token as a base64 encoded string.
         """
-        from .state import token_backend
-
-        return token_backend.encode(self.payload)
+        return self.get_token_backend().encode(self.payload)
 
     def verify(self):
         """
@@ -167,6 +165,10 @@ def for_user(cls, user):
 
         return token
 
+    def get_token_backend(self):
+        from .state import token_backend
+        return token_backend
+
 
 class BlacklistMixin:
     """

tests/test_tokens.py

@@ -7,6 +7,7 @@
 
 from rest_framework_simplejwt.exceptions import TokenError
 from rest_framework_simplejwt.settings import api_settings
+from rest_framework_simplejwt.state import token_backend
 from rest_framework_simplejwt.tokens import (
     AccessToken, RefreshToken, SlidingToken, Token, UntypedToken,
 )
@@ -301,6 +302,11 @@ def test_for_user(self):
 
         self.assertEqual(token[api_settings.USER_ID_CLAIM], username)
 
+    def test_get_token_backend(self):
+        token = MyToken()
+
+        self.assertEqual(token.get_token_backend(), token_backend)
+
 
 class TestSlidingToken(TestCase):
     def test_init(self):