Added support for multiple defaults on options and nargs

1 files changed, 46 insertions, 0 deletions

diff --git a/tests/test_defaults.py b/tests/test_defaults.py
new file mode 100644
index 0000000..dba5a9c
--- /dev/null
+++ b/tests/test_defaults.py
@@ -0,0 +1,46 @@
+import click
+
+
+def test_basic_defaults(runner):
+    @click.command()
+    @click.option('--foo', default=42, type=click.FLOAT)
+    def cli(foo):
+        assert type(foo) is float
+        click.echo('FOO:[%s]' % foo)
+
+    result = runner.invoke(cli, [])
+    assert not result.exception
+    assert 'FOO:[42.0]' in result.output
+
+
+def test_multiple_defaults(runner):
+    @click.command()
+    @click.option('--foo', default=[23, 42], type=click.FLOAT,
+                  multiple=True)
+    def cli(foo):
+        for item in foo:
+            assert type(item) is float
+            click.echo(item)
+
+    result = runner.invoke(cli, [])
+    assert not result.exception
+    assert result.output.splitlines() == [
+        '23.0',
+        '42.0',
+    ]
+
+
+def test_nargs_plus_multiple(runner):
+    @click.command()
+    @click.option('--arg', default=((1, 2), (3, 4)),
+                  nargs=2, multiple=True, type=click.INT)
+    def cli(arg):
+        for item in arg:
+            click.echo('<%d|%d>' % item)
+
+    result = runner.invoke(cli, [])
+    assert not result.exception
+    assert result.output.splitlines() == [
+        '<1|2>',
+        '<3|4>',
+    ]