1 files changed, 382 insertions, 0 deletions
diff --git a/repoze/bfg/tests/test_i18n.py b/repoze/bfg/tests/test_i18n.py
new file mode 100644
index 000000000..9c4b226e4
--- /dev/null
+++ b/repoze/bfg/tests/test_i18n.py
@@ -0,0 +1,382 @@
+# -*- coding: utf-8 -*-
+#
+
+import unittest
+from repoze.bfg.testing import cleanUp
+
+class TestTranslationString(unittest.TestCase):
+    def _makeOne(self, *arg, **kw):
+        from repoze.bfg.i18n import TranslationString
+        return TranslationString(*arg, **kw)
+
+    def test_it(self):
+        # this is part of the API, we don't actually need to test much more
+        # than that it's importable
+        ts = self._makeOne('a')
+        self.assertEqual(ts, 'a')
+
+class TestTranslationStringFactory(unittest.TestCase):
+    def _makeOne(self, *arg, **kw):
+        from repoze.bfg.i18n import TranslationStringFactory
+        return TranslationStringFactory(*arg, **kw)
+
+    def test_it(self):
+        # this is part of the API, we don't actually need to test much more
+        # than that it's importable
+        factory = self._makeOne('a')
+        self.assertEqual(factory('').domain, 'a')
+
+class TestLocalizer(unittest.TestCase):
+    def _makeOne(self, *arg, **kw):
+        from repoze.bfg.i18n import Localizer
+        return Localizer(*arg, **kw)
+
+    def test_ctor(self):
+        localizer = self._makeOne('en_US', None)
+        self.assertEqual(localizer.locale_name, 'en_US')
+        self.assertEqual(localizer.translations, None)
+
+    def test_translate(self):
+        translations = DummyTranslations()
+        localizer = self._makeOne(None, translations)
+        self.assertEqual(localizer.translate('123', domain='1',
+                                             mapping={}), '123')
+        self.failUnless(localizer.translator)
+
+    def test_pluralize(self):
+        translations = DummyTranslations()
+        localizer = self._makeOne(None, translations)
+        self.assertEqual(localizer.pluralize('singular', 'plural', 1,
+                                             domain='1', mapping={}),
+                         'singular')
+        self.failUnless(localizer.pluralizer)
+
+class Test_negotiate_locale_name(unittest.TestCase):
+    def setUp(self):
+        cleanUp()
+
+    def tearDown(self):
+        cleanUp()
+
+    def _callFUT(self, request):
+        from repoze.bfg.i18n import negotiate_locale_name
+        return negotiate_locale_name(request)
+
+    def _registerImpl(self, impl):
+        from repoze.bfg.threadlocal import get_current_registry
+        registry = get_current_registry()
+        from repoze.bfg.interfaces import ILocaleNegotiator
+        registry.registerUtility(impl, ILocaleNegotiator)
+
+    def test_no_registry_on_request(self):
+        self._registerImpl(dummy_negotiator)
+        request = DummyRequest()
+        result = self._callFUT(request)
+        self.assertEqual(result, 'bogus')
+
+    def test_with_registry_on_request(self):
+        from repoze.bfg.threadlocal import get_current_registry
+        registry = get_current_registry()
+        self._registerImpl(dummy_negotiator)
+        request = DummyRequest()
+        request.registry = registry
+        result = self._callFUT(request)
+        self.assertEqual(result, 'bogus')
+
+    def test_default_from_settings(self):
+        from repoze.bfg.threadlocal import get_current_registry
+        registry = get_current_registry()
+        settings = {'default_locale_name':'settings'}
+        from repoze.bfg.interfaces import ISettings
+        registry.registerUtility(settings, ISettings)
+        request = DummyRequest()
+        request.registry = registry
+        result = self._callFUT(request)
+        self.assertEqual(result, 'settings')
+
+    def test_default_default(self):
+        request = DummyRequest()
+        result = self._callFUT(request)
+        self.assertEqual(result, 'en')
+
+class Test_get_locale_name(unittest.TestCase):
+    def setUp(self):
+        cleanUp()
+
+    def tearDown(self):
+        cleanUp()
+
+    def _callFUT(self, request):
+        from repoze.bfg.i18n import get_locale_name
+        return get_locale_name(request)
+
+    def _registerImpl(self, impl):
+        from repoze.bfg.threadlocal import get_current_registry
+        registry = get_current_registry()
+        from repoze.bfg.interfaces import ILocaleNegotiator
+        registry.registerUtility(impl, ILocaleNegotiator)
+
+    def test_name_on_request(self):
+        request = DummyRequest()
+        request.bfg_locale_name = 'ie'
+        result = self._callFUT(request)
+        self.assertEqual(result, 'ie')
+
+    def test_name_not_on_request(self):
+        self._registerImpl(dummy_negotiator)
+        request = DummyRequest()
+        result = self._callFUT(request)
+        self.assertEqual(result, 'bogus')
+        self.assertEqual(request.bfg_locale_name, 'bogus')
+
+class Test_get_localizer(unittest.TestCase):
+    def setUp(self):
+        cleanUp()
+
+    def tearDown(self):
+        cleanUp()
+
+    def _callFUT(self, request):
+        from repoze.bfg.i18n import get_localizer
+        return get_localizer(request)
+
+    def test_no_registry_on_request(self):
+        request = DummyRequest()
+        request.bfg_localizer = '123'
+        result = self._callFUT(request)
+        self.assertEqual(result, '123')
+
+    def test_with_registry_on_request(self):
+        from repoze.bfg.threadlocal import get_current_registry
+        registry = get_current_registry()
+        request = DummyRequest()
+        request.bfg_localizer = '123'
+        request.registry = registry
+        result = self._callFUT(request)
+        self.assertEqual(result, '123')
+
+    def test_locale_on_request(self):
+        request = DummyRequest()
+        request.bfg_localizer = 'abc'
+        result = self._callFUT(request)
+        self.assertEqual(result, 'abc')
+
+    def test_locale_from_registry(self):
+        from repoze.bfg.threadlocal import get_current_registry
+        from repoze.bfg.interfaces import ILocalizer
+        registry = get_current_registry()
+        locale = 'abc'
+        registry.registerUtility(locale, ILocalizer, name='en')
+        request = DummyRequest()
+        request.bfg_locale_name = 'en'
+        result = self._callFUT(request)
+        self.assertEqual(result, 'abc')
+
+    def test_locale_from_mo(self):
+        import os
+        from repoze.bfg.threadlocal import get_current_registry
+        from repoze.bfg.interfaces import ITranslationDirectories
+        from repoze.bfg.i18n import Localizer
+        registry = get_current_registry()
+        here = os.path.dirname(__file__)
+        localedir = os.path.join(here, 'fixtures', 'locale')
+        localedirs = [localedir]
+        registry.registerUtility(localedirs, ITranslationDirectories)
+        request = DummyRequest()
+        request.bfg_locale_name = 'de'
+        result = self._callFUT(request)
+        self.assertEqual(result.__class__, Localizer)
+        self.assertEqual(result.translate('Approve', 'deformsite'),
+                         'Genehmigen')
+        self.assertEqual(result.translate('Approve'), 'Approve')
+        self.failUnless(hasattr(result, 'pluralize'))
+
+    def test_locale_from_mo_bad_mo(self):
+        import os
+        from repoze.bfg.threadlocal import get_current_registry
+        from repoze.bfg.interfaces import ITranslationDirectories
+        from repoze.bfg.i18n import Localizer
+        registry = get_current_registry()
+        here = os.path.dirname(__file__)
+        localedir = os.path.join(here, 'fixtures', 'locale')
+        localedirs = [localedir]
+        registry.registerUtility(localedirs, ITranslationDirectories)
+        request = DummyRequest()
+        request.bfg_locale_name = 'be'
+        result = self._callFUT(request)
+        self.assertEqual(result.__class__, Localizer)
+        self.assertEqual(result.translate('Approve', 'deformsite'),
+                         'Approve')
+
+class Test_default_locale_negotiator(unittest.TestCase):
+    def setUp(self):
+        cleanUp()
+
+    def tearDown(self):
+        cleanUp()
+
+    def _callFUT(self, request):
+        from repoze.bfg.i18n import default_locale_negotiator
+        return default_locale_negotiator(request)
+
+    def test_from_settings(self):
+        from repoze.bfg.interfaces import ISettings
+        from repoze.bfg.threadlocal import get_current_registry
+        settings = {'default_locale_name':'dude'}
+        registry = get_current_registry()
+        registry.registerUtility(settings, ISettings)
+        request = DummyRequest()
+        result = self._callFUT(request)
+        self.assertEqual(result, 'dude')
+        
+    def test_settings_empty(self):
+        request = DummyRequest()
+        result = self._callFUT(request)
+        self.assertEqual(result, 'en')
+        
+    def test_from_params(self):
+        request = DummyRequest()
+        request.params['locale'] = 'foo'
+        result = self._callFUT(request)
+        self.assertEqual(result, 'foo')
+
+class TestTranslations(unittest.TestCase):
+    def _getTargetClass(self):
+        from repoze.bfg.i18n import Translations
+        return Translations
+        
+    def _makeOne(self):
+        messages1 = [
+            ('foo', 'Voh'),
+            (('foo1', 1), 'Voh1'),
+        ]
+        messages2 = [
+            ('foo', 'VohD'),
+            (('foo1', 1), 'VohD1'),
+        ]
+
+        klass = self._getTargetClass()
+        
+        translations1 = klass(None, domain='messages')
+        translations1._catalog = dict(messages1)
+        translations1.plural = lambda *arg: 1
+        translations2 = klass(None, domain='messages1')
+        translations2._catalog = dict(messages2)
+        translations2.plural = lambda *arg: 1
+        translations = translations1.add(translations2, merge=False)
+        return translations
+
+    def test_load_domain_None(self):
+        import gettext
+        import os
+        here = os.path.dirname(__file__)
+        localedir = os.path.join(here, 'fixtures', 'locale')
+        locales = ['de', 'en']
+        klass = self._getTargetClass()
+        result = klass.load(localedir, locales, domain=None)
+        self.assertEqual(result.__class__, gettext.NullTranslations)
+
+    def test_load_found_locale_and_domain(self):
+        import os
+        here = os.path.dirname(__file__)
+        localedir = os.path.join(here, 'fixtures', 'locale')
+        locales = ['de', 'en']
+        klass = self._getTargetClass()
+        result = klass.load(localedir, locales, domain='deformsite')
+        self.assertEqual(result.__class__, klass)
+
+    def test_load_found_locale_and_domain_locale_is_string(self):
+        import os
+        here = os.path.dirname(__file__)
+        localedir = os.path.join(here, 'fixtures', 'locale')
+        locales = 'de'
+        klass = self._getTargetClass()
+        result = klass.load(localedir, locales, domain='deformsite')
+        self.assertEqual(result.__class__, klass)
+
+    def test___repr__(self):
+        inst = self._makeOne()
+        result = repr(inst)
+        self.assertEqual(result, '<Translations: "None">')
+
+    def test_merge_not_gnutranslations(self):
+        inst = self._makeOne()
+        self.assertEqual(inst.merge(None), inst)
+
+    def test_merge_gnutranslations(self):
+        inst = self._makeOne()
+        inst2 = self._makeOne()
+        inst2._catalog['a'] = 'b'
+        inst.merge(inst2)
+        self.assertEqual(inst._catalog['a'], 'b')
+
+    def test_add_different_domain_merge_true_notexisting(self):
+        inst = self._makeOne()
+        inst2 = self._makeOne()
+        inst2.domain = 'domain2'
+        inst.add(inst2)
+        self.assertEqual(inst._domains['domain2'], inst2)
+
+    def test_add_different_domain_merge_true_existing(self):
+        inst = self._makeOne()
+        inst2 = self._makeOne()
+        inst3 = self._makeOne()
+        inst2.domain = 'domain2'
+        inst2._catalog['a'] = 'b'
+        inst3.domain = 'domain2'
+        inst._domains['domain2'] = inst3
+        inst.add(inst2)
+        self.assertEqual(inst._domains['domain2'], inst3)
+        self.assertEqual(inst3._catalog['a'], 'b')
+
+    def test_add_same_domain_merge_true(self):
+        inst = self._makeOne()
+        inst2 = self._makeOne()
+        inst2._catalog['a'] = 'b'
+        inst.add(inst2)
+        self.assertEqual(inst._catalog['a'], 'b')
+
+    def test_dgettext(self):
+        t = self._makeOne()
+        self.assertEqual(t.dgettext('messages', 'foo'), 'Voh')
+        self.assertEqual(t.dgettext('messages1', 'foo'), 'VohD')
+
+    def test_ldgettext(self):
+        t = self._makeOne()
+        self.assertEqual(t.ldgettext('messages', 'foo'), 'Voh')
+        self.assertEqual(t.ldgettext('messages1', 'foo'), 'VohD')
+
+    def test_dugettext(self):
+        t = self._makeOne()
+        self.assertEqual(t.dugettext('messages', 'foo'), 'Voh')
+        self.assertEqual(t.dugettext('messages1', 'foo'), 'VohD')
+
+    def test_dngettext(self):
+        t = self._makeOne()
+        self.assertEqual(t.dngettext('messages', 'foo1', 'foos1', 1), 'Voh1')
+        self.assertEqual(t.dngettext('messages1', 'foo1', 'foos1', 1), 'VohD1')
+        
+    def test_ldngettext(self):
+        t = self._makeOne()
+        self.assertEqual(t.ldngettext('messages', 'foo1', 'foos1', 1), 'Voh1')
+        self.assertEqual(t.ldngettext('messages1', 'foo1', 'foos1', 1), 'VohD1')
+
+    def test_dungettext(self):
+        t = self._makeOne()
+        self.assertEqual(t.dungettext('messages', 'foo1', 'foos1', 1), 'Voh1')
+        self.assertEqual(t.dungettext('messages1', 'foo1', 'foos1', 1), 'VohD1')
+
+
+class DummyRequest(object):
+    def __init__(self):
+        self.params = {}
+
+def dummy_negotiator(request):
+    return 'bogus'
+
+class DummyTranslations(object):
+    def ugettext(self, text):
+        return text
+
+    def ungettext(self, singular, plural, n):
+        return singular