fix exit_code for click.testing.CliRunner.invoke

pallets · Feb 21, 2017 · 2a9762f · 2a9762f
1 parent 8d9dd46
commit 2a9762f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/click/testing.py b/click/testing.py
@@ -292,9 +292,13 @@ def invoke(self, cli, args=None, input=None, env=None,
 
                 exit_code = e.code
                 if not isinstance(exit_code, int):
-                    sys.stdout.write(str(exit_code))
+                    if exit_code is None:
+                        exit_code = 0
+                    else:
+                        exit_code = 1
+
+                    sys.stdout.write(str(e.code))
                     sys.stdout.write('\n')
-                    exit_code = 1
             except Exception as e:
                 if not catch_exceptions:
                     raise