From ddaa94003bca853819855014b7d99c165c890de8 Mon Sep 17 00:00:00 2001
From: Johnny Robeson <johnny@localmomentum.net>
Date: Sun, 18 Sep 2016 03:17:44 -0400
Subject: [PATCH] add yaml output command func to config_command tests

---
 test/test_config_command.py | 31 ++++++++++++++++---------------
 1 file changed, 16 insertions(+), 15 deletions(-)

diff --git a/test/test_config_command.py b/test/test_config_command.py
index b0240574c..3f0da5f18 100644
--- a/test/test_config_command.py
+++ b/test/test_config_command.py
@@ -43,40 +43,41 @@ class ConfigCommandTest(unittest.TestCase, TestHelper):
     def tearDown(self):
         rmtree(self.temp_dir)
 
-    def test_show_user_config(self):
+    def _run_with_yaml_output(self, *args):
         with capture_stdout() as output:
-            self.run_command('config', '-c')
-        output = yaml.load(output.getvalue())
+            self.run_command(*args)
+        return yaml.load(output.getvalue())
+
+    def test_show_user_config(self):
+        output = self._run_with_yaml_output('config', '-c')
+
         self.assertEqual(output['option'], 'value')
         self.assertEqual(output['password'], 'password_value')
 
     def test_show_user_config_with_defaults(self):
-        with capture_stdout() as output:
-            self.run_command('config', '-dc')
-        output = yaml.load(output.getvalue())
+        output = self._run_with_yaml_output('config', '-dc')
+
         self.assertEqual(output['option'], 'value')
         self.assertEqual(output['password'], 'password_value')
         self.assertEqual(output['library'], 'lib')
         self.assertEqual(output['import']['timid'], False)
 
     def test_show_user_config_with_cli(self):
-        with capture_stdout() as output:
-            self.run_command('--config', self.cli_config_path, 'config')
-        output = yaml.load(output.getvalue())
+        output = self._run_with_yaml_output('--config', self.cli_config_path,
+                                            'config')
+
         self.assertEqual(output['library'], 'lib')
         self.assertEqual(output['option'], 'cli overwrite')
 
     def test_show_redacted_user_config(self):
-        with capture_stdout() as output:
-            self.run_command('config')
-        output = yaml.load(output.getvalue())
+        output = self._run_with_yaml_output('config')
+
         self.assertEqual(output['option'], 'value')
         self.assertEqual(output['password'], 'REDACTED')
 
     def test_show_redacted_user_config_with_defaults(self):
-        with capture_stdout() as output:
-            self.run_command('config', '-d')
-        output = yaml.load(output.getvalue())
+        output = self._run_with_yaml_output('config', '-d')
+
         self.assertEqual(output['option'], 'value')
         self.assertEqual(output['password'], 'REDACTED')
         self.assertEqual(output['import']['timid'], False)