Added tests for prompts

1 files changed, 22 insertions, 0 deletions

diff --git a/tests/test_testing.py b/tests/test_testing.py
index 3e11f61..933b426 100644
--- a/tests/test_testing.py
+++ b/tests/test_testing.py
@@ -55,3 +55,25 @@ def test_runner_with_stream():
     result = runner.invoke(test, input=ReasonableBytesIO(b'Hello World!\n'))
     assert not result.exception
     assert result.output == 'Hello World!\nHello World!\n'
+
+
+def test_prompts():
+    @click.command()
+    @click.option('--foo', prompt=True)
+    def test(foo):
+        click.echo('foo=%s' % foo)
+
+    runner = CliRunner()
+    result = runner.invoke(test, input='wau wau')
+    assert not result.exception
+    assert result.output == 'Foo: wau wau\nfoo=wau wau\n'
+
+    @click.command()
+    @click.option('--foo', prompt=True, hide_input=True)
+    def test(foo):
+        click.echo('foo=%s' % foo)
+
+    runner = CliRunner()
+    result = runner.invoke(test, input='wau wau')
+    assert not result.exception
+    assert result.output == 'Foo: \nfoo=wau wau\n'