Advertisement
Guest User

Untitled

a guest
Mar 3rd, 2016
442
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.24 KB | None | 0 0
  1. # creating and testing permissions and test groups in django tests.
  2. from django.contrib.auth.models import User, Permission, Group
  3. from django.test import TestCase
  4. from django.test import Client
  5.  
  6.  
  7. class ExampleGroupPermissionsTests(TestCase):
  8.  
  9. def setUp(self):
  10. #create permissions group
  11. group_name = "My Test Group"
  12. self.group = Group(name=group_name)
  13. self.group.save()
  14. self.c = Client()
  15. self.user = User.objects.create_user(username="test", email="test@test.com", password="test")
  16.  
  17. def tearDown(self):
  18. self.user.delete()
  19. self.group.delete()
  20.  
  21. def test_user_cannot_access(self):
  22. """user NOT in group should not have access
  23. """
  24. self.c.login(username='test', password='test')
  25. response = self.c.get("/my_view")
  26. self.assertEqual(response.status_code, 302, u'user in group should have access')
  27.  
  28.  
  29. def test_user_can_access(self):
  30. """user in group should have access
  31. """
  32. self.user.groups.add(self.group)
  33. self.user.save()
  34. self.c.login(username='test', password='test')
  35. response = self.c.get("/my_view")
  36. self.assertEqual(response.status_code, 200, u'user in group should have access')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement