python中的模板方法调用顺序

Question

单元测试 run 函数的最简单方法是什么？这里我对 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])

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

Answer 1

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

Answer 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__ 是一个不错的方法。