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)