fixed #228: make exit code customizable to indicated whether files were changed
diff --git a/yapftests/main_test.py b/yapftests/main_test.py
index e5a5a74..163fc88 100644
--- a/yapftests/main_test.py
+++ b/yapftests/main_test.py
@@ -91,7 +91,7 @@
with patched_input(code):
with captured_output() as (out, err):
ret = yapf.main(['-', '--style=chromium'])
- self.assertEqual(ret, 2)
+ self.assertEqual(ret, 0)
self.assertEqual(out.getvalue(), chromium_code)
def testEchoBadInput(self):
@@ -116,3 +116,25 @@
self.assertEqual(ret, 0)
version = 'yapf {}\n'.format(yapf.__version__)
self.assertEqual(version, out.getvalue())
+
+ def testUnchangedFileExitCode(self):
+ code = "a = 1"
+ with patched_input(code):
+ with captured_output() as (out, err):
+ ret = yapf.main([])
+ self.assertEqual(ret, 0)
+
+ def testChangedFileExitCode(self):
+ code = "a=1"
+ with patched_input(code):
+ with captured_output() as (out, err):
+ ret = yapf.main([])
+ self.assertEqual(ret, 0)
+
+ def testCustomChangedFileExitCode(self):
+ code = "a=1"
+ with patched_input(code):
+ with captured_output() as (out, err):
+ ret = yapf.main(['-', '-c', '2'])
+ self.assertEqual(ret, 2)
+
