blob: 10a4d56dc45146d7bef22db77bedeb77aac37470 [file] [log] [blame]
Greg Clayton825ab582015-09-22 16:29:15 +00001#!/usr/bin/env python
2
3"""
4 The LLVM Compiler Infrastructure
5
6 This file is distributed under the University of Illinois Open Source
7 License. See LICENSE.TXT for details.
8
9Configuration options for lldbtest.py set by dotest.py during initialization
10"""
11
12import curses
13import lldbcurses
14import sys
15import test_results
16
17class Curses(test_results.ResultsFormatter):
18 """Receives live results from tests that are running and reports them to the terminal in a curses GUI"""
19
20 def __init__(self, out_file, options):
21 # Initialize the parent
22 super(Curses, self).__init__(out_file, options)
23 self.using_terminal = True
24 self.have_curses = True
25 self.initialize_event = None
26 self.jobs = [None] * 64
27 self.job_tests = [None] * 64
28 self.results = list()
29 self.saved_first_responder = None
30 try:
31 self.main_window = lldbcurses.intialize_curses()
Greg Claytond13c4fb2015-09-22 17:18:15 +000032 self.main_window.delegate = self.handle_main_window_key
Greg Clayton825ab582015-09-22 16:29:15 +000033 self.main_window.refresh()
34 self.job_panel = None
35 self.results_panel = None
36 self.status_panel = None
37 self.info_panel = None
Greg Claytond13c4fb2015-09-22 17:18:15 +000038 self.hide_status_list = list()
Greg Clayton825ab582015-09-22 16:29:15 +000039 except:
40 self.have_curses = False
41 lldbcurses.terminate_curses()
42 self.using_terminal = False
43 print "Unexpected error:", sys.exc_info()[0]
44 raise
45
46
47 self.line_dict = dict()
48 #self.events_file = open("/tmp/events.txt", "w")
49 # self.formatters = list()
50 # if tee_results_formatter:
51 # self.formatters.append(tee_results_formatter)
52
53 def status_to_short_str(self, status):
54 if status == 'success':
55 return '.'
56 elif status == 'failure':
57 return 'F'
58 elif status == 'unexpected_success':
59 return '?'
60 elif status == 'expected_failure':
61 return 'X'
62 elif status == 'skip':
63 return 'S'
64 elif status == 'error':
65 return 'E'
66 else:
67 return status
68
Greg Claytond13c4fb2015-09-22 17:18:15 +000069 def handle_main_window_key(self, window, key):
70 if key == ord('\t'):
71 self.main_window.select_next_first_responder()
72
Greg Clayton825ab582015-09-22 16:29:15 +000073 def handle_info_panel_key(self, window, key):
74 window.resign_first_responder(remove_from_parent=True, new_first_responder=self.saved_first_responder)
75 window.hide()
76 self.saved_first_responder = None
77 self.main_window.refresh()
78 return True
79
80 def handle_job_panel_key(self, window, key):
81 return False
82
83 def handle_result_panel_key(self, window, key):
Greg Claytond13c4fb2015-09-22 17:18:15 +000084 toggle_status = None
Greg Clayton825ab582015-09-22 16:29:15 +000085 if key == ord('\r') or key == ord('\n') or key == curses.KEY_ENTER:
86 selected_idx = self.results_panel.get_selected_idx()
87 if selected_idx >= 0 and selected_idx < len(self.results):
88 if self.info_panel is None:
89 info_frame = self.results_panel.get_contained_rect(top_inset=10, left_inset=10, right_inset=10, height=30)
90 self.info_panel = lldbcurses.BoxedPanel(info_frame, "Result Details", delegate=self.handle_info_panel_key)
91 self.info_panel.top()
92 else:
93 self.info_panel.show()
94
Greg Claytond13c4fb2015-09-22 17:18:15 +000095 self.saved_first_responder = self.main_window.first_responder
Greg Clayton825ab582015-09-22 16:29:15 +000096 self.main_window.set_first_responder(self.info_panel)
97 test_start = self.results[selected_idx][0]
98 test_result = self.results[selected_idx][1]
99 self.info_panel.set_line(0, "File: %s" % (test_start['test_filename']))
100 self.info_panel.set_line(1, "Test: %s.%s" % (test_start['test_class'], test_start['test_name']))
101 self.info_panel.set_line(2, "Time: %s" % (test_result['elapsed_time']))
102 self.info_panel.set_line(3, "Status: %s" % (test_result['status']))
103 elif key == curses.KEY_HOME:
104 self.results_panel.scroll_begin()
105 elif key == curses.KEY_END:
106 self.results_panel.scroll_end()
107 elif key == curses.KEY_UP:
108 self.results_panel.select_prev()
109 elif key == curses.KEY_DOWN:
110 self.results_panel.select_next()
Greg Claytond13c4fb2015-09-22 17:18:15 +0000111 elif key == ord('.'):
112 toggle_status = 'success'
113 elif key == ord('e') or key == ord('E'):
114 toggle_status = 'error'
115 elif key == ord('f') or key == ord('F'):
116 toggle_status = 'failure'
117 elif key == ord('s') or key == ord('S'):
118 toggle_status = 'skip'
119 elif key == ord('x') or key == ord('X'):
120 toggle_status = 'expected_failure'
121 elif key == ord('?'):
122 toggle_status = 'unexpected_success'
Greg Clayton825ab582015-09-22 16:29:15 +0000123 else:
124 return False
Greg Clayton825ab582015-09-22 16:29:15 +0000125
Greg Claytond13c4fb2015-09-22 17:18:15 +0000126 if toggle_status:
127 # Toggle showing and hiding results whose status matches "toggle_status" in "Results" window
128 if toggle_status in self.hide_status_list:
129 self.hide_status_list.remove(toggle_status)
130 else:
131 self.hide_status_list.append(toggle_status)
132 self.update_results(update=False) # Will update below
133 self.main_window.refresh()
134
135 def update_results(self, update=True):
136 '''Called after a category of test have been show/hidden to update the results list with
137 what the user desires to see.'''
138 self.results_panel.clear(update=False)
139 for result in self.results:
140 test_result = result[1]
141 status = test_result['status']
142 if status in self.hide_status_list:
143 continue
144 name = test_result['test_class'] + '.' + test_result['test_name']
145 self.results_panel.append_line('%s (%6.2f sec) %s' % (self.status_to_short_str(status), test_result['elapsed_time'], name))
146 if update:
147 self.main_window.refresh()
148
Greg Clayton825ab582015-09-22 16:29:15 +0000149 def handle_event(self, test_event):
150 with self.lock:
151 super(Curses, self).handle_event(test_event)
152 # for formatter in self.formatters:
153 # formatter.process_event(test_event)
154 if self.have_curses:
155 worker_index = -1
156 if 'worker_index' in test_event:
157 worker_index = test_event['worker_index']
158 if 'event' in test_event:
159 check_for_one_key = True
160 #print >>self.events_file, str(test_event)
161 event = test_event['event']
162 if event == 'test_start':
163 name = test_event['test_class'] + '.' + test_event['test_name']
164 self.job_tests[worker_index] = test_event
165 if 'pid' in test_event:
166 line = 'pid: %5d ' % (test_event['pid']) + name
167 else:
168 line = name
169 self.job_panel.set_line(worker_index, line)
170 self.main_window.refresh()
171 elif event == 'test_result':
172 status = test_event['status']
173 self.status_panel.increment_status(status)
174 if 'pid' in test_event:
175 line = 'pid: %5d ' % (test_event['pid'])
176 else:
177 line = ''
178 self.job_panel.set_line(worker_index, line)
179 # if status != 'success' and status != 'skip' and status != 'expect_failure':
180 name = test_event['test_class'] + '.' + test_event['test_name']
181 elapsed_time = test_event['event_time'] - self.job_tests[worker_index]['event_time']
Greg Claytond13c4fb2015-09-22 17:18:15 +0000182 if not status in self.hide_status_list:
183 self.results_panel.append_line('%s (%6.2f sec) %s' % (self.status_to_short_str(status), elapsed_time, name))
Greg Clayton825ab582015-09-22 16:29:15 +0000184 self.main_window.refresh()
185 # Append the result pairs
186 test_event['elapsed_time'] = elapsed_time
187 self.results.append([self.job_tests[worker_index], test_event])
188 self.job_tests[worker_index] = ''
189 elif event == 'job_begin':
190 self.jobs[worker_index] = test_event
191 if 'pid' in test_event:
192 line = 'pid: %5d ' % (test_event['pid'])
193 else:
194 line = ''
195 self.job_panel.set_line(worker_index, line)
196 elif event == 'job_end':
197 self.jobs[worker_index] = ''
198 self.job_panel.set_line(worker_index, '')
199 elif event == 'initialize':
200 self.initialize_event = test_event
201 num_jobs = test_event['worker_count']
202 job_frame = self.main_window.get_contained_rect(height=num_jobs+2)
203 results_frame = self.main_window.get_contained_rect(top_inset=num_jobs+2, bottom_inset=1)
204 status_frame = self.main_window.get_contained_rect(height=1, top_inset=self.main_window.get_size().h-1)
205 self.job_panel = lldbcurses.BoxedPanel(frame=job_frame, title="Jobs", delegate=self.handle_job_panel_key)
206 self.results_panel = lldbcurses.BoxedPanel(frame=results_frame, title="Results", delegate=self.handle_result_panel_key)
207 self.status_panel = lldbcurses.StatusPanel(frame=status_frame)
208
209 self.main_window.add_child(self.job_panel)
210 self.main_window.add_child(self.results_panel)
211 self.main_window.add_child(self.status_panel)
212 self.main_window.set_first_responder(self.results_panel)
213
214 self.status_panel.add_status_item(name="success", title="Success", format="%u", width=20, value=0, update=False)
215 self.status_panel.add_status_item(name="failure", title="Failure", format="%u", width=20, value=0, update=False)
216 self.status_panel.add_status_item(name="error", title="Error", format="%u", width=20, value=0, update=False)
217 self.status_panel.add_status_item(name="skip", title="Skipped", format="%u", width=20, value=0, update=True)
218 self.status_panel.add_status_item(name="expected_failure", title="Expected Failure", format="%u", width=30, value=0, update=False)
219 self.status_panel.add_status_item(name="unexpected_success", title="Unexpected Success", format="%u", width=30, value=0, update=False)
220 self.main_window.refresh()
221 elif event == 'terminate':
222 self.main_window.key_event_loop()
223 lldbcurses.terminate_curses()
224 check_for_one_key = False
225 self.using_terminal = False
226 # Check for 1 keypress with no delay
227
228 # Check for 1 keypress with no delay
229 if check_for_one_key:
230 self.main_window.key_event_loop(0, 1)
231