tests.main(): disable colors when appropriate

diff --git a/pym/portage/tests/__init__.py b/pym/portage/tests/__init__.py
index ee1e5639ccf2d0dcd6caf464ec6808f81b64c7d3..a46c2db42587b59ee6f21be214347c1aceb6f613 100644 (file)
--- a/pym/portage/tests/__init__.py
+++ b/pym/portage/tests/__init__.py
@@ -14,6 +14,7 @@ try:
 except ImportError:
        from unittest import _TextTestResult
 
+import portage
 from portage import os
 from portage import _encodings
 from portage import _unicode_decode
@@ -28,6 +29,11 @@ def main():
                action="store_true", dest="list_tests")
        (options, args) = parser.parse_args(args=sys.argv)
 
+       if (os.environ.get('NOCOLOR') in ('yes', 'true') or
+               os.environ.get('TERM') == 'dumb' or
+               not sys.stdout.isatty()):
+               portage.output.nocolor()
+
        if options.list_tests:
                testdir = os.path.dirname(sys.argv[0])
                for mydir in getTestDirs(basedir):