python - Python的模拟库，在模拟函数中只返回一次值？

Question

使用 mock 模拟方法很简单，例如，像这样:

    o.get_idlist = mock.Mock(return_value=[1])

但是，这将使 get_idlist() 每次都返回 [1]。有没有办法让它只返回 [1] 一次，后续调用返回 [] (例如)而不使用 side_effect=somefun ？我想我想要的是一种副作用，但就是这样。

(我只是觉得为此创建一个大的命名函数太笨拙了......或者有没有办法在这里使用 lambda？)

Answer 1

假设您没有任何反对 side_effect 本身的东西，但只是不想创建整个函数，您可以粗略地执行如下操作

>>> from unittest.mock import Mock
>>> m = Mock()
>>> m.side_effect = [1] + [[]]*100
>>> m() 
1
>>> m()
[] 
>>> m()
[]
>>> m()
[]
>>> m()
[]

我承认它有点脏，但如果你只是想进行快速测试，它应该足够了。当然，您可以将 [] 替换为 [0] 或任何您喜欢的内容。