diff --git a/src/borg/testsuite/__init__.py b/src/borg/testsuite/__init__.py index 0a75d743d..724fcf894 100644 --- a/src/borg/testsuite/__init__.py +++ b/src/borg/testsuite/__init__.py @@ -249,6 +249,9 @@ class FakeInputs: def __init__(self, inputs): self.inputs = inputs + def available(self): + return bool(self.inputs) + def __call__(self, prompt=None): if prompt is not None: print(prompt, end="") diff --git a/src/borg/testsuite/helpers/yes_no_test.py b/src/borg/testsuite/helpers/yes_no_test.py index 508de36f7..8246f47fe 100644 --- a/src/borg/testsuite/helpers/yes_no_test.py +++ b/src/borg/testsuite/helpers/yes_no_test.py @@ -7,21 +7,21 @@ from .. import FakeInputs def test_yes_input(): inputs = list(TRUISH) input = FakeInputs(inputs) - for i in inputs: + while input.available(): assert yes(input=input) inputs = list(FALSISH) input = FakeInputs(inputs) - for i in inputs: + while input.available(): assert not yes(input=input) def test_yes_input_defaults(): inputs = list(DEFAULTISH) input = FakeInputs(inputs) - for i in inputs: + while input.available(): assert yes(default=True, input=input) input = FakeInputs(inputs) - for i in inputs: + while input.available(): assert not yes(default=False, input=input)