Question

假设我有一个名为Client的类，它创建Request类的对象并将其传递给Connection对象的方法：

class Client(object):
    def __init__(self, connection):
        self._conn = connection

    def sendText(plaintext):
        self._conn.send(Request(0, plaintext))

我想断言传递给Connection.send方法的对象来检查它的属性。我首先创建一个模拟的Connection类：

conn = Mock()

client = Client(conn)
client.sendText('some message')

然后我想要类似的东西：

conn.send.assert_called_with(
    (Request,
    {'type': 0, 'text': 'some message'})
)

其中'type'和'text'是Request的属性。有没有办法在python的模拟中执行此操作？我在文档中找到的只是简单的数据示例。我可以用mock.patch装饰器完成它，用一个断言对象字段的方法替换原来的'send'方法：

def patchedSend(self, req):
    assert req.Type == 0

with mock.patch.object(Connection, 'send', TestClient.patchedSend):
    ...

但是在这种情况下，我必须为每个方法检查定义一个separete mocked函数，如果函数已被调用，我无法检查（没有额外的编码）。

Answer 1

您可以使用

获取模拟的最后一个参数

request, = conn.send.call_args

然后断言关于它的属性。如果您希望工具表达更复杂的关于事物的断言，您可以安装PyHamcrest。

注意：不要在单元测试中使用assert。使用assertEqual或assertTrue等断言方法。断言方法无法被意外关闭，并且它们可以提供比assert语句更有用的消息。

Answer 2

嗯，我认为在这种特定情况下，最简单和更好的方法是创建一个函数来创建请求，然后模拟它。

例如，它就像它一样：

class Client(object):
    def __init__(self, connection):
        self._conn = connection

    def _create_request(self, plain_text):
        return Request(0, plain_text)

    def send_text(self, plaintext):
        self._conn.send(self._create_request(plain_text))

然后，在测试中，您可以模拟_create_request以返回一些特定值，然后断言用它调用send_text。

您也可以按照建议通过call_args获取参数，但我认为这样看起来更好。

断言传递给python mock中的模拟方法的对象

2 个答案: