2

单元测试运行功能的最简单方法是什么？在这里，我对fun * 函数的作用不感兴趣，但前提是它们的调用顺序正确。

from mock import Mock

(many, arguments, four, fun) = ('many', 'arguments', 'four', 'fun') 

class TemplateMethod(object):

    def fun1(self, many, parameters, go, here):
        raise NotImplementedError()

    @classmethod
    def fun2(cls, many, parameters, go, here):
        pass

    @staticmethod
    def fun3(many, parameters, go, here):
        pass

    def run(self):
        result1 = self.fun1(many, arguments, four, fun)
        result1.do_something()

        self.fun2(many, arguments, four, fun)
        self.fun3(many, arguments, four, fun)

该解决方案的唯一要求是对被测类的非侵入性。

解决方案：

实际上，这有点像草稿，这个简单的类可以以多种方式进行修复和扩展，我什至都懒得去想它。关键是您现在可以简单地将所有方法的所有调用记录在模板方法中。您还可以指定不应被该类模拟的对象列表（即您正在记录调用的函数）。

特别感谢@Damian Schenkelman，他给了我一些重要的指示。

class MockingInvocationRecorder(object):

    FUNCTION_TYPES = ('function', 'classmethod', 'staticmethod')

    def __init__(self, obj, dont_mock_list):
        self._invocation_list = []

        name_list = [exc.__name__ for exc in dont_mock_list]
        self._wrap_memfuns(obj, name_list)

    @property
    def invocations(self):
        return tuple(self._invocation_list)

    def _wrap_single(self, memfun, exclude_list):
        def wrapper(*args, **kwargs):
            self._invocation_list.append(memfun.__name__)

            if memfun.__name__ in exclude_list:
                return memfun(*args, **kwargs)
            else:
                return Mock()

        return wrapper

    def _wrap_memfuns(self, obj, exclude_list):
        for (mem_name, mem) in type(obj).__dict__.iteritems():
            if type(mem).__name__ in self.FUNCTION_TYPES:
                wrapper = self._wrap_single(getattr(obj, mem_name), exclude_list)
                setattr(obj, mem_name, wrapper)

您现在可以通过以下方式测试调用顺序：

tm = TemplateMethod()
ir = MockingInvocationRecorder(tm, [tm.run])

tm.run()

print ir.invocations => ('run', 'fun1', 'fun2', 'fun3')

您可以通过类名引用成员：

ir = MockingInvocationRecorder(tm, [TemplateMethod.run])

请注意，您列出了所有不应该被嘲笑的方法。

4

2 回答 2

3

Monkeypatch fun1、fun2、fun3 函数并使用列表来跟踪调用顺序。之后，断言在列表中。

于 2012-09-22T16:22:32.913 回答
2

在这里使用__call__可能会有所帮助。

class Function1(object):
    def __call__(self, params, logger=None):
        if logger is not None:
            logger.append(self.__repr__())
        return self.function1(params) 
        # Or no return if only side effects are needed

    def __init__(self, ...):
        #...

    def __repr__(self):
        # Some string representation of your function.

    def function1(params):
        print "Function 1 stuff here..."

然后做类似的事情

class TemplateMethod(object):
    def __init__(self, params):
        self.logger = []
        self.fun1 = Function1(...)
        self.fun1(params, self.logger)

这很老套；可能有一些方法可以清理我得到的东西，但是将函数封装在类中然后使用__call__是一个很好的方法。

于 2012-09-22T20:43:44.543 回答