Coverage for api/serializers.py: 60%

1from rest_framework_simplejwt.serializers import TokenObtainPairSerializer 

2 

3from apps.users.serializers import UserPolymorphicSerializer 

4 

5 

6class UserToken(TokenObtainPairSerializer): 

7 def validate(self, attrs): 

8 data = super().validate(attrs) 

9 user_serializer = UserPolymorphicSerializer(self.user, context={"request": self.context.get("request")}) 

10 

11 data["user"] = user_serializer.data 

12 

13 return data