def test_remove_user_token(self):
Token.objects.get_or_create(user=self.user)
self.assertEqual(Token.objects.filter(user=self.user).count(), 1)
users.remove_user_token(self.user)
self.assertEqual(Token.objects.filter(user=self.user).count(), 0)
def post(self, request, *args, **kwargs):
"""
Override post method to remove token and custom response
"""
# Remove user token
if drfr_settings.LOGOUT_REMOVE_TOKEN:
remove_user_token(self.request.user)
return Response(status=status.HTTP_204_NO_CONTENT)
def update(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
user = serializer.save()
# Reset old token
remove_user_token(user)
# Response data include new token
data = get_user_profile_data(user)
return Response(data, status=status.HTTP_200_OK)