diff options
Diffstat (limited to 'lldb/test/test_results.py')
| -rw-r--r-- | lldb/test/test_results.py | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/lldb/test/test_results.py b/lldb/test/test_results.py index 4ed0d53cdb7..441a4bf4aee 100644 --- a/lldb/test/test_results.py +++ b/lldb/test/test_results.py @@ -382,6 +382,8 @@ class ResultsFormatter(object): super(ResultsFormatter, self).__init__() self.out_file = out_file self.options = options + self.using_terminal = False + self.lock = None # used when coordinating output is needed if not self.out_file: raise Exception("ResultsFormatter created with no file object") self.start_time_by_test = {} @@ -392,6 +394,9 @@ class ResultsFormatter(object): # entirely consistent from the outside. self.lock = threading.Lock() + def set_lock(self, lock): + self.lock = lock + def handle_event(self, test_event): """Handles the test event for collection into the formatter output. @@ -436,6 +441,9 @@ class ResultsFormatter(object): del self.start_time_by_test[test_key] return end_time - start_time + def is_using_terminal(self): + """Returns True if this results formatter is using the terminal and output should be avoided""" + return self.using_terminal class XunitFormatter(ResultsFormatter): """Provides xUnit-style formatted output. @@ -881,6 +889,114 @@ class RawPickledFormatter(ResultsFormatter): self.out_file.send( "{}#{}".format(len(pickled_message), pickled_message)) +class Curses(ResultsFormatter): + """Receives live results from tests that are running and reports them to the terminal in a curses GUI""" + + def clear_line(self, y): + self.out_file.write("\033[%u;0H\033[2K" % (y)) + self.out_file.flush() + + def print_line(self, y, str): + self.out_file.write("\033[%u;0H\033[2K%s" % (y, str)) + self.out_file.flush() + + def __init__(self, out_file, options): + # Initialize the parent + super(Curses, self).__init__(out_file, options) + self.using_terminal = True + self.have_curses = True + self.initialize_event = None + self.jobs = [None] * 64 + self.job_tests = [None] * 64 + try: + import lldbcurses + self.main_window = lldbcurses.intialize_curses() + num_jobs = 8 # TODO: need to dynamically determine this + job_frame = self.main_window.get_contained_rect(height=num_jobs+2) + fail_frame = self.main_window.get_contained_rect(top_inset=num_jobs+2, bottom_inset=1) + status_frame = self.main_window.get_contained_rect(height=1, top_inset=self.main_window.get_size().h-1) + self.job_panel = lldbcurses.BoxedPanel(job_frame, "Jobs") + self.fail_panel = lldbcurses.BoxedPanel(fail_frame, "Failures") + self.status_panel = lldbcurses.StatusPanel(status_frame) + self.status_panel.add_status_item(name="success", title="Success (%s)" % self.status_to_short_str('success'), format="%u", width=20, value=0, update=False) + self.status_panel.add_status_item(name="failure", title="Failure (%s)" % self.status_to_short_str('failure'), format="%u", width=20, value=0, update=False) + self.status_panel.add_status_item(name="error", title="Error (%s)" % self.status_to_short_str('error'), format="%u", width=20, value=0, update=False) + self.status_panel.add_status_item(name="skip", title="Skipped (%s)" % self.status_to_short_str('skip'), format="%u", width=20, value=0, update=True) + self.status_panel.add_status_item(name="expected_failure", title="Expected Failure (%s)" % self.status_to_short_str('expected_failure'), format="%u", width=30, value=0, update=False) + self.status_panel.add_status_item(name="unexpected_success", title="Unexpected Success (%s)" % self.status_to_short_str('unexpected_success'), format="%u", width=30, value=0, update=False) + self.main_window.refresh() + except: + self.have_curses = False + lldbcurses.terminate_curses() + self.using_terminal = False + print "Unexpected error:", sys.exc_info()[0] + raise + + + self.line_dict = dict() + self.events_file = open("/tmp/events.txt", "w") + # self.formatters = list() + # if tee_results_formatter: + # self.formatters.append(tee_results_formatter) + + def status_to_short_str(self, status): + if status == 'success': + return '.' + elif status == 'failure': + return 'F' + elif status == 'unexpected_success': + return '?' + elif status == 'expected_failure': + return 'X' + elif status == 'skip': + return 'S' + elif status == 'error': + return 'E' + else: + return status + def handle_event(self, test_event): + if self.lock: + self.lock.acquire() + super(Curses, self).handle_event(test_event) + # for formatter in self.formatters: + # formatter.process_event(test_event) + if self.have_curses: + worker_index = -1 + if 'worker_index' in test_event: + worker_index = test_event['worker_index'] + if 'event' in test_event: + print >>self.events_file, str(test_event) + event = test_event['event'] + if event == 'test_start': + name = test_event['test_class'] + '.' + test_event['test_name'] + self.job_tests[worker_index] = test_event + if 'pid' in test_event: + line = 'pid: ' + str(test_event['pid']) + ' ' + name + else: + line = name + self.job_panel.set_line(worker_index, line) + self.main_window.refresh() + elif event == 'test_result': + status = test_event['status'] + self.status_panel.increment_status(status) + self.job_panel.set_line(worker_index, '') + # if status != 'success' and status != 'skip' and status != 'expect_failure': + name = test_event['test_class'] + '.' + test_event['test_name'] + time = test_event['event_time'] - self.job_tests[worker_index]['event_time'] + self.fail_panel.append_line('%s (%6.2f sec) %s' % (self.status_to_short_str(status), time, name)) + self.main_window.refresh() + self.job_tests[worker_index] = '' + elif event == 'job_begin': + self.jobs[worker_index] = test_event + elif event == 'job_end': + self.jobs[worker_index] = '' + elif event == 'initialize': + self.initialize_event = test_event + num_jobs = test_event['worker_count'] + self.main_window.refresh() + + if self.lock: + self.lock.release() class DumpFormatter(ResultsFormatter): """Formats events to the file as their raw python dictionary format.""" |
