Advertisement
Guest User

Untitled

a guest
Jul 19th, 2017
550
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.34 KB | None | 0 0
  1. class UserDetail(RetrieveModelMixin, DestroyModelMixin, UpdateModelMixin, GenericViewSet):
  2. authentication_classes = (BasicAuthentication, TokenAuthentication)
  3. permission_classes = IsAuthenticated,
  4. queryset = CustomUser.objects.all()
  5. serializer_class = UserSerializer
  6.  
  7. class AccountTests(APITestCase):
  8.  
  9. def setUp(self):
  10. self.user = CustomUser.objects.create_user(email="user1@test.com", password="password1", is_staff=True)
  11. self.user.save()
  12. self.user = CustomUser.objects.get(email="user1@test.com")
  13. self.client = APIClient()
  14.  
  15. def test_add_name(self):
  16. self.client.login(email="user1@test.com", password='password1')
  17. url = reverse('customuser-detail', args=(self.user.id,))
  18. data = {'first_name': 'test', 'last_name': 'user'}
  19.  
  20. self.client.login(email="user1@test.com", password='password1')
  21. response = self.client.put(url, data, format='json')
  22.  
  23. self.assertEqual(response.status_code, status.HTTP_200_OK)
  24.  
  25. {u'detail': u'Authentication credentials were not provided.'}
  26.  
  27. self.user = CustomUser.objects.create_user(email="user1@test.com", is_staff=True)
  28. self.user.set_password("password1")
  29. self.user.save()
  30.  
  31. self.client.force_login(
  32. user=User.objects.first(),
  33. backend='django.contrib.auth.backends.ModelBackend' # one of your AUTHENTICATION_BACKENDS
  34. )
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement