Python verspottet mehrere Rückgabewerte

Ich verwende pythons mock.patch und möchte den Rückgabewert für jeden Aufruf ändern. Hier ist die Einschränkung: Die zu patchende Funktion hat keine Eingaben, daher kann ich den Rückgabewert nicht basierend auf der Eingabe ändern.

Hier ist mein Code als Referenz.

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

Mein Testcode:

@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.promptist nur eine plattformunabhängige (Python 2 und 3) Version von "input". Letztendlich versuche ich also, die Benutzereingaben zu verspotten. Ich habe versucht, eine Liste für den Rückgabewert zu verwenden, aber das funktioniert nicht.

Sie können sehen, dass ich hier nur eine Endlosschleife bekomme, wenn der Rückgabewert etwas Ungültiges ist. Ich brauche also eine Möglichkeit, den Rückgabewert zu ändern, damit mein Test tatsächlich beendet wird.

(Eine andere Möglichkeit, diese Frage zu beantworten, könnte darin bestehen, zu erklären, wie ich Benutzereingaben in einem Komponententest nachahmen kann.)

Kein Problem mit dieser Frage, vor allem, weil ich nicht in der Lage bin, die Eingaben zu variieren.

Einer der Kommentare der Antwort auf diese Frage ist in die gleiche Richtung, es wurde jedoch keine Antwort / kein Kommentar abgegeben.

Sie können einen zuweisen iterable zu side_effect, und das Mock wird den nächsten Wert in der Folge bei jedem Aufruf zurückkehren wird:

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

Zitieren der Mock()Dokumentation :

Wenn side_effect eine Iterable ist, gibt jeder Aufruf des Mocks den nächsten Wert aus der Iterable zurück.

Als Nebenwirkung der Test response is not 'y' or 'n' or 'yes' or 'no'wird nicht funktionieren; Sie fragen, ob der Ausdruck (response is not 'y')wahr oder 'y'wahr ist (immer der Fall, eine nicht leere Zeichenfolge ist immer wahr) usw. Die verschiedenen Ausdrücke auf beiden Seiten der orOperatoren sind unabhängig . Siehe Wie teste ich eine Variable gegen mehrere Werte?

Sie sollten auch nicht iszum Testen gegen eine Zeichenfolge verwenden. Der CPython-Interpreter kann unter bestimmten Umständen Zeichenfolgenobjekte wiederverwenden. Dies ist jedoch kein Verhalten, auf das Sie zählen sollten.

Verwenden Sie daher:

response not in ('y', 'n', 'yes', 'no')

stattdessen; Dabei werden Gleichheitstests ( ==) verwendet, um festzustellen, ob responseauf eine Zeichenfolge mit demselben Inhalt (Wert) verwiesen wird.

Gleiches gilt für response == 'y' or 'yes'; Verwenden Sie response in ('y', 'yes')stattdessen.