34 lines
1.4 KiB
Python
34 lines
1.4 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from django.conf import settings
|
||
|
|
from django.test import override_settings
|
||
|
|
|
||
|
|
from .factories import GalleryFactory
|
||
|
|
from .helpers import PhotologueBaseTest
|
||
|
|
|
||
|
|
|
||
|
|
@unittest.skipUnless('django.contrib.sitemaps' in settings.INSTALLED_APPS,
|
||
|
|
'Sitemaps not installed in this project, nothing to test.')
|
||
|
|
@override_settings(ROOT_URLCONF='photologue.tests.test_urls')
|
||
|
|
class SitemapTest(PhotologueBaseTest):
|
||
|
|
|
||
|
|
def test_get_photo(self):
|
||
|
|
"""Default test setup contains one photo, this should appear in the sitemap."""
|
||
|
|
response = self.client.get('/sitemap.xml')
|
||
|
|
self.assertContains(response,
|
||
|
|
'<url><loc>http://example.com/ptests/photo/landscape/</loc>'
|
||
|
|
'<lastmod>2011-12-23</lastmod></url>')
|
||
|
|
|
||
|
|
def test_get_gallery(self):
|
||
|
|
"""if we add a gallery to the site, we should see both the gallery and
|
||
|
|
the photo in the sitemap."""
|
||
|
|
self.gallery = GalleryFactory(slug='test-gallery')
|
||
|
|
|
||
|
|
response = self.client.get('/sitemap.xml')
|
||
|
|
self.assertContains(response,
|
||
|
|
'<url><loc>http://example.com/ptests/photo/landscape/</loc>'
|
||
|
|
'<lastmod>2011-12-23</lastmod></url>')
|
||
|
|
self.assertContains(response,
|
||
|
|
'<url><loc>http://example.com/ptests/gallery/test-gallery/</loc>'
|
||
|
|
'<lastmod>2011-12-23</lastmod></url>')
|