Filter GTEST failed tests from stdin

filter_gtest_failed.py

"""
To use to `./gest_test_binary | python filter_gtest_failed.py`
Only failed tests' output will be printed.

@author Dimitar Tasev
@license MIT
"""
from __future__ import print_function

import sys

test_running = False
test_output = []
print_everything_forever = False
failed_line_end = "\n"


def add_color(line, color):
    closing_bracket_idx = line.rfind("]")
    return color + line[0:closing_bracket_idx+1] + "\033[0m" + line[closing_bracket_idx+1:]


for line in sys.stdin:
    if test_running:
        test_output.append(line)

    if "[----------] Global test environment tear-down" in line or print_everything_forever:
        print_everything_forever = True
        print(line, end="")
        failed_line_end = ""

    elif "[  FAILED  ]" in line:
        # if test failed, print out test output
        test_running = False
        # make the [ FAILED ] red
        test_output[-1] = add_color(test_output[-1], "\033[31m")
        print("".join(test_output), end=failed_line_end)

    elif "[       OK ]" in line:
        # test was OK, if you want to see output from that... don't use this script
        test_running = False
        test_output = []

    elif "[ RUN      ]" in line:
        # test started
        test_running = True

        test_output = [add_color(line, "\033[32m")]