python - 关闭单元测试的翻译

Question

我有一些 python 代码，比如

from pylons.i18n.translation import _

def get_message():
     message = _(u"Translated message")
     # interesting code to test
     # [...]
     return 'result'

我想像这样进行单元测试:

class MyTest(TestCase):
    def test_get_message(self):
        assertTrue(get_message(), 'result')

现在在 nosetests 中运行这个测试给我:

TypeError: No object (name: translator) has been registered for this thread

有没有办法在单元测试时停用与翻译相关的任何内容？

Answer 1

假设您的生产代码在 my_module.py 中:

from unittest import TestCase
from mock import patch
from my_module import get_message


class MyTest(TestCase):
    def test_get_message(self):
        with patch("my_module._"):
            result = get_message()
            self.assertEqual("result", result)

使用补丁，您的测试将_() 函数更改为MagicMock() 对象。文档 here .

注意 mock 是 Python 3.3 及更高版本标准库的一部分。否则，您应该首先使用 pip install mock 安装它。