2

如何使用 Pythonmock库断言对不同模拟对象的特定调用序列？

例如，我想断言：

  • 一个电话foo(spam, eggs)；然后
  • 一个电话bar(beans, ham)；然后
  • 的呼唤foo(sausage)。

我可以修补每个fooand bar，并且生成的模拟对象每个都允许我对对该模拟的调用进行断言。但是我需要访问整个调用序列来对该序列进行断言。

是的，理想情况下，我只需要检查结果状态并在事后对其进行断言。但这对于某些系统是不可行的，正确状态的唯一可行描述是“这些调用是按照这个特定的顺序进行的”。

我可以使用库的哪些功能mock来访问对不同对象的一系列调用，并断言调用按正确的顺序进行？

4

2 回答 2

3

Mock 实际上提供了类似这样的内置功能。模拟经常有一个父模拟......例如

somemock.foo  # parent is somemock

父级没有直接暴露在模拟 API 中，但是对子级的调用是在父级上注册的。

import mock
m = mock.Mock()
m.a('hello world')
m.b('goodbye world')
m.c('kittens!')
m.a('Howdy')
m.assert_has_calls([
  mock.call.a('hello world'),
  mock.call.b('goodbye world'),
  mock.call.c('kittens!'),
  mock.call.a('Howdy')
])  # passes silently
m.assert_has_calls([
  mock.call.a('hello world'),
  mock.call.b('goodbye world'),
  mock.call.a('Howdy'),
  mock.call.c('kittens!')
]) # Error
# Traceback (most recent call last):
#   File "<stdin>", line 1, in <module>
#   File "/usr/local/lib/python2.7/dist-packages/mock.py", line 863, in assert_has_calls
#     'Actual: %r' % (calls, self.mock_calls)
# AssertionError: Calls not found.
# Expected: [call.a('hello world'), call.b('goodbye world'), call.a('Howdy'), call.c('kittens!')]
# Actual: [call.a('hello world'),
#  call.b('goodbye world'),
#  call.c('kittens!'),
#  call.a('Howdy')]

但是，你可能会说“我的模拟并不都来自同一个父母”。一切还没有丢失！您可以创建一个父级并在事后将它们附加到它。

parent_mock = mock.Mock()
parent_mock.attach_mock(foomock, 'foo')
parent_mock.attach_mock(barmock, 'bar')

现在你可以做我们上面做的同样的断言（只要你不需要保留原始模拟的父母......那么我不知道该告诉你什么......）

于 2014-12-05T00:44:51.547 回答
0

解决这个问题的最初尝试是使用一个专门的Mock子类，它在提供的序列对象中注册调用，该对象可以是您喜欢的任何共享序列。

from copy import deepcopy
import mock

class CallRegisterMock(mock.MagicMock):
    """ A mock object that registers each call. """

    def __init__(self, call_register, *args, **kwargs):
        super(CallRegisterMock, self).__init__(*args, **kwargs)

        self.call_register = call_register

    def __call__(self, *args, **kwargs):
        args = deepcopy(args)
        kwargs = deepcopy(kwargs)
        call = mock.call(*args, **kwargs)
        qualified_call = (self, call)
        self.call_register.append(qualified_call)
        super(CallRegisterMock, self).__call__(*args, **kwargs)

这有缺点：

  • 它可能正在重新发明一个或多个轮子。（如果您认为是，请添加更好的答案。）
于 2014-12-05T00:32:21.287 回答