0

作为一个单元测试，我希望能够测试在一堆 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 的导入，以便它们始终成功？同样，他们不需要做任何事情。我打赌有一个简单的方法，但它现在正在逃避我......

4

1 回答 1

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
于 2013-05-20T13:48:56.097 回答