1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
#!/usr/bin/env python
import unittest
from unittest import mock
from mkdocs.config.base import ValidationError
from mkdocs.localization import install_translations, parse_locale
from mkdocs.tests.base import tempdir
class LocalizationTests(unittest.TestCase):
def setUp(self):
self.env = mock.Mock()
def test_jinja_extension_installed(self):
install_translations(self.env, parse_locale('en'), [])
self.env.add_extension.assert_called_once_with('jinja2.ext.i18n')
def test_valid_language(self):
locale = parse_locale('en')
self.assertEqual(locale.language, 'en')
def test_valid_language_territory(self):
locale = parse_locale('en_US')
self.assertEqual(locale.language, 'en')
self.assertEqual(locale.territory, 'US')
self.assertEqual(str(locale), 'en_US')
def test_unknown_locale(self):
self.assertRaises(ValidationError, parse_locale, 'foo')
def test_invalid_locale(self):
self.assertRaises(ValidationError, parse_locale, '42')
@tempdir()
def test_no_translations_found(self, dir_without_translations):
with self.assertLogs('mkdocs') as cm:
install_translations(self.env, parse_locale('fr_CA'), [dir_without_translations])
self.assertEqual(
'\n'.join(cm.output),
"WARNING:mkdocs.localization:No translations could be found for the locale 'fr_CA'. "
"Defaulting to English.",
)
self.env.install_null_translations.assert_called_once()
@tempdir()
def test_translations_found(self, tdir):
translations = mock.Mock()
with mock.patch('mkdocs.localization.Translations.load', return_value=translations):
install_translations(self.env, parse_locale('en'), [tdir])
self.env.install_gettext_translations.assert_called_once_with(translations)
@tempdir()
@tempdir()
def test_merge_translations(self, custom_dir, theme_dir):
custom_dir_translations = mock.Mock()
theme_dir_translations = mock.Mock()
def side_effet(*args, **kwargs):
dirname = args[0]
if dirname.startswith(custom_dir):
return custom_dir_translations
elif dirname.startswith(theme_dir):
return theme_dir_translations
else:
self.fail()
with mock.patch('mkdocs.localization.Translations.load', side_effect=side_effet):
install_translations(self.env, parse_locale('en'), [custom_dir, theme_dir])
theme_dir_translations.merge.assert_called_once_with(custom_dir_translations)
|
