mirror of https://github.com/django/django.git
62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.test import TestCase
|
|
from django.test.utils import setup_test_template_loader, restore_template_loaders
|
|
|
|
from ..models import Author, Article, UrlArticle
|
|
|
|
|
|
class DefaultsTests(TestCase):
|
|
"""Test django views in django/views/defaults.py"""
|
|
fixtures = ['testdata.json']
|
|
non_existing_urls = ['/views/non_existing_url/', # this is in urls.py
|
|
'/views/other_non_existing_url/'] # this NOT in urls.py
|
|
|
|
def test_page_not_found(self):
|
|
"A 404 status is returned by the page_not_found view"
|
|
for url in self.non_existing_urls:
|
|
response = self.client.get(url)
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
def test_csrf_token_in_404(self):
|
|
"""
|
|
The 404 page should have the csrf_token available in the context
|
|
"""
|
|
# See ticket #14565
|
|
for url in self.non_existing_urls:
|
|
response = self.client.get(url)
|
|
csrf_token = response.context['csrf_token']
|
|
self.assertNotEqual(str(csrf_token), 'NOTPROVIDED')
|
|
self.assertNotEqual(str(csrf_token), '')
|
|
|
|
def test_server_error(self):
|
|
"The server_error view raises a 500 status"
|
|
response = self.client.get('/views/server_error/')
|
|
self.assertEqual(response.status_code, 500)
|
|
|
|
def test_custom_templates(self):
|
|
"""
|
|
Test that 404.html and 500.html templates are picked by their respective
|
|
handler.
|
|
"""
|
|
setup_test_template_loader(
|
|
{'404.html': 'This is a test template for a 404 error.',
|
|
'500.html': 'This is a test template for a 500 error.'}
|
|
)
|
|
try:
|
|
for code, url in ((404, '/views/non_existing_url/'), (500, '/views/server_error/')):
|
|
response = self.client.get(url)
|
|
self.assertContains(response, "test template for a %d error" % code,
|
|
status_code=code)
|
|
finally:
|
|
restore_template_loaders()
|
|
|
|
def test_get_absolute_url_attributes(self):
|
|
"A model can set attributes on the get_absolute_url method"
|
|
self.assertTrue(getattr(UrlArticle.get_absolute_url, 'purge', False),
|
|
'The attributes of the original get_absolute_url must be added.')
|
|
article = UrlArticle.objects.get(pk=1)
|
|
self.assertTrue(getattr(article.get_absolute_url, 'purge', False),
|
|
'The attributes of the original get_absolute_url must be added.')
|