Python simula múltiples valores de retorno

Estoy usando pythons mock.patch y me gustaría cambiar el valor de retorno para cada llamada. Aquí está la advertencia: la función que se parchea no tiene entradas, por lo que no puedo cambiar el valor de retorno en función de la entrada.

Aquí está mi código para referencia.

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')

Mi código de prueba:

@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 es solo una versión independiente de la plataforma (python 2 y 3) de “input”. Entonces, en última instancia, estoy tratando de simular la entrada de los usuarios. He intentado usar una lista para el valor de retorno, pero eso no parece funcionar.

Puede ver que si el valor de retorno es algo inválido, simplemente obtendré un bucle infinito aquí. Así que necesito una forma de cambiar eventualmente el valor de retorno, para que mi prueba realmente termine.

(otra forma posible de responder a esta pregunta podría ser explicar cómo podría imitar la entrada del usuario en una prueba unitaria)

No es un duplicado de esta pregunta principalmente porque no tengo la capacidad de variar las entradas.

Uno de los comentarios de la Respuesta sobre esta pregunta está en la misma línea, pero no se ha proporcionado ninguna respuesta/comentario.

Puede asignar un iterable a side_effecty el simulacro devolverá el siguiente valor en la secuencia cada vez que se llame:

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

citando el Mock() documentación:

Si efecto secundario es un iterable, cada llamada al simulacro devolverá el siguiente valor del iterable.

para múltiples valores de retorno, podemos usar efecto secundario durante la inicialización del parche también y pase iterable a él

muestra.py

def hello_world():
    pass

prueba_muestra.py

from unittest.mock import patch
from sample import hello_world

@patch('sample.hello_world', side_effect=[{'a': 1, 'b': 2}, {'a': 4, 'b': 5}])
def test_first_test(self, hello_world_patched):
    assert hello_world() == {'a': 1, 'b': 2}
    assert hello_world() == {'a': 4, 'b': 5}
    assert hello_world_patched.call_count == 2

También puede usar patch para múltiples valores de retorno:

@patch('Function_to_be_patched', return_value=['a', 'b', 'c'])

Recuerde que si está utilizando más de un parche para un método, el orden se verá así:

@patch('a')
@patch('b')
def test(mock_b, mock_a);
    pass

como se puede ver aquí, se revertirá. El primer parche mencionado estará en la última posición.