from django.conf import settings

from django.contrib.flatpages.forms import FlatpageForm

from django.contrib.flatpages.models import FlatPage

from django.contrib.sites.models import Site

from django.test import TestCase, modify_settings, override_settings

from django.utils import translation

@modify_settings(INSTALLED_APPS={"append": ["django.contrib.flatpages"]})

@override_settings(SITE_ID=1)

class FlatpageAdminFormTests(TestCase):

    @classmethod

    def setUpTestData(cls):

        # don't use the manager because we want to ensure the site exists

        # with pk=1, regardless of whether or not it already exists.

        cls.site1 = Site(pk=1, domain="example.com", name="example.com")

        cls.site1.save()

    def setUp(self):

        # Site fields cache needs to be cleared after flatpages is added to

        # INSTALLED_APPS

        Site._meta._expire_cache()

        self.form_data = {

            "title": "A test page",

            "content": "This is a test",

            "sites": [settings.SITE_ID],

        }

    def test_flatpage_admin_form_url_validation(self):

        "The flatpage admin form correctly validates urls"

        self.assertTrue(

            FlatpageForm(data=dict(url="/new_flatpage/", **self.form_data)).is_valid()

        )

        self.assertTrue(

            FlatpageForm(

                data=dict(url="/some.special~chars/", **self.form_data)

            ).is_valid()

        )

        self.assertTrue(

            FlatpageForm(

                data=dict(url="/some.very_special~chars-here/", **self.form_data)

            ).is_valid()

        )

        self.assertFalse(

            FlatpageForm(data=dict(url="/a space/", **self.form_data)).is_valid()

        )

        self.assertFalse(

            FlatpageForm(data=dict(url="/a % char/", **self.form_data)).is_valid()

        )

        self.assertFalse(

            FlatpageForm(data=dict(url="/a ! char/", **self.form_data)).is_valid()

        )

        self.assertFalse(

            FlatpageForm(data=dict(url="/a & char/", **self.form_data)).is_valid()

        )

        self.assertFalse(

            FlatpageForm(data=dict(url="/a ? char/", **self.form_data)).is_valid()

        )

    def test_flatpage_requires_leading_slash(self):

        form = FlatpageForm(data=dict(url="no_leading_slash/", **self.form_data))

        with translation.override("en"):

            self.assertFalse(form.is_valid())

            self.assertEqual(form.errors["url"], ["URL is missing a leading slash."])

    @override_settings(

        APPEND_SLASH=True, MIDDLEWARE=["django.middleware.common.CommonMiddleware"]

    )

    def test_flatpage_requires_trailing_slash_with_append_slash(self):

        form = FlatpageForm(data=dict(url="/no_trailing_slash", **self.form_data))

        with translation.override("en"):

            self.assertEqual(

                form.fields["url"].help_text,

                "Example: “/about/contact/”. Make sure to have leading and "

                "trailing slashes.",

            )

            self.assertFalse(form.is_valid())

            self.assertEqual(form.errors["url"], ["URL is missing a trailing slash."])

    @override_settings(

        APPEND_SLASH=False, MIDDLEWARE=["django.middleware.common.CommonMiddleware"]

    )

    def test_flatpage_doesnt_requires_trailing_slash_without_append_slash(self):

        form = FlatpageForm(data=dict(url="/no_trailing_slash", **self.form_data))

        self.assertTrue(form.is_valid())

        with translation.override("en"):

            self.assertEqual(

                form.fields["url"].help_text,

                "Example: “/about/contact”. Make sure to have a leading slash.",

            )

    def test_flatpage_admin_form_url_uniqueness_validation(self):

        """

        The flatpage admin form correctly enforces url uniqueness among

        flatpages of the same site.

        """

        data = dict(url="/myflatpage1/", **self.form_data)

        FlatpageForm(data=data).save()

        f = FlatpageForm(data=data)

        with translation.override("en"):

            self.assertFalse(f.is_valid())

            self.assertEqual(

                f.errors,

                {

                    "__all__": [

                        "Flatpage with url /myflatpage1/ already exists for site "

                        "example.com"

                    ]

                },

            )

    def test_flatpage_admin_form_edit(self):

        """

        Existing flatpages can be edited in the admin form without triggering

        the url-uniqueness validation.

        """

        existing = FlatPage.objects.create(

            url="/myflatpage1/", title="Some page", content="The content"

        )

        existing.sites.add(settings.SITE_ID)

        data = dict(url="/myflatpage1/", **self.form_data)

        f = FlatpageForm(data=data, instance=existing)

        self.assertTrue(f.is_valid(), f.errors)

        updated = f.save()

        self.assertEqual(updated.title, "A test page")

    def test_flatpage_nosites(self):

        data = dict(url="/myflatpage1/", **self.form_data)

        data.update({"sites": ""})

        f = FlatpageForm(data=data)

        self.assertFalse(f.is_valid())

        self.assertEqual(

            f.errors, {"sites": [translation.gettext("This field is required.")]}

        )