1 files changed, 57 insertions, 0 deletions
diff --git a/tests/test_locale.py b/tests/test_locale.py
new file mode 100644
index 0000000..1dcad64
--- /dev/null
+++ b/tests/test_locale.py
@@ -0,0 +1,57 @@
+"""Test locale."""
+
+import pytest
+
+from sphinx import locale
+
+
+@pytest.fixture(autouse=True)
+def cleanup_translations():
+    yield
+    locale.translators.clear()
+
+
+def test_init(rootdir):
+    # not initialized yet
+    _ = locale.get_translation('myext')
+    assert _('Hello world') == 'Hello world'
+    assert _('Hello sphinx') == 'Hello sphinx'
+    assert _('Hello reST') == 'Hello reST'
+
+    # load locale1
+    locale.init([rootdir / 'test-locale' / 'locale1'], 'en', 'myext')
+    _ = locale.get_translation('myext')
+    assert _('Hello world') == 'HELLO WORLD'
+    assert _('Hello sphinx') == 'Hello sphinx'
+    assert _('Hello reST') == 'Hello reST'
+
+    # load a catalog to unrelated namespace
+    locale.init([rootdir / 'test-locale' / 'locale2'], 'en', 'myext', 'mynamespace')
+    _ = locale.get_translation('myext')
+    assert _('Hello world') == 'HELLO WORLD'
+    assert _('Hello sphinx') == 'Hello sphinx'  # nothing changed here
+    assert _('Hello reST') == 'Hello reST'
+
+    # load locale2 in addition
+    locale.init([rootdir / 'test-locale' / 'locale2'], 'en', 'myext')
+    _ = locale.get_translation('myext')
+    assert _('Hello world') == 'HELLO WORLD'
+    assert _('Hello sphinx') == 'HELLO SPHINX'
+    assert _('Hello reST') == 'Hello reST'
+
+
+def test_init_with_unknown_language(rootdir):
+    locale.init([rootdir / 'test-locale' / 'locale1'], 'unknown', 'myext')
+    _ = locale.get_translation('myext')
+    assert _('Hello world') == 'Hello world'
+    assert _('Hello sphinx') == 'Hello sphinx'
+    assert _('Hello reST') == 'Hello reST'
+
+
+def test_add_message_catalog(app, rootdir):
+    app.config.language = 'en'
+    app.add_message_catalog('myext', rootdir / 'test-locale' / 'locale1')
+    _ = locale.get_translation('myext')
+    assert _('Hello world') == 'HELLO WORLD'
+    assert _('Hello sphinx') == 'Hello sphinx'
+    assert _('Hello reST') == 'Hello reST'