Add isatty fixture and use monkeypatch

openSUSE · Sep 29, 2016 · 68d8317 · 68d8317
1 parent 0fa65e2
commit 68d8317
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_color.py b/tests/test_color.py
@@ -5,20 +5,22 @@
 from sdsc import printcolor
 
 
-@pytest.mark.parametrize("msg",
-                         ("a", "b\nc")
+@pytest.mark.parametrize("msg",  ("hello",)
 )
 @pytest.mark.parametrize("msgtype",
                          ("error", "debug", None)
 )
-# @pytest.mark.parametrize('isatty', (True, False))
-def test_printcolor(capsys, msg, msgtype):
+@pytest.mark.parametrize('isatty', (True, False))
+def test_printcolor(capsys, monkeypatch, msg, msgtype, isatty):
     """Checks printcolor() function
 
+    :param capsys:
     :param str msg:
     :param str msgtype:
-    :param capsys:
+    :param bool isatty:
     """
+    monkeypatch.setattr('sys.stdout.isatty', lambda: isatty)
+
     printcolor(msg, msgtype)
     if msgtype in ('debug', 'error'):
         assert capsys.readouterr()[0].count(msg) == 0