Merge pull request #999 from kporangehat/invoke_with_prog_name

Fix #616: allow setting prog_name as extra in CliRunner.invoke
pallets · May 14, 2018 · 467fb8d · 467fb8d
2 parents 43cb986 + 2bc3b2c
commit 467fb8d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/click/testing.py b/click/testing.py
@@ -318,8 +318,12 @@ def invoke(self, cli, args=None, input=None, env=None,
                 args = shlex.split(args)
 
             try:
-                cli.main(args=args or (),
-                         prog_name=self.get_default_prog_name(cli), **extra)
+                prog_name = extra.pop("prog_name")
+            except KeyError:
+                prog_name = self.get_default_prog_name(cli)
+
+            try:
+                cli.main(args=args or (), prog_name=prog_name, **extra)
             except SystemExit as e:
                 exc_info = sys.exc_info()
                 exit_code = e.code

diff --git a/tests/test_testing.py b/tests/test_testing.py
@@ -246,3 +246,14 @@ def cli_args(foo):
     result = runner.invoke(cli_args, args=args)
     assert result.exit_code == 0
     assert result.output == expected_output
+
+
+def test_setting_prog_name_in_extra():
+    @click.command()
+    def cli():
+        click.echo("ok")
+
+    runner = CliRunner()
+    result = runner.invoke(cli, prog_name="foobar")
+    assert not result.exception
+    assert result.output == 'ok\n'