mirror of https://github.com/Textualize/rich.git
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import io
|
|
|
|
from rich.console import Console
|
|
from rich.prompt import Prompt, IntPrompt, Confirm
|
|
|
|
|
|
def test_prompt_str():
|
|
INPUT = "egg\nfoo"
|
|
console = Console(file=io.StringIO())
|
|
name = Prompt.ask(
|
|
"what is your name",
|
|
console=console,
|
|
choices=["foo", "bar"],
|
|
default="baz",
|
|
stream=io.StringIO(INPUT),
|
|
)
|
|
assert name == "foo"
|
|
expected = "what is your name [foo/bar] (baz): Please select one of the available options\nwhat is your name [foo/bar] (baz): "
|
|
output = console.file.getvalue()
|
|
print(repr(output))
|
|
assert output == expected
|
|
|
|
|
|
def test_prompt_str_default():
|
|
INPUT = ""
|
|
console = Console(file=io.StringIO())
|
|
name = Prompt.ask(
|
|
"what is your name", console=console, default="Will", stream=io.StringIO(INPUT),
|
|
)
|
|
assert name == "Will"
|
|
expected = "what is your name (Will): "
|
|
output = console.file.getvalue()
|
|
print(repr(output))
|
|
assert output == expected
|
|
|
|
|
|
def test_prompt_int():
|
|
INPUT = "foo\n100"
|
|
console = Console(file=io.StringIO())
|
|
number = IntPrompt.ask(
|
|
"Enter a number", console=console, stream=io.StringIO(INPUT),
|
|
)
|
|
assert number == 100
|
|
expected = "Enter a number: Please enter a valid integer number\nEnter a number: "
|
|
output = console.file.getvalue()
|
|
print(repr(output))
|
|
assert output == expected
|
|
|
|
|
|
def test_prompt_confirm_no():
|
|
INPUT = "foo\nNO\nn"
|
|
console = Console(file=io.StringIO())
|
|
answer = Confirm.ask("continue", console=console, stream=io.StringIO(INPUT),)
|
|
assert answer is False
|
|
expected = "continue [y/N]: Please enter Y or N\ncontinue [y/N]: Please enter Y or N\ncontinue [y/N]: "
|
|
output = console.file.getvalue()
|
|
print(repr(output))
|
|
assert output == expected
|
|
|
|
|
|
def test_prompt_confirm_yes():
|
|
INPUT = "foo\nNO\ny"
|
|
console = Console(file=io.StringIO())
|
|
answer = Confirm.ask("continue", console=console, stream=io.StringIO(INPUT),)
|
|
assert answer is True
|
|
expected = "continue [y/N]: Please enter Y or N\ncontinue [y/N]: Please enter Y or N\ncontinue [y/N]: "
|
|
output = console.file.getvalue()
|
|
print(repr(output))
|
|
assert output == expected
|
|
|