test_intercept.py 文件源码

def test_monitor():
    """A CallMonitor records each call to a method, and its arguments."""

    class SomethingElse(object):
        def foo(self, n, y=None):
            self.n = n
            return y

    s = SomethingElse()
    original_method = six.get_method_function(s.foo)

    with CallMonitor(s.foo) as monitor:
        assert s.foo(1, y='a') == 'a'
        assert s.foo(2) == None

    assert six.get_method_function(s.foo) is original_method
    assert s.n == 2
    assert monitor.times_called == 2
    assert monitor.calls[0].args == (1,)
    assert monitor.calls[0].kwargs == {'y':'a'}
    assert monitor.calls[0].return_value == 'a'

    assert monitor.calls[1].args == (2,)
    assert monitor.calls[1].kwargs == {}
    assert monitor.calls[1].return_value == None