prompt test

This commit is contained in:
Will McGugan 2020-07-23 16:59:48 +01:00
parent 70d43d7c92
commit 6ece06d960

View file

@ -68,3 +68,15 @@ def test_prompt_confirm_yes():
print(repr(output))
assert output == expected
def test_prompt_confirm_default():
INPUT = "foo\nNO\ny"
console = Console(file=io.StringIO())
answer = Confirm.ask(
"continue", console=console, stream=io.StringIO(INPUT), default=True
)
assert answer is True
expected = "continue [y/n] (y): Please enter Y or N\ncontinue [y/n] (y): Please enter Y or N\ncontinue [y/n] (y): "
output = console.file.getvalue()
print(repr(output))
assert output == expected