mockmethod.py

mockmethod.py

# I couldn't figure out how to do this with Mock or MagicMock

class MockMethod(object):
    def __init__(self, *return_values):
        self.return_values = return_values
        self.called = 0
        self.call_args = []

    def __call__(self, *args, **kwargs):
        self.called += 1
        self.call_args.append([args, kwargs])
        return self.return_value

    @property
    def return_value(self):
        ret = self.return_values[0]
        if len(self.return_values) > 1:
            self.return_values = self.return_values[1:]
        return ret

This allows the method being mocked to return potentially different values with subsequent calls. Interesting for mocking user-input, for example when validating or prompting repeatedly and needing different return values.