zulip/zerver/tests/test_i18n.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import

from typing import Any

import mock
from django.test import TestCase
from django.conf import settings
from django.http import HttpResponse
from six.moves.http_cookies import SimpleCookie

from zerver.lib.test_helpers import AuthedTestCase
from zerver.management.commands import makemessages


class TranslationTestCase(AuthedTestCase):
    """
    Tranlations strings should change with locale. URLs should be locale
    aware.
    """

    # e.g. self.client.post(url) if method is "post"
    def fetch(self, method, url, expected_status, **kwargs):
        # type: (str, str, int, **Any) -> HttpResponse
        response = getattr(self.client, method)(url, **kwargs)
        self.assertEqual(response.status_code, expected_status,
                         msg="Expected %d, received %d for %s to %s" % (
                expected_status, response.status_code, method, url))
        return response

    def test_accept_language_header(self):
        # type: () -> None
        languages = [('en', u'Register'),
                     ('de', u'Registrieren'),
                     ('sr', u'Региструј се'),
                     ('zh-cn', u'注册'),
                     ]

        for lang, word in languages:
            response = self.fetch('get', '/integrations/', 200,
                                  HTTP_ACCEPT_LANGUAGE=lang)
            self.assert_in_response(word, response)

    def test_cookie(self):
        # type: () -> None
        languages = [('en', u'Register'),
                     ('de', u'Registrieren'),
                     ('sr', u'Региструј се'),
                     ('zh-cn', u'注册'),
                     ]

        for lang, word in languages:
            self.client.cookies = SimpleCookie({settings.LANGUAGE_COOKIE_NAME: lang}) # type: ignore # SimpleCookie has incomplete stubs in python 3

            response = self.fetch('get', '/integrations/', 200)
            self.assert_in_response(word, response)

    def test_i18n_urls(self):
        # type: () -> None
        languages = [('en', u'Register'),
                     ('de', u'Registrieren'),
                     ('sr', u'Региструј се'),
                     ('zh-cn', u'注册'),
                     ]

        for lang, word in languages:
            response = self.fetch('get', '/{}/integrations/'.format(lang), 200)
            self.assert_in_response(word, response)


class JsonTranslationTestCase(AuthedTestCase):
    @mock.patch('zerver.lib.request._')
    def test_json_error(self, mock_gettext):
        dummy_value = "Some other language '%s'"
        mock_gettext.return_value = dummy_value

        self.login("hamlet@zulip.com")
        result = self.client.post("/json/refer_friend",
                                  HTTP_ACCEPT_LANGUAGE='de')

        self.assert_json_error_contains(result,
                                        dummy_value % 'email',
                                        status_code=400)

    @mock.patch('zerver.views._')
    def test_jsonable_error(self, mock_gettext):
        dummy_value = "Some other language"
        mock_gettext.return_value = dummy_value

        self.login("hamlet@zulip.com")
        result = self.client.get("/de/accounts/login/jwt/")

        self.assert_json_error_contains(result,
                                        dummy_value,
                                        status_code=400)


class FrontendRegexTestCase(TestCase):
    def test_regexes(self):
        command = makemessages.Command()

        data = [
            ('{{#tr context}}english text with __variable__{{/tr}}{{/tr}}',
             'english text with __variable__'),

            ('{{t "english text" }}, "extra"}}',
             'english text'),

            ("{{t 'english text' }}, 'extra'}}",
             'english text'),

            ('i18n.t("english text"), "extra",)',
             'english text'),

            ('i18n.t("english text", context), "extra",)',
             'english text'),

            ("i18n.t('english text'), 'extra',)",
             'english text'),

            ("i18n.t('english text', context), 'extra',)",
             'english text'),
        ]

        for input_text, expected in data:
            result = list(command.extract_strings(input_text).keys())
            self.assertEqual(len(result), 1)
            self.assertEqual(result[0], expected)