diff options
Diffstat (limited to 'tests/special_headers/tests.py')
-rw-r--r-- | tests/special_headers/tests.py | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/tests/special_headers/tests.py b/tests/special_headers/tests.py new file mode 100644 index 0000000000..b890adbbad --- /dev/null +++ b/tests/special_headers/tests.py @@ -0,0 +1,62 @@ +from django.contrib.auth.models import User +from django.test import TestCase +from django.test.utils import override_settings + + +@override_settings(PASSWORD_HASHERS=('django.contrib.auth.hashers.SHA1PasswordHasher',)) +class SpecialHeadersTest(TestCase): + fixtures = ['data.xml'] + urls = 'regressiontests.special_headers.urls' + + def test_xheaders(self): + user = User.objects.get(username='super') + response = self.client.get('/special_headers/article/1/') + self.assertFalse('X-Object-Type' in response) + self.client.login(username='super', password='secret') + response = self.client.get('/special_headers/article/1/') + self.assertTrue('X-Object-Type' in response) + user.is_staff = False + user.save() + response = self.client.get('/special_headers/article/1/') + self.assertFalse('X-Object-Type' in response) + user.is_staff = True + user.is_active = False + user.save() + response = self.client.get('/special_headers/article/1/') + self.assertFalse('X-Object-Type' in response) + + def test_xview_func(self): + user = User.objects.get(username='super') + response = self.client.head('/special_headers/xview/func/') + self.assertFalse('X-View' in response) + self.client.login(username='super', password='secret') + response = self.client.head('/special_headers/xview/func/') + self.assertTrue('X-View' in response) + self.assertEqual(response['X-View'], 'regressiontests.special_headers.views.xview') + user.is_staff = False + user.save() + response = self.client.head('/special_headers/xview/func/') + self.assertFalse('X-View' in response) + user.is_staff = True + user.is_active = False + user.save() + response = self.client.head('/special_headers/xview/func/') + self.assertFalse('X-View' in response) + + def test_xview_class(self): + user = User.objects.get(username='super') + response = self.client.head('/special_headers/xview/class/') + self.assertFalse('X-View' in response) + self.client.login(username='super', password='secret') + response = self.client.head('/special_headers/xview/class/') + self.assertTrue('X-View' in response) + self.assertEqual(response['X-View'], 'regressiontests.special_headers.views.XViewClass') + user.is_staff = False + user.save() + response = self.client.head('/special_headers/xview/class/') + self.assertFalse('X-View' in response) + user.is_staff = True + user.is_active = False + user.save() + response = self.client.head('/special_headers/xview/class/') + self.assertFalse('X-View' in response) |