test_generic.py 789 B

12345678910111213141516171819202122
  1. from __future__ import unicode_literals
  2. from django.test import override_settings
  3. from .base import TestModel, SitemapTestsBase
  4. @override_settings(ABSOLUTE_URL_OVERRIDES={})
  5. class GenericViewsSitemapTests(SitemapTestsBase):
  6. def test_generic_sitemap(self):
  7. "A minimal generic sitemap can be rendered"
  8. response = self.client.get('/generic/sitemap.xml')
  9. expected = ''
  10. for pk in TestModel.objects.values_list("id", flat=True):
  11. expected += "<url><loc>%s/testmodel/%s/</loc></url>" % (self.base_url, pk)
  12. expected_content = """<?xml version="1.0" encoding="UTF-8"?>
  13. <urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">
  14. %s
  15. </urlset>
  16. """ % expected
  17. self.assertXMLEqual(response.content.decode('utf-8'), expected_content)