43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from django.apps import apps
|
|
from django.contrib.sites.models import Site
|
|
from django.test import TestCase
|
|
from django.test.utils import modify_settings, override_settings
|
|
|
|
|
|
@override_settings(
|
|
ROOT_URLCONF="flatpages_tests.urls",
|
|
SITE_ID=1,
|
|
)
|
|
@modify_settings(
|
|
INSTALLED_APPS={
|
|
"append": ["django.contrib.sitemaps", "django.contrib.flatpages"],
|
|
},
|
|
)
|
|
class FlatpagesSitemapTests(TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
# This cleanup is necessary because contrib.sites cache
|
|
# makes tests interfere with each other, see #11505
|
|
Site.objects.clear_cache()
|
|
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
Site = apps.get_model("sites.Site")
|
|
current_site = Site.objects.get_current()
|
|
current_site.flatpage_set.create(url="/foo/", title="foo")
|
|
current_site.flatpage_set.create(
|
|
url="/private-foo/", title="private foo", registration_required=True
|
|
)
|
|
|
|
def test_flatpage_sitemap(self):
|
|
response = self.client.get("/flatpages/sitemap.xml")
|
|
self.assertIn(
|
|
b"<url><loc>http://example.com/flatpage_root/foo/</loc></url>",
|
|
response.getvalue(),
|
|
)
|
|
self.assertNotIn(
|
|
b"<url><loc>http://example.com/flatpage_root/private-foo/</loc></url>",
|
|
response.getvalue(),
|
|
)
|