codekingpro/portable-devtools

1import datetime
2import os
3import shutil
4import tempfile
5import unittest
6
7import tornado.locale
8from tornado.escape import utf8, to_unicode
9from tornado.util import unicode_type
10
11
12class TranslationLoaderTest(unittest.TestCase):
13    # TODO: less hacky way to get isolated tests
14    SAVE_VARS = ["_translations", "_supported_locales", "_use_gettext"]
15
16    def clear_locale_cache(self):
17        tornado.locale.Locale._cache = {}
18
19    def setUp(self):
20        self.saved = {}  # type: dict
21        for var in TranslationLoaderTest.SAVE_VARS:
22            self.saved[var] = getattr(tornado.locale, var)
23        self.clear_locale_cache()
24
25    def tearDown(self):
26        for k, v in self.saved.items():
27            setattr(tornado.locale, k, v)
28        self.clear_locale_cache()
29
30    def test_csv(self):
31        tornado.locale.load_translations(
32            os.path.join(os.path.dirname(__file__), "csv_translations")
33        )
34        locale = tornado.locale.get("fr_FR")
35        self.assertTrue(isinstance(locale, tornado.locale.CSVLocale))
36        self.assertEqual(locale.translate("school"), "\u00e9cole")
37
38    def test_csv_bom(self):
39        with open(
40            os.path.join(os.path.dirname(__file__), "csv_translations", "fr_FR.csv"),
41            "rb",
42        ) as f:
43            char_data = to_unicode(f.read())
44        # Re-encode our input data (which is utf-8 without BOM) in
45        # encodings that use the BOM and ensure that we can still load
46        # it. Note that utf-16-le and utf-16-be do not write a BOM,
47        # so we only test whichver variant is native to our platform.
48        for encoding in ["utf-8-sig", "utf-16"]:
49            tmpdir = tempfile.mkdtemp()
50            try:
51                with open(os.path.join(tmpdir, "fr_FR.csv"), "wb") as f:
52                    f.write(char_data.encode(encoding))
53                tornado.locale.load_translations(tmpdir)
54                locale = tornado.locale.get("fr_FR")
55                self.assertIsInstance(locale, tornado.locale.CSVLocale)
56                self.assertEqual(locale.translate("school"), "\u00e9cole")
57            finally:
58                shutil.rmtree(tmpdir)
59
60    def test_gettext(self):
61        tornado.locale.load_gettext_translations(
62            os.path.join(os.path.dirname(__file__), "gettext_translations"),
63            "tornado_test",
64        )
65        locale = tornado.locale.get("fr_FR")
66        self.assertTrue(isinstance(locale, tornado.locale.GettextLocale))
67        self.assertEqual(locale.translate("school"), "\u00e9cole")
68        self.assertEqual(locale.pgettext("law", "right"), "le droit")
69        self.assertEqual(locale.pgettext("good", "right"), "le bien")
70        self.assertEqual(locale.pgettext("organization", "club", "clubs", 1), "le club")
71        self.assertEqual(
72            locale.pgettext("organization", "club", "clubs", 2), "les clubs"
73        )
74        self.assertEqual(locale.pgettext("stick", "club", "clubs", 1), "le b\xe2ton")
75        self.assertEqual(locale.pgettext("stick", "club", "clubs", 2), "les b\xe2tons")
76
77
78class LocaleDataTest(unittest.TestCase):
79    def test_non_ascii_name(self):
80        name = tornado.locale.LOCALE_NAMES["es_LA"]["name"]
81        self.assertTrue(isinstance(name, unicode_type))
82        self.assertEqual(name, "Espa\u00f1ol")
83        self.assertEqual(utf8(name), b"Espa\xc3\xb1ol")
84
85
86class EnglishTest(unittest.TestCase):
87    def test_format_date(self):
88        locale = tornado.locale.get("en_US")
89        date = datetime.datetime(2013, 4, 28, 18, 35)
90        self.assertEqual(
91            locale.format_date(date, full_format=True), "April 28, 2013 at 6:35 pm"
92        )
93
94        aware_dt = datetime.datetime.now(datetime.timezone.utc)
95        naive_dt = aware_dt.replace(tzinfo=None)
96        for name, now in {"aware": aware_dt, "naive": naive_dt}.items():
97            with self.subTest(dt=name):
98                self.assertEqual(
99                    locale.format_date(
100                        now - datetime.timedelta(seconds=2), full_format=False
101                    ),
102                    "2 seconds ago",
103                )
104                self.assertEqual(
105                    locale.format_date(
106                        now - datetime.timedelta(minutes=2), full_format=False
107                    ),
108                    "2 minutes ago",
109                )
110                self.assertEqual(
111                    locale.format_date(
112                        now - datetime.timedelta(hours=2), full_format=False
113                    ),
114                    "2 hours ago",
115                )
116
117                self.assertEqual(
118                    locale.format_date(
119                        now - datetime.timedelta(days=1),
120                        full_format=False,
121                        shorter=True,
122                    ),
123                    "yesterday",
124                )
125
126                date = now - datetime.timedelta(days=2)
127                self.assertEqual(
128                    locale.format_date(date, full_format=False, shorter=True),
129                    locale._weekdays[date.weekday()],
130                )
131
132                date = now - datetime.timedelta(days=300)
133                self.assertEqual(
134                    locale.format_date(date, full_format=False, shorter=True),
135                    "%s %d" % (locale._months[date.month - 1], date.day),
136                )
137
138                date = now - datetime.timedelta(days=500)
139                self.assertEqual(
140                    locale.format_date(date, full_format=False, shorter=True),
141                    "%s %d, %d" % (locale._months[date.month - 1], date.day, date.year),
142                )
143
144    def test_friendly_number(self):
145        locale = tornado.locale.get("en_US")
146        self.assertEqual(locale.friendly_number(1000000), "1,000,000")
147
148    def test_list(self):
149        locale = tornado.locale.get("en_US")
150        self.assertEqual(locale.list([]), "")
151        self.assertEqual(locale.list(["A"]), "A")
152        self.assertEqual(locale.list(["A", "B"]), "A and B")
153        self.assertEqual(locale.list(["A", "B", "C"]), "A, B and C")
154
155    def test_format_day(self):
156        locale = tornado.locale.get("en_US")
157        date = datetime.datetime(2013, 4, 28, 18, 35)
158        self.assertEqual(locale.format_day(date=date, dow=True), "Sunday, April 28")
159        self.assertEqual(locale.format_day(date=date, dow=False), "April 28")
160