模拟按顺序修补多个用户输入答案

【问题标题】：mock patch multiple user inputs in sequence模拟按顺序修补多个用户输入
【发布时间】：2021-02-16 10:38:44
【问题描述】：

类似的问题已被问过很多次，但我似乎无法弄清楚我正在尝试构建的这个简单测试：我想先提供一个“y”，然后为一个复杂的函数提供一个“n”需要用户输入（即它需要按顺序输入两个输入）。这是我的尝试 - with 语句不会推进迭代器，但我不知道我将如何实现修补输入。

import mock

m = mock.Mock()
m.side_effect = ["y","n"]
    
@pytest.fixture(scope="module")
def test_my_complex_function():
    with mock.patch('builtins.input', return_value=m()):
        out = my_complex_function(some_args)
    return out

【问题讨论】：

这个输入/答案是连续的？
字符串 - 首先是“y”然后是“n”

标签： python testing mocking

【解决方案1】：

如果我很好地理解了这个问题，那么您就有一个具有类似行为的函数。

module.py

def complex_function():
    first = input("First input")
    second = input("Second input")
    return first, second

并且您想模拟 input 内置方法。你的方法是对的，唯一要解决的问题是你必须构建 2 个模拟。每个input 实例一个。

test_module.py

import pytest

from unittest.mock import Mock, patch

from module import complex_function

input_mock_y = Mock()  # First mock for first input call
input_mock_n = Mock()  # Second mock for second input call

input_mock = Mock()    # Combine the 2 mocks in another mock to patch the input call.
input_mock.side_effect = [input_mock_y.return_value, input_mock_n.return_value]

def test_my_complex_function():
    with patch('builtins.input', input_mock) as mock_input:
        result = complex_function()
    
    assert mock_method.call_count == 2

您可能会说：好的，但我怎么知道每个输入都被正确修补了？因此，您可以为任何输入模拟指定一些返回值，以便进行比较。

input_mock_y = Mock()
input_mock_y.return_value = "Y"

input_mock_n = Mock()
input_mock_n.return_value = "N"

input_mock = Mock()
input_mock.side_effect = [input_mock_y.return_value, input_mock_n.return_value]

def test_my_complex_function():
    with patch('builtins.input', input_mock) as mock_method:
        result = function()
    
    assert mock_method.call_count == 2
    assert result == ('Y', 'N')

【讨论】：