Advertisement
Guest User

Untitled

a guest
Jun 25th, 2019
143
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.21 KB | None | 0 0
  1. class FormTest(TestCase):
  2. def setUp(self):
  3.  
  4. self.user = ProjectUser.objects.create(username='testUser',
  5. email='testUser@user.com')
  6. self.user.set_password('testPassword')
  7. self.user.save()
  8. self.client = Client()
  9. self.factory = RequestFactory()
  10.  
  11.  
  12. def sometest(self):
  13.  
  14. self.client.login(username='testUser', password='testPassword')
  15. user = auth.get_user(self.client)
  16. assert user.is_authenticated
  17.  
  18. new_object_2 = SampleModel.objects.create(unique_id='123456', name='sample')
  19.  
  20. request = self.factory.post(reverse('page1:favourite_post', kwargs={'fav_id': new_object_2.id}))
  21. response = favourite_post(request, new_object_2.id)
  22.  
  23. path('favourite_post/<int:fav_id>', views.favourite_post, name='favourite_post'),
  24.  
  25. def favourite_post(request, fav_id):
  26. post = get_object_or_404(Post, id=fav_id)
  27.  
  28. if request.method == 'POST':
  29. if post.favourite.filter(id=request.user.id).exists():
  30. post.favourite.remove(request.user)
  31.  
  32.  
  33. else:
  34. post.favourite.add(request.user)
  35.  
  36. return HttpResponseRedirect(reverse('page:some_page', args=(fav_id,)))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement