[lit] Change progress bar color to red on first failure

2 files changed, 8 insertions, 3 deletions

diff --git a/llvm/utils/lit/lit/ProgressBar.py b/llvm/utils/lit/lit/ProgressBar.py
index cfe6d70ff2f..57b354efa35 100644
--- a/llvm/utils/lit/lit/ProgressBar.py
+++ b/llvm/utils/lit/lit/ProgressBar.py
@@ -218,7 +218,7 @@ class ProgressBar:
     The progress bar is colored, if the terminal supports color
     output; and adjusts to the width of the terminal.
     """
-    BAR = '%s${GREEN}[${BOLD}%s%s${NORMAL}${GREEN}]${NORMAL}%s'
+    BAR = '%s${%s}[${BOLD}%s%s${NORMAL}${%s}]${NORMAL}%s'
     HEADER = '${BOLD}${CYAN}%s${NORMAL}\n\n'
         
     def __init__(self, term, header, useETA=True):
@@ -235,7 +235,7 @@ class ProgressBar:
                 self.XNL = "" # Cursor must be fed to the next line
         else:
             self.width = 75
-        self.bar = term.render(self.BAR)
+        self.barColor = 'GREEN'
         self.header = self.term.render(self.HEADER % header.center(self.width))
         self.cleared = 1 #: true if we haven't drawn the bar yet.
         self.useETA = useETA
@@ -264,9 +264,12 @@ class ProgressBar:
             message = message + ' '*(self.width - len(message))
         else:
             message = '... ' + message[-(self.width-4):]
+        bc = self.barColor
+        bar = self.BAR % (prefix, bc, '='*n, '-'*(barWidth-n), bc, suffix)
+        bar = self.term.render(bar)
         sys.stdout.write(
             self.BOL + self.term.UP + self.term.CLEAR_EOL +
-            (self.bar % (prefix, '='*n, '-'*(barWidth-n), suffix)) +
+            bar +
             self.XNL +
             self.term.CLEAR_EOL + message)
         if not self.term.XN:
diff --git a/llvm/utils/lit/lit/display.py b/llvm/utils/lit/lit/display.py
index 8a3857c9c32..5d8bee130fe 100644
--- a/llvm/utils/lit/lit/display.py
+++ b/llvm/utils/lit/lit/display.py
@@ -52,6 +52,8 @@ class ProgressDisplay(object):
             self.print_result(test)
 
         if self.progressBar:
+            if test.isFailure():
+                self.progressBar.barColor = 'RED'
             percent = float(self.completed) / self.numTests
             self.progressBar.update(percent, test.getFullName())