revert logfile redirection and make colors work by temporarily switching stdout

This commit is contained in:
Vishvananda Ishaya
2011-02-23 15:29:02 -08:00
parent d0d61174f0
commit 8430f45c3b
2 changed files with 9 additions and 8 deletions

View File

@@ -41,4 +41,3 @@ FLAGS.iscsi_num_targets = 8
FLAGS.verbose = True
FLAGS.sql_connection = 'sqlite:///tests.sqlite'
FLAGS.use_ipv6 = True
FLAGS.logfile = 'tests.log'

View File

@@ -97,7 +97,7 @@ class _AnsiColorizer(object):
try:
return curses.tigetnum("colors") > 2
except curses.error:
curses.setupterm(fd=stream.fileno())
curses.setupterm()
return curses.tigetnum("colors") > 2
except:
raise
@@ -122,13 +122,13 @@ class _Win32Colorizer(object):
See _AnsiColorizer docstring.
"""
def __init__(self, stream):
from win32console import GetStdHandle, STD_ERROR_HANDLE, \
from win32console import GetStdHandle, STD_OUT_HANDLE, \
FOREGROUND_RED, FOREGROUND_BLUE, FOREGROUND_GREEN, \
FOREGROUND_INTENSITY
red, green, blue, bold = (FOREGROUND_RED, FOREGROUND_GREEN,
FOREGROUND_BLUE, FOREGROUND_INTENSITY)
self.stream = stream
self.screenBuffer = GetStdHandle(STD_ERROR_HANDLE)
self.screenBuffer = GetStdHandle(STD_OUT_HANDLE)
self._colors = {
'normal': red | green | blue,
'red': red | bold,
@@ -144,7 +144,7 @@ class _Win32Colorizer(object):
try:
import win32console
screenBuffer = win32console.GetStdHandle(
win32console.STD_ERROR_HANDLE)
win32console.STD_OUT_HANDLE)
except ImportError:
return False
import pywintypes
@@ -185,12 +185,14 @@ class NovaTestResult(result.TextTestResult):
result.TextTestResult.__init__(self, *args, **kw)
self._last_case = None
self.colorizer = None
# NOTE(vish): reset stdout for the terminal check
stdout = sys.stdout
sys.stdout = sys.__stdout__
for colorizer in [_Win32Colorizer, _AnsiColorizer, _NullColorizer]:
# NOTE(vish): nose does funky stuff with stdout, so use stderr
# to setup the colorizer
if colorizer.supported(sys.stderr):
if colorizer.supported():
self.colorizer = colorizer(self.stream)
break
sys.stdout = stdout
def getDescription(self, test):
return str(test)