Add mute decorator for tests where output unneeded

Author: tony

tmuxp/testsuite/helpers.py

@@ -13,6 +13,7 @@
 import inspect
 import io
 import logging
+import os
 import sys
 from contextlib import contextmanager
 from random import randint
@@ -336,3 +337,30 @@ def test_foo(self, stdout, stderr):
         sys.stderr = prev_err
         sys.__stdout__ = prev_rout
         sys.__stderr__ = prev_rerr
+
+
+@decorator
+def mute():
+    """Redirect `sys.stdout` and `sys.stderr` to /dev/null, silencent them.
+    Decorator example::
+        @mute
+        def test_foo(self):
+            something()
+    Context example::
+        with mute():
+            something()
+    """
+    prev_out, prev_err = sys.stdout, sys.stderr
+    prev_rout, prev_rerr = sys.__stdout__, sys.__stderr__
+    devnull = open(os.devnull, 'w')
+    mystdout, mystderr = devnull, devnull
+    sys.stdout = sys.__stdout__ = mystdout
+    sys.stderr = sys.__stderr__ = mystderr
+
+    try:
+        yield
+    finally:
+        sys.stdout = prev_out
+        sys.stderr = prev_err
+        sys.__stdout__ = prev_rout
+        sys.__stderr__ = prev_rerr

tmuxp/testsuite/util.py

@@ -113,7 +113,7 @@ def test_raise_BeforeLoadScriptError_if_retcode(self):
         with self.assertRaises(BeforeLoadScriptError):
             run_before_script(script_file)
 
-    @stdouts()
+    @stdouts
     def test_return_stdout_if_ok(self, stdout, stderr):
         script_file = os.path.join(fixtures_dir, 'script_complete.sh')
 

tmuxp/testsuite/workspacebuilder.py

@@ -19,7 +19,7 @@
 
 from tmuxp import Window, config, exc
 from tmuxp._compat import text_type
-from tmuxp.testsuite.helpers import TmuxTestCase, stdouts
+from tmuxp.testsuite.helpers import TmuxTestCase, mute
 from tmuxp.workspacebuilder import WorkspaceBuilder
 
 logger = logging.getLogger(__name__)
@@ -883,8 +883,8 @@ class BeforeLoadScript(TmuxTestCase):
       - pane
     """
 
-    @stdouts()
-    def test_throw_error_if_retcode_error(self, stdout, stderr):
+    @mute()
+    def test_throw_error_if_retcode_error(self):
         sconfig = kaptan.Kaptan(handler='yaml')
         yaml = self.config_script_fails.format(
             fixtures_dir=fixtures_dir,
@@ -909,8 +909,8 @@ def test_throw_error_if_retcode_error(self, stdout, stderr):
                 msg="Kills session if before_script exits with errcode"
             )
 
-    @stdouts()
-    def test_throw_error_if_file_not_exists(self, stdout, stderr):
+    @mute()
+    def test_throw_error_if_file_not_exists(self):
         sconfig = kaptan.Kaptan(handler='yaml')
         yaml = self.config_script_not_exists.format(
             fixtures_dir=fixtures_dir,
@@ -941,8 +941,8 @@ def test_throw_error_if_file_not_exists(self, stdout, stderr):
                 msg="Kills session if before_script doesn't exist"
             )
 
-    @stdouts()
-    def test_true_if_test_passes(self, stdout, stderr):
+    @mute()
+    def test_true_if_test_passes(self):
         assert(
             os.path.exists(os.path.join(fixtures_dir, 'script_complete.sh'))
         )
@@ -961,8 +961,8 @@ def test_true_if_test_passes(self, stdout, stderr):
         with self.temp_session():
             builder.build(session=self.session)
 
-    @stdouts()
-    def test_true_if_test_passes_with_args(self, *args):
+    @mute()
+    def test_true_if_test_passes_with_args(self):
         assert(
             os.path.exists(os.path.join(fixtures_dir, 'script_complete.sh'))
         )