from django.core.checks import Error

from django.core.checks.compatibility.django_4_0 import check_csrf_trusted_origins

from django.test import SimpleTestCase

from django.test.utils import override_settings

class CheckCSRFTrustedOrigins(SimpleTestCase):

    @override_settings(CSRF_TRUSTED_ORIGINS=["example.com"])

    def test_invalid_url(self):

        self.assertEqual(

            check_csrf_trusted_origins(None),

            [

                Error(

                    "As of Django 4.0, the values in the CSRF_TRUSTED_ORIGINS "

                    "setting must start with a scheme (usually http:// or "

                    "https://) but found example.com. See the release notes for "

                    "details.",

                    id="4_0.E001",

                )

            ],

        )

    @override_settings(

        CSRF_TRUSTED_ORIGINS=["http://example.com", "https://example.com"],

    )

    def test_valid_urls(self):

        self.assertEqual(check_csrf_trusted_origins(None), [])