Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-04-15 12:31:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improvements to sitemaps tests

Browse source
This commit is contained in:
Karl Hobley 2014-07-02 12:57:21 +01:00
parent f642e78f60
commit 3be8c0374c
1 changed files with 13 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
wagtail/contrib/wagtailsitemaps/tests.py
View file

@ -1,5 +1,6 @@
from django.test import TestCase
from django.core.urlresolvers import reverse
from django.core.cache import cache
from wagtail.wagtailcore.models import Page, Site
from wagtail.tests.models import SimplePage
@ -46,6 +47,9 @@ class TestSitemapGenerator(TestCase):
# Check that a URL has made it into the xml
self.assertIn('/hello-world/', xml)
# Make sure the unpublished page didn't make it into the xml
self.assertNotIn('/unpublished/', xml)
class TestSitemapView(TestCase):
def test_sitemap_view(self):
@ -56,11 +60,20 @@ class TestSitemapView(TestCase):
self.assertEqual(response['Content-Type'], 'text/xml; charset=utf-8')
def test_sitemap_view_cache(self):
cache_key = 'wagtail-sitemap:%d' % Site.objects.get(is_default_site=True).id
# Check that the key is not in the cache
self.assertFalse(cache.has_key(cache_key))
# Hit the view
first_response = self.client.get('/sitemap.xml')
self.assertEqual(first_response.status_code, 200)
self.assertTemplateUsed(first_response, 'wagtailsitemaps/sitemap.xml')
# Check that the key is in the cache
self.assertTrue(cache.has_key(cache_key))
# Hit the view again. Should come from the cache this time
second_response = self.client.get('/sitemap.xml')