python - assert_has_calls 不会 stub 方法调用

Question

我正在使用模拟Python库中的assert_has_calls。当我做这样的事情时遇到了问题:

mocks = mock.Mock()
mocks.assert_has_calls([mock.call.method_to_be_mocked(mock.ANY),
                        mock.call.method_to_be_mocked(mock.ANY)])

我想验证该方法是否被调用了一定次数，但同时我也想删除对该方法的调用。 assert_has_calls 似乎没有做 stub 部分。实际的方法被调用，但在开发环境中失败。

我该如何解决这个问题？

Answer 1

您想断言它被调用了多少次，所以使用 call_count并使用 assert_equal (来自 unittest 或您正在使用的测试框架中的任何其他等效项)来验证它是否等于您期望调用它的次数:

assert_equal(mock.call.method_to_be_mocked.call_count, 2)