Question

单元测试运行功能的最简单方法是什么？在这里，我对 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)

解决方案的唯一要求是对被测试阶级的非侵入性。

SOLUTION：

实际上，这有点像草案，这个简单的类可以通过很多方式修复和扩展，我甚至不在乎考虑它。关键是你现在可以简单地记录模板方法中所有方法的所有调用。您还可以指定不应该被类模拟的对象列表（即您正在记录调用的函数）。

特别感谢@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)

这非常hacky;可能有一些方法可以清理我所获得的内容，但是将这些函数封装在类中然后使用__call__是一个很好的方法。

python中的模板方法调用顺序

2 个答案: