kopia lustrzana https://github.com/wagtail/wagtail
29 wiersze
1.2 KiB
Python
29 wiersze
1.2 KiB
Python
from django.test import TestCase
|
|
|
|
from wagtail.contrib.settings.forms import SiteSwitchForm
|
|
from wagtail.core.models import Page, Site
|
|
from wagtail.test.testapp.models import TestSetting
|
|
|
|
|
|
class TestSiteSwitchFromSiteOrdering(TestCase):
|
|
def setUp(self):
|
|
self.root_page = Page.objects.get(pk=2)
|
|
Site.objects.all().delete() # Drop the initial site.
|
|
|
|
def test_site_order_by_hostname(self):
|
|
site_1 = Site.objects.create(hostname="charly.com", root_page=self.root_page)
|
|
site_2 = Site.objects.create(
|
|
hostname="bravo.com", root_page=self.root_page, is_default_site=True
|
|
)
|
|
site_3 = Site.objects.create(hostname="alfa.com", root_page=self.root_page)
|
|
form = SiteSwitchForm(site_1, TestSetting)
|
|
expected_choices = [
|
|
("/admin/settings/tests/testsetting/{}/".format(site_3.id), "alfa.com"),
|
|
(
|
|
"/admin/settings/tests/testsetting/{}/".format(site_2.id),
|
|
"bravo.com [default]",
|
|
),
|
|
("/admin/settings/tests/testsetting/{}/".format(site_1.id), "charly.com"),
|
|
]
|
|
self.assertEqual(form.fields["site"].choices, expected_choices)
|