From 2a9762f5809f2f270fc37af2e5422995affa1b4a Mon Sep 17 00:00:00 2001
From: timfeirg <kkcocogogo@gmail.com>
Date: Tue, 21 Feb 2017 20:36:36 +0800
Subject: [PATCH] fix exit_code for click.testing.CliRunner.invoke

---
 click/testing.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/click/testing.py b/click/testing.py
index 09d37a72e..368fe492c 100644
--- 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