包含属性的伪造模块

Question

作为一个单元测试，我希望能够以编程方式测试在一堆 GUI 屏幕中定义的导入是否有意义。我不想执行任何操作，只是

import main_screen

并查看在导入链中是否有任何问题。我已经使用 sys.modules 技巧了：

import my_fake_module
sys.modules['module_i_want_to_fake'] = my_fake_module

效果很好，直到有什么效果：

from module_i_want_to_fake import real_attribute

然后我当然会得到一个 ImportError，除非我向 my_fake_module 添加一个存根的方法/类，但是有太多的方法无法实现。

有什么方法可以挂钩来自 my_fake_module 的导入，以便它们始终成功？同样，他们不需要做任何事情。我敢打赌有一个简单的方法，但它现在正在逃避我......

Answer 1

一种选择是创建一个自定义的FakeModule 类来处理__getattr__ 并分发适当的存根函数/值：

>>> class FakeModule:
...     def __getattr__(self, name):
...         print name
...         return name
...
>>> sys.modules['fakeModule'] = FakeModule()
>>> from fakeModule import something
__path__
something
something
>>> something
'something'

实际代码需要比上面的快速概念证明多一点思考，可能会返回一个 lambda，因此您至少可以执行导入的值。

你可能想做这样的事情：

from functools import wraps

class FakeModule(object):
    def __init__(self):
        self.module = __import__("module_to_fake")

    @staticmethod
    def __testWrapper(fn):
        @wraps(fn)
        def wrapped(*args):
            print "%s called with args %s" % (fn.__name__, args)
            result = fn(*args)
            print "    Result was %s" % (result,)
            return result
        return wrapped

    def __getattr__(self, name):
        try:
            result = getattr(module, name)
        except AttributeError:
            print "tried to get %s, which does not exist" % name
            result = None

        if hasattr(result, "__call__"):
            result = self.__testWrapper(result)

        return result