/django/contrib/flatpages/tests/middleware.py
Python | 70 lines | 61 code | 9 blank | 0 comment | 1 complexity | d1e39fec5614f488d4336c3596cec3b5 MD5 | raw file
Possible License(s): BSD-3-Clause
- import os
- from django.conf import settings
- from django.contrib.auth.models import User
- from django.test import TestCase
- class FlatpageMiddlewareTests(TestCase):
- fixtures = ['sample_flatpages']
- urls = 'django.contrib.flatpages.tests.urls'
- def setUp(self):
- self.old_MIDDLEWARE_CLASSES = settings.MIDDLEWARE_CLASSES
- flatpage_middleware_class = 'django.contrib.flatpages.middleware.FlatpageFallbackMiddleware'
- if flatpage_middleware_class not in settings.MIDDLEWARE_CLASSES:
- settings.MIDDLEWARE_CLASSES += (flatpage_middleware_class,)
- self.old_TEMPLATE_DIRS = settings.TEMPLATE_DIRS
- settings.TEMPLATE_DIRS = (
- os.path.join(
- os.path.dirname(__file__),
- 'templates'
- ),
- )
- self.old_LOGIN_URL = settings.LOGIN_URL
- settings.LOGIN_URL = '/accounts/login/'
- def tearDown(self):
- settings.MIDDLEWARE_CLASSES = self.old_MIDDLEWARE_CLASSES
- settings.TEMPLATE_DIRS = self.old_TEMPLATE_DIRS
- settings.LOGIN_URL = self.old_LOGIN_URL
- def test_view_flatpage(self):
- "A flatpage can be served through a view, even when the middleware is in use"
- response = self.client.get('/flatpage_root/flatpage/')
- self.assertEqual(response.status_code, 200)
- self.assertContains(response, "<p>Isn't it flat!</p>")
- def test_view_non_existent_flatpage(self):
- "A non-existent flatpage raises 404 when served through a view, even when the middleware is in use"
- response = self.client.get('/flatpage_root/no_such_flatpage/')
- self.assertEqual(response.status_code, 404)
- def test_view_authenticated_flatpage(self):
- "A flatpage served through a view can require authentication"
- response = self.client.get('/flatpage_root/sekrit/')
- self.assertRedirects(response, '/accounts/login/?next=/flatpage_root/sekrit/')
- User.objects.create_user('testuser', 'test@example.com', 's3krit')
- self.client.login(username='testuser',password='s3krit')
- response = self.client.get('/flatpage_root/sekrit/')
- self.assertEqual(response.status_code, 200)
- self.assertContains(response, "<p>Isn't it sekrit!</p>")
- def test_fallback_flatpage(self):
- "A flatpage can be served by the fallback middlware"
- response = self.client.get('/flatpage/')
- self.assertEqual(response.status_code, 200)
- self.assertContains(response, "<p>Isn't it flat!</p>")
- def test_fallback_non_existent_flatpage(self):
- "A non-existent flatpage raises a 404 when served by the fallback middlware"
- response = self.client.get('/no_such_flatpage/')
- self.assertEqual(response.status_code, 404)
- def test_fallback_authenticated_flatpage(self):
- "A flatpage served by the middleware can require authentication"
- response = self.client.get('/sekrit/')
- self.assertRedirects(response, '/accounts/login/?next=/sekrit/')
- User.objects.create_user('testuser', 'test@example.com', 's3krit')
- self.client.login(username='testuser',password='s3krit')
- response = self.client.get('/sekrit/')
- self.assertEqual(response.status_code, 200)
- self.assertContains(response, "<p>Isn't it sekrit!</p>")