检索字符串以进行单元测试

Question

考虑这段代码：

def get_some_text():
    return _(u"We need this text here")

编写单元测试以确保函数返回 this - 由 Babel 引用和翻译 - 字符串的最佳方法是什么？

这个幼稚的代码不会包含引用——所以这实际上是一个不同的“Babel 字符串”：

def test_get_some_text(self):
    self.assertEqual(get_some_text(), _(u"We need this text here"))

Answer 1

如果您使用的是flask-babel，请不要介意使用模拟，您只是在测试_ 返回的内容（即get_some_text 不会对_ 的结果进行任何额外的转换） , 然后你可以模拟_ 的返回值并测试你得到了你期望的结果：

import mock


@mock.patch('gettext.ugettext')
def test_get_some_text(self, mock_ugettext):
  mock_ugettext.return_value = u'Hello World'

  self.assertEqual(get_some_text(), u'Hello World')

我们知道_ 从here 调用ugettext，如果我们在前面删除import ipdb; ipdb.set_trace() 行，我们可以跳入python shell，调用get_some_text，然后使用this answer 来查找ugettext的导入路径，原来是gettext.ugettext。

如果你只使用 babel，并且你知道你的翻译目录的路径，你也许可以 make your own translations 只是为了测试：

import os
import shutil

import polib    


os_locale = ''
translations_directory = '/absolute/path/to/your/translations'
# choose a locale that isn't already in translations_directory
test_locale = 'en_GB'

def setUp(self):
    mo_file_path = os.path.join(
        translations_directory,
        test_locale,
        'LC_MESSAGES',
        'messages.mo'
    )
    mo = polib.MOFile()
    entry = polib.MOEntry(
        msgid=u'We need this text here',
        msgstr='My favourite colour is grey'
    )
    mo.append(entry)
    mo.save(mo_file_path)

    # modify our locale for the duration of this test
    os_locale = os.environ.pop('LANG', 'en')
    os.environ['LANG'] = test_locale

def tearDown(self):
    # restore our locale
    os.environ['LANG'] = os_locale
    shutil.rmtree(os.path.join(translations_directory, test_locale))

def test_get_some_text(self):
    self.assertEqual(get_some_text(), u'My favourite colour is grey')