summaryrefslogtreecommitdiffstats
path: root/lldb/test/curses_results.py
blob: 10a4d56dc45146d7bef22db77bedeb77aac37470 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
#!/usr/bin/env python

"""
                     The LLVM Compiler Infrastructure

 This file is distributed under the University of Illinois Open Source
 License. See LICENSE.TXT for details.

Configuration options for lldbtest.py set by dotest.py during initialization
"""

import curses
import lldbcurses
import sys
import test_results

class Curses(test_results.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):
        # 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
        self.results = list()
        self.saved_first_responder = None
        try:
            self.main_window = lldbcurses.intialize_curses()
            self.main_window.delegate = self.handle_main_window_key
            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()
        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_main_window_key(self, window, key):
        if key == ord('\t'):
            self.main_window.select_next_first_responder()

    def handle_info_panel_key(self, window, key):
        window.resign_first_responder(remove_from_parent=True, new_first_responder=self.saved_first_responder)
        window.hide()        
        self.saved_first_responder = None
        self.main_window.refresh()
        return True

    def handle_job_panel_key(self, window, key):
        return False

    def handle_result_panel_key(self, window, key):
        toggle_status = None
        if key == ord('\r') or key == ord('\n') or key == curses.KEY_ENTER:
            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", delegate=self.handle_info_panel_key)
                    self.info_panel.top()
                else:
                    self.info_panel.show()
                
                self.saved_first_responder = self.main_window.first_responder  
                self.main_window.set_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']))
        elif key == curses.KEY_HOME:
            self.results_panel.scroll_begin()
        elif key == curses.KEY_END:
            self.results_panel.scroll_end()
        elif key == curses.KEY_UP:
            self.results_panel.select_prev()
        elif key == curses.KEY_DOWN:
            self.results_panel.select_next()
        elif key == ord('.'):                                        
            toggle_status = 'success'
        elif key == ord('e') or key == ord('E'):
            toggle_status = 'error'
        elif key == ord('f') or key == ord('F'):
            toggle_status = 'failure'
        elif key == ord('s') or key == ord('S'):
            toggle_status = 'skip'
        elif key == ord('x') or key == ord('X'):
            toggle_status = 'expected_failure'
        elif key == ord('?'):
            toggle_status = 'unexpected_success'
        else:
            return False
        
        if toggle_status:
            # Toggle showing and hiding results whose status matches "toggle_status" in "Results" window
            if toggle_status in self.hide_status_list:
                self.hide_status_list.remove(toggle_status)
            else:
                self.hide_status_list.append(toggle_status)   
            self.update_results(update=False) # Will update below
        self.main_window.refresh()

    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['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 >>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: %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)
                        # if status != 'success' and status != 'skip' and status != 'expect_failure':
                        name = test_event['test_class'] + '.' + test_event['test_name']
                        elapsed_time = test_event['event_time'] - self.job_tests[worker_index]['event_time']
                        if not status in self.hide_status_list:
                            self.results_panel.append_line('%s (%6.2f sec) %s' % (self.status_to_short_str(status), 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", delegate=self.handle_job_panel_key)
                        self.results_panel = lldbcurses.BoxedPanel(frame=results_frame, title="Results", delegate=self.handle_result_panel_key)
                        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="success", title="Success", format="%u", width=20, value=0, update=False)
                        self.status_panel.add_status_item(name="failure", title="Failure", format="%u", width=20, value=0, update=False)
                        self.status_panel.add_status_item(name="error", title="Error", format="%u", width=20, value=0, update=False)
                        self.status_panel.add_status_item(name="skip", title="Skipped", format="%u", width=20, value=0, update=True)
                        self.status_panel.add_status_item(name="expected_failure", title="Expected Failure", format="%u", width=30, value=0, update=False)
                        self.status_panel.add_status_item(name="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) 
                        
OpenPOWER on IntegriCloud