forked from OSchip/llvm-project
342 lines
16 KiB
Python
342 lines
16 KiB
Python
"""
|
|
Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
See https://llvm.org/LICENSE.txt for license information.
|
|
SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
# System modules
|
|
import curses
|
|
import datetime
|
|
import math
|
|
import sys
|
|
import time
|
|
|
|
# Third-party modules
|
|
|
|
# LLDB modules
|
|
from lldbsuite.test import lldbcurses
|
|
|
|
from . import results_formatter
|
|
from ..event_builder import EventBuilder
|
|
|
|
|
|
class Curses(results_formatter.ResultsFormatter):
|
|
"""Receives live results from tests that are running and reports them to the terminal in a curses GUI"""
|
|
|
|
def __init__(self, out_file, options, file_is_stream):
|
|
# Initialize the parent
|
|
super(Curses, self).__init__(out_file, options, file_is_stream)
|
|
self.using_terminal = True
|
|
self.have_curses = True
|
|
self.initialize_event = None
|
|
self.jobs = [None] * 64
|
|
self.job_tests = [None] * 64
|
|
self.results = list()
|
|
try:
|
|
self.main_window = lldbcurses.intialize_curses()
|
|
self.main_window.add_key_action(
|
|
'\t',
|
|
self.main_window.select_next_first_responder,
|
|
"Switch between views that can respond to keyboard input")
|
|
self.main_window.refresh()
|
|
self.job_panel = None
|
|
self.results_panel = None
|
|
self.status_panel = None
|
|
self.info_panel = None
|
|
self.hide_status_list = list()
|
|
self.start_time = time.time()
|
|
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, test_event):
|
|
if status == EventBuilder.STATUS_SUCCESS:
|
|
return '.'
|
|
elif status == EventBuilder.STATUS_FAILURE:
|
|
return 'F'
|
|
elif status == EventBuilder.STATUS_UNEXPECTED_SUCCESS:
|
|
return '?'
|
|
elif status == EventBuilder.STATUS_EXPECTED_FAILURE:
|
|
return 'X'
|
|
elif status == EventBuilder.STATUS_SKIP:
|
|
return 'S'
|
|
elif status == EventBuilder.STATUS_ERROR:
|
|
if test_event.get("issue_phase", None) == "build":
|
|
# Build failure
|
|
return 'B'
|
|
else:
|
|
return 'E'
|
|
elif status == EventBuilder.STATUS_TIMEOUT:
|
|
return 'T'
|
|
elif status == EventBuilder.STATUS_EXPECTED_TIMEOUT:
|
|
return 't'
|
|
else:
|
|
return status
|
|
|
|
def show_info_panel(self):
|
|
selected_idx = self.results_panel.get_selected_idx()
|
|
if selected_idx >= 0 and selected_idx < len(self.results):
|
|
if self.info_panel is None:
|
|
info_frame = self.results_panel.get_contained_rect(
|
|
top_inset=10, left_inset=10, right_inset=10, height=30)
|
|
self.info_panel = lldbcurses.BoxedPanel(
|
|
info_frame, "Result Details")
|
|
# Add a key action for any key that will hide this panel when
|
|
# any key is pressed
|
|
self.info_panel.add_key_action(-1,
|
|
self.hide_info_panel,
|
|
'Hide the info panel')
|
|
self.info_panel.top()
|
|
else:
|
|
self.info_panel.show()
|
|
|
|
self.main_window.push_first_responder(self.info_panel)
|
|
test_start = self.results[selected_idx][0]
|
|
test_result = self.results[selected_idx][1]
|
|
self.info_panel.set_line(
|
|
0, "File: %s" %
|
|
(test_start['test_filename']))
|
|
self.info_panel.set_line(
|
|
1, "Test: %s.%s" %
|
|
(test_start['test_class'], test_start['test_name']))
|
|
self.info_panel.set_line(
|
|
2, "Time: %s" %
|
|
(test_result['elapsed_time']))
|
|
self.info_panel.set_line(3, "Status: %s" % (test_result['status']))
|
|
|
|
def hide_info_panel(self):
|
|
self.main_window.pop_first_responder(self.info_panel)
|
|
self.info_panel.hide()
|
|
self.main_window.refresh()
|
|
|
|
def toggle_status(self, status):
|
|
if status:
|
|
# Toggle showing and hiding results whose status matches "status"
|
|
# in "Results" window
|
|
if status in self.hide_status_list:
|
|
self.hide_status_list.remove(status)
|
|
else:
|
|
self.hide_status_list.append(status)
|
|
self.update_results()
|
|
|
|
def update_results(self, update=True):
|
|
'''Called after a category of test have been show/hidden to update the results list with
|
|
what the user desires to see.'''
|
|
self.results_panel.clear(update=False)
|
|
for result in self.results:
|
|
test_result = result[1]
|
|
status = test_result['status']
|
|
if status in self.hide_status_list:
|
|
continue
|
|
name = test_result['test_class'] + '.' + test_result['test_name']
|
|
self.results_panel.append_line(
|
|
'%s (%6.2f sec) %s' %
|
|
(self.status_to_short_str(
|
|
status,
|
|
test_result),
|
|
test_result['elapsed_time'],
|
|
name))
|
|
if update:
|
|
self.main_window.refresh()
|
|
|
|
def handle_event(self, test_event):
|
|
with self.lock:
|
|
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:
|
|
check_for_one_key = True
|
|
#print(str(test_event), file=self.events_file)
|
|
event = test_event['event']
|
|
if self.status_panel:
|
|
self.status_panel.update_status(
|
|
'time', str(
|
|
datetime.timedelta(
|
|
seconds=math.floor(
|
|
time.time() - self.start_time))))
|
|
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: %5d ' % (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)
|
|
if 'pid' in test_event:
|
|
line = 'pid: %5d ' % (test_event['pid'])
|
|
else:
|
|
line = ''
|
|
self.job_panel.set_line(worker_index, line)
|
|
name = test_event['test_class'] + \
|
|
'.' + test_event['test_name']
|
|
elapsed_time = test_event[
|
|
'event_time'] - self.job_tests[worker_index]['event_time']
|
|
if status not in self.hide_status_list:
|
|
self.results_panel.append_line(
|
|
'%s (%6.2f sec) %s' %
|
|
(self.status_to_short_str(
|
|
status, test_event), elapsed_time, name))
|
|
self.main_window.refresh()
|
|
# Append the result pairs
|
|
test_event['elapsed_time'] = elapsed_time
|
|
self.results.append(
|
|
[self.job_tests[worker_index], test_event])
|
|
self.job_tests[worker_index] = ''
|
|
elif event == 'job_begin':
|
|
self.jobs[worker_index] = test_event
|
|
if 'pid' in test_event:
|
|
line = 'pid: %5d ' % (test_event['pid'])
|
|
else:
|
|
line = ''
|
|
self.job_panel.set_line(worker_index, line)
|
|
elif event == 'job_end':
|
|
self.jobs[worker_index] = ''
|
|
self.job_panel.set_line(worker_index, '')
|
|
elif event == 'initialize':
|
|
self.initialize_event = test_event
|
|
num_jobs = test_event['worker_count']
|
|
job_frame = self.main_window.get_contained_rect(
|
|
height=num_jobs + 2)
|
|
results_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(
|
|
frame=job_frame, title="Jobs")
|
|
self.results_panel = lldbcurses.BoxedPanel(
|
|
frame=results_frame, title="Results")
|
|
|
|
self.results_panel.add_key_action(
|
|
curses.KEY_UP,
|
|
self.results_panel.select_prev,
|
|
"Select the previous list entry")
|
|
self.results_panel.add_key_action(
|
|
curses.KEY_DOWN, self.results_panel.select_next, "Select the next list entry")
|
|
self.results_panel.add_key_action(
|
|
curses.KEY_HOME,
|
|
self.results_panel.scroll_begin,
|
|
"Scroll to the start of the list")
|
|
self.results_panel.add_key_action(
|
|
curses.KEY_END, self.results_panel.scroll_end, "Scroll to the end of the list")
|
|
self.results_panel.add_key_action(
|
|
curses.KEY_ENTER,
|
|
self.show_info_panel,
|
|
"Display info for the selected result item")
|
|
self.results_panel.add_key_action(
|
|
'.',
|
|
lambda: self.toggle_status(
|
|
EventBuilder.STATUS_SUCCESS),
|
|
"Toggle showing/hiding tests whose status is 'success'")
|
|
self.results_panel.add_key_action(
|
|
'e',
|
|
lambda: self.toggle_status(
|
|
EventBuilder.STATUS_ERROR),
|
|
"Toggle showing/hiding tests whose status is 'error'")
|
|
self.results_panel.add_key_action(
|
|
'f',
|
|
lambda: self.toggle_status(
|
|
EventBuilder.STATUS_FAILURE),
|
|
"Toggle showing/hiding tests whose status is 'failure'")
|
|
self.results_panel.add_key_action('s', lambda: self.toggle_status(
|
|
EventBuilder.STATUS_SKIP), "Toggle showing/hiding tests whose status is 'skip'")
|
|
self.results_panel.add_key_action(
|
|
'x',
|
|
lambda: self.toggle_status(
|
|
EventBuilder.STATUS_EXPECTED_FAILURE),
|
|
"Toggle showing/hiding tests whose status is 'expected_failure'")
|
|
self.results_panel.add_key_action(
|
|
'?',
|
|
lambda: self.toggle_status(
|
|
EventBuilder.STATUS_UNEXPECTED_SUCCESS),
|
|
"Toggle showing/hiding tests whose status is 'unexpected_success'")
|
|
self.status_panel = lldbcurses.StatusPanel(
|
|
frame=status_frame)
|
|
|
|
self.main_window.add_child(self.job_panel)
|
|
self.main_window.add_child(self.results_panel)
|
|
self.main_window.add_child(self.status_panel)
|
|
self.main_window.set_first_responder(
|
|
self.results_panel)
|
|
|
|
self.status_panel.add_status_item(
|
|
name="time",
|
|
title="Elapsed",
|
|
format="%s",
|
|
width=20,
|
|
value="0:00:00",
|
|
update=False)
|
|
self.status_panel.add_status_item(
|
|
name=EventBuilder.STATUS_SUCCESS,
|
|
title="Success",
|
|
format="%u",
|
|
width=20,
|
|
value=0,
|
|
update=False)
|
|
self.status_panel.add_status_item(
|
|
name=EventBuilder.STATUS_FAILURE,
|
|
title="Failure",
|
|
format="%u",
|
|
width=20,
|
|
value=0,
|
|
update=False)
|
|
self.status_panel.add_status_item(
|
|
name=EventBuilder.STATUS_ERROR,
|
|
title="Error",
|
|
format="%u",
|
|
width=20,
|
|
value=0,
|
|
update=False)
|
|
self.status_panel.add_status_item(
|
|
name=EventBuilder.STATUS_SKIP,
|
|
title="Skipped",
|
|
format="%u",
|
|
width=20,
|
|
value=0,
|
|
update=True)
|
|
self.status_panel.add_status_item(
|
|
name=EventBuilder.STATUS_EXPECTED_FAILURE,
|
|
title="Expected Failure",
|
|
format="%u",
|
|
width=30,
|
|
value=0,
|
|
update=False)
|
|
self.status_panel.add_status_item(
|
|
name=EventBuilder.STATUS_UNEXPECTED_SUCCESS,
|
|
title="Unexpected Success",
|
|
format="%u",
|
|
width=30,
|
|
value=0,
|
|
update=False)
|
|
self.main_window.refresh()
|
|
elif event == 'terminate':
|
|
# self.main_window.key_event_loop()
|
|
lldbcurses.terminate_curses()
|
|
check_for_one_key = False
|
|
self.using_terminal = False
|
|
# Check for 1 keypress with no delay
|
|
|
|
# Check for 1 keypress with no delay
|
|
if check_for_one_key:
|
|
self.main_window.key_event_loop(0, 1)
|