0
0
mirror of https://github.com/django/django.git synced 2024-11-22 03:18:31 +01:00
django/tests/sitemaps_tests/base.py

32 lines
1.1 KiB
Python

from django.apps import apps
from django.contrib.sites.models import Site
from django.core.cache import cache
from django.test import TestCase, modify_settings, override_settings
from .models import I18nTestModel, TestModel
@modify_settings(INSTALLED_APPS={'append': 'django.contrib.sitemaps'})
@override_settings(ROOT_URLCONF='sitemaps_tests.urls.http')
class SitemapTestsBase(TestCase):
protocol = 'http'
sites_installed = apps.is_installed('django.contrib.sites')
domain = 'example.com' if sites_installed else 'testserver'
@classmethod
def setUpTestData(cls):
# Create an object for sitemap content.
TestModel.objects.create(name='Test Object')
cls.i18n_model = I18nTestModel.objects.create(name='Test Object')
def setUp(self):
self.base_url = '%s://%s' % (self.protocol, self.domain)
cache.clear()
@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()