Misago/misago/core/tests/test_context_processors.py

from django.test import TestCase
from django.utils import translation

from misago.core import context_processors


class MockRequest(object):
    path = "/"

    def __init__(self, secure, host):
        self.secure = secure
        self.host = host

    def is_secure(self):
        return self.secure

    def get_host(self):
        return self.host


class MetaMockRequest(object):
    def __init__(self, meta):
        self.META = meta


class MomentjsLocaleTests(TestCase):
    def test_momentjs_locale(self):
        """momentjs_locale adds MOMENTJS_LOCALE_URL to context"""
        with translation.override("no-no"):
            self.assertEqual(
                context_processors.momentjs_locale(True), {"MOMENTJS_LOCALE_URL": None}
            )

        with translation.override("en-us"):
            self.assertEqual(
                context_processors.momentjs_locale(True), {"MOMENTJS_LOCALE_URL": None}
            )

        with translation.override("de"):
            self.assertEqual(
                context_processors.momentjs_locale(True),
                {"MOMENTJS_LOCALE_URL": "misago/momentjs/de.js"},
            )

        with translation.override("pl-de"):
            self.assertEqual(
                context_processors.momentjs_locale(True),
                {"MOMENTJS_LOCALE_URL": "misago/momentjs/pl.js"},
            )


class SiteAddressTests(TestCase):
    def test_site_address_for_http(self):
        """Correct SITE_ADDRESS set for HTTP request"""
        mock_request = MockRequest(False, "somewhere.com")
        self.assertEqual(
            context_processors.site_address(mock_request),
            {
                "REQUEST_PATH": "/",
                "SITE_ADDRESS": "http://somewhere.com",
                "SITE_HOST": "somewhere.com",
                "SITE_PROTOCOL": "http",
            },
        )

    def test_site_address_for_https(self):
        """Correct SITE_ADDRESS set for HTTPS request"""
        mock_request = MockRequest(True, "somewhere.com")
        self.assertEqual(
            context_processors.site_address(mock_request),
            {
                "REQUEST_PATH": "/",
                "SITE_ADDRESS": "https://somewhere.com",
                "SITE_HOST": "somewhere.com",
                "SITE_PROTOCOL": "https",
            },
        )


class FrontendContextTests(TestCase):
    def test_frontend_context(self):
        """frontend_context is available in templates"""
        mock_request = MockRequest(False, "somewhere.com")
        mock_request.include_frontend_context = True
        mock_request.frontend_context = {"someValue": "Something"}

        self.assertEqual(
            context_processors.frontend_context(mock_request),
            {"frontend_context": {"someValue": "Something"}},
        )

        mock_request.include_frontend_context = False
        self.assertEqual(context_processors.frontend_context(mock_request), {})