4

假设我有一个名为 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 装饰器通过用断言对象字段的方法替换原始的“发送”方法来完成它：

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

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

但在这种情况下，我必须为每个方法检查定义一个单独的模拟函数，并且我无法检查（没有额外的编码）该函数是否已被调用。

4

2 回答 2

7

你可以得到模拟的最后一个参数

request, = conn.send.call_args

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

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

于 2013-07-12T17:20:51.527 回答
1

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

例如，它会是这样的：

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 获取参数，但我认为这样看起来更好。

于 2013-07-12T17:35:58.700 回答