fix tests

Author: hishamkaram

tests/test_integration.py

@@ -1,12 +1,14 @@
 from datetime import timedelta
 
+from django.contrib.auth import get_user_model
 from rest_framework_simplejwt.compat import reverse
 from rest_framework_simplejwt.settings import api_settings
-from rest_framework_simplejwt.state import User
 from rest_framework_simplejwt.tokens import AccessToken
 
 from .utils import APIViewTestCase, override_api_settings
 
+User = get_user_model()
+
 
 class TestTestView(APIViewTestCase):
     view_name = 'test_view'

tests/test_serializers.py

@@ -1,17 +1,16 @@
 from datetime import timedelta
 from unittest.mock import MagicMock, patch
 
+from django.contrib.auth import get_user_model
 from django.test import TestCase
 from rest_framework import exceptions as drf_exceptions
-
 from rest_framework_simplejwt.exceptions import TokenError
 from rest_framework_simplejwt.serializers import (
     TokenObtainPairSerializer, TokenObtainSerializer,
     TokenObtainSlidingSerializer, TokenRefreshSerializer,
     TokenRefreshSlidingSerializer, TokenVerifySerializer,
 )
 from rest_framework_simplejwt.settings import api_settings
-from rest_framework_simplejwt.state import User
 from rest_framework_simplejwt.token_blacklist.models import (
     BlacklistedToken, OutstandingToken,
 )
@@ -24,6 +23,8 @@
 
 from .utils import override_api_settings
 
+User = get_user_model()
+
 
 class TestTokenObtainSerializer(TestCase):
     def setUp(self):

tests/test_tokens.py

@@ -1,21 +1,23 @@
 from datetime import datetime, timedelta
 from unittest.mock import patch
 
+from django.contrib.auth import get_user_model
 from django.test import TestCase
-from jose import jwt
-
 from rest_framework_simplejwt.exceptions import TokenError
 from rest_framework_simplejwt.settings import api_settings
-from rest_framework_simplejwt.state import User
 from rest_framework_simplejwt.tokens import (
     AccessToken, RefreshToken, SlidingToken, Token, UntypedToken,
 )
 from rest_framework_simplejwt.utils import (
     aware_utcnow, datetime_to_epoch, make_utc,
 )
 
+from jose import jwt
+
 from .utils import override_api_settings
 
+User = get_user_model()
+
 
 class MyToken(Token):
     token_type = 'test'

tests/test_views.py

@@ -2,11 +2,10 @@
 from importlib import reload
 from unittest.mock import patch
 
+from django.contrib.auth import get_user_model
 from django.utils import timezone
-
 from rest_framework_simplejwt import serializers
 from rest_framework_simplejwt.settings import api_settings
-from rest_framework_simplejwt.state import User
 from rest_framework_simplejwt.tokens import (
     AccessToken, RefreshToken, SlidingToken,
 )
@@ -16,6 +15,8 @@
 
 from .utils import APIViewTestCase, override_api_settings
 
+User = get_user_model()
+
 
 class TestTokenObtainPairView(APIViewTestCase):
     view_name = 'token_obtain_pair'