Python：在测试期间模拟 gettext 的装饰器

Question

我正在编写一个 Django 应用程序，我想编写一个装饰器，在测试期间模拟 _() 函数，只需在要翻译的字符串后添加“_translated”

我基本上让我的装饰器在我的测试中替换以下指令：

with mock.patch('module_which_is_being_tested._', lambda s: s+'_translated'):

Answer 1

我没有在网上找到类似的东西，所以我分享它：

from unittest import mock
import functools

def mock_translation(tested_object):
    """
    A decorator that mock the '_()' function during testing
    just adds '_translated' after the string to translate

    the class/function being tested needs to be passed as parameter

    Use: 
    from my_module.sub_module import my_function_to_be_tested

    @mock_translation(my_function_to_be_tested)
    def test_my_function_to_be_tested():
        pass
    """
 
    def actual_decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            module_name = tested_object.__module__
            tested_obj = tested_object
            breakpoint()
            import_path = module_name + '._'
            # with mock.patch('battletech.models._', lambda s: s+'_translated'):
            with mock.patch(import_path, lambda s: s+'_translated'):
                value = func(*args, **kwargs)
                return value
        return wrapper
    return actual_decorator