Python模拟多个返回值

Question

我正在使用pythons mock.patch并想更改每个调用的返回值。请注意，正在修补的函数没有输入，因此无法基于输入更改返回值。

这是我的代码供参考。

def get_boolean_response():
    response = io.prompt('y/n').lower()
    while response not in ('y', 'n', 'yes', 'no'):
        io.echo('Not a valid input. Try again'])
        response = io.prompt('y/n').lower()

    return response in ('y', 'yes')

我的测试代码：

@mock.patch('io')
def test_get_boolean_response(self, mock_io):
    #setup
    mock_io.prompt.return_value = ['x','y']
    result = operations.get_boolean_response()

    #test
    self.assertTrue(result)
    self.assertEqual(mock_io.prompt.call_count, 2)

io.prompt只是“ input”的独立于平台的版本（python
2和3）。因此，最终我将尝试模拟用户的输入。我已经尝试过使用列表作为返回值，但这并不能正常工作。

您可以看到，如果返回值无效，那么我将在此处得到一个无限循环。因此，我需要一种最终更改返回值的方法，以便测试实际上完成。

（回答这个问题的另一种可能的方式可能是解释如何在单元测试中模仿用户输入）

Answer 1

你可以指定一个 迭代
来side_effect，并模拟将序列中的每个被调用时，返回的下一个值：

>>> from unittest.mock import Mock
>>> m = Mock()
>>> m.side_effect = ['foo', 'bar', 'baz']
>>> m()
'foo'
>>> m()
'bar'
>>> m()
'baz'

引用Mock()文档：

如果 side_effect 是可迭代的，则每次对模拟的调用都将返回可迭代的下一个值。