gtest: modified Xcode integration so gtest assert/expect failure now shows error content in the error marker at the line of failure.

This change modifies the python test runner to combine lines starting with a
failure report including up to 3 more lines, terminated by the next
"[  FAILED  ]" line.  These are all emitted on the same line as the file:line
indication, which allows Xcode's failure marker code to pick it up and display
it along with the error badge in the Xcode editor window.  Makes for a nice
gtest development experience.

llvm-svn: 218518

1 files changed, 46 insertions, 10 deletions

diff --git a/lldb/gtest/do-gtest.py b/lldb/gtest/do-gtest.py
index e3bb0fcd9ba..f2e81912b76 100755
--- a/lldb/gtest/do-gtest.py
+++ b/lldb/gtest/do-gtest.py
@@ -17,10 +17,43 @@ def find_makefile_dirs():
     return makefile_dirs
 
 _TESTDIR_RELATIVE_REGEX = re.compile(r"^([^/:]+:\d+:)")
+_COMBINER_TERMINATION_REGEX = re.compile(r"^.+FAILED.+$")
 
 def filter_run_line(sub_expr, line):
-    return _TESTDIR_RELATIVE_REGEX.sub(sub_expr, line)
- 
+    return _TESTDIR_RELATIVE_REGEX.subn(sub_expr, line)
+
+def line_combine_printer(file, previous_data, new_line_subn_result):
+    (accumulated_line, combine_lines_left) = previous_data
+    (incoming_line, sub_match_count) = new_line_subn_result
+
+    if sub_match_count > 0:
+        # New line was a match.  Don't print yet, start an accumulation.
+        if len(accumulated_line) > 0:
+            # Flush anything previously there.
+            print(accumulated_line, file=file)
+        return (incoming_line + ": ", 3)
+    else:
+        # If we're combining and incoming is a "[  FAILED ]" line, we've gone too far on a combine.
+        if (len(accumulated_line) > 0) and _COMBINER_TERMINATION_REGEX.match(incoming_line):
+            # Stop the combine.
+            print(accumulated_line, file=file)
+            print(incoming_line, file=file)
+            return ("", 0)
+
+        if len(accumulated_line) > 0:
+            new_line = accumulated_line + ", " + incoming_line
+        else:
+            new_line = incoming_line
+
+        remaining_count = combine_lines_left - 1
+        if remaining_count > 0:
+            return (new_line, remaining_count)
+        else:
+            # Time to write it out.
+            if len(new_line) > 0:
+                print(new_line, file=file)
+            return ("", 0)
+
 def call_make(makefile_dir, extra_args=None):
     command = ["make", "-C", makefile_dir]
     if extra_args:
@@ -31,6 +64,9 @@ def call_make(makefile_dir, extra_args=None):
 
     proc = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 
+    stdout_data = ("", 0)
+    stderr_data = ("", 0)
+
     while True:
         reads = [proc.stdout.fileno(), proc.stderr.fileno()]
         select_result = select.select(reads, [], [])
@@ -38,10 +74,10 @@ def call_make(makefile_dir, extra_args=None):
         for fd in select_result[0]:
             if fd == proc.stdout.fileno():
                 line = proc.stdout.readline()
-                print(filter_run_line(sub_expr, line.rstrip()))
+                stdout_data = line_combine_printer(sys.stdout, stdout_data, filter_run_line(sub_expr, line.rstrip()))
             elif fd == proc.stderr.fileno():
                 line = proc.stderr.readline()
-                print(filter_run_line(sub_expr, line.rstrip()), file=sys.stderr)
+                stderr_data = line_combine_printer(sys.stderr, stderr_data, filter_run_line(sub_expr, line.rstrip()))
 
         proc_retval = proc.poll()
         if proc_retval != None:
@@ -49,17 +85,17 @@ def call_make(makefile_dir, extra_args=None):
 
             # Drain stdout.
             while True:
-                line = proc.stdout.readline()
-                if line:
-                    print(filter_run_line(sub_expr, line.rstrip()))
+                line = proc.stdout.readline().rstrip()
+                if line and len(line) > 0:
+                    stdout_data = line_combine_printer(sys.stdout, stdout_data, filter_run_line(sub_expr, line))
                 else:
                     break
 
             # Drain stderr.
             while True:
-                line = proc.stderr.readline()
-                if line:
-                    print(filter_run_line(sub_expr, line.rstrip()), file=sys.stderr)
+                line = proc.stderr.readline().rstrip()
+                if line and len(line) > 0:
+                    stderr_data = line_combine_printer(sys.stderr, stderr_data, filter_run_line(sub_expr, line))
                 else:
                     break