old_contents: import json
  
  from django.test import TestCase
  from django.contrib.auth.models import User
  
  from rest_framework_jwt import utils
  from rest_framework_jwt.jwt import base64url_decode
  
  
  class UtilsTests(TestCase):
      def setUp(self):
          self.username = 'jpueblo'
          self.email = 'jpueblo@example.com'
          self.user = User.objects.create_user(self.username, self.email)
  
      def test_jwt_payload_handler(self):
          payload = utils.jwt_payload_handler(self.user)
  
          self.assertTrue(isinstance(payload, dict))
          self.assertEqual(payload['user_id'], self.user.id)
          self.assertEqual(payload['email'], self.email)
          self.assertEqual(payload['username'], self.username)
          self.assertTrue('exp' in payload)
  
      def test_jwt_encode(self):
          payload = utils.jwt_payload_handler(self.user)
          token = utils.jwt_encode_handler(payload)
  
          payload_data = base64url_decode(token.split('.')[1].encode('utf-8'))
          payload_from_token = json.loads(payload_data.decode('utf-8'))
  
          self.assertEqual(payload_from_token, payload)
  
      def test_jwt_decode(self):
          payload = utils.jwt_payload_handler(self.user)
          token = utils.jwt_encode_handler(payload)
          decoded_payload = utils.jwt_decode_handler(token)
  
          self.assertEqual(decoded_payload, payload)
  
new_contents: import json
  from jwt import base64url_decode
  
  from django.test import TestCase
  from django.contrib.auth.models import User
  from rest_framework_jwt import utils
  
  
  class UtilsTests(TestCase):
      def setUp(self):
          self.username = 'jpueblo'
          self.email = 'jpueblo@example.com'
          self.user = User.objects.create_user(self.username, self.email)
  
      def test_jwt_payload_handler(self):
          payload = utils.jwt_payload_handler(self.user)
  
          self.assertTrue(isinstance(payload, dict))
          self.assertEqual(payload['user_id'], self.user.id)
          self.assertEqual(payload['email'], self.email)
          self.assertEqual(payload['username'], self.username)
          self.assertTrue('exp' in payload)
  
      def test_jwt_encode(self):
          payload = utils.jwt_payload_handler(self.user)
          token = utils.jwt_encode_handler(payload)
  
          payload_data = base64url_decode(token.split('.')[1].encode('utf-8'))
          payload_from_token = json.loads(payload_data.decode('utf-8'))
  
          self.assertEqual(payload_from_token, payload)
  
      def test_jwt_decode(self):
          payload = utils.jwt_payload_handler(self.user)
          token = utils.jwt_encode_handler(payload)
          decoded_payload = utils.jwt_decode_handler(token)
  
          self.assertEqual(decoded_payload, payload)
  
subject: Fix missed jwt import