blob: 87784ac4cafe7e3aa907736427e0465df880d677 [file] [log] [blame]
Michael Walsh3ba8ecd2018-04-24 11:33:25 -05001#!/usr/bin/env python
2
3r"""
4Check for stop conditions. Return code of 2 if stop conditions are found.
5"""
6
7import sys
8import subprocess
9
10save_path_0 = sys.path[0]
11del sys.path[0]
12
13from gen_print import *
14from gen_valid import *
15from gen_arg import *
16from gen_misc import *
17from gen_cmd import *
18from gen_plug_in_utils import *
19from gen_call_robot import *
20
21# Restore sys.path[0].
22sys.path.insert(0, save_path_0)
23
24# Initialize.
25STOP_REST_FAIL = ''
26STOP_COMMAND = ''
27stop_test_rc = 2
28
29# Create parser object to process command line parameters and args.
30
31# Create parser object.
32parser = argparse.ArgumentParser(
33 usage='%(prog)s [OPTIONS]',
34 description="If the \"Stop\" plug-in is selected by the user, %(prog)s" +
35 " is called by OBMC Boot Test after each boot test. If %(prog)s returns" +
36 " " + str(stop_test_rc) + ", then OBMC Boot Test will stop. The user" +
37 " may set environment variable STOP_COMMAND to contain any valid bash" +
38 " command or program. %(prog)s will run this stop command. If the stop" +
39 " command returns non-zero, then %(prog)s will return " +
40 str(stop_test_rc) + ". %(prog)s recognizes some special values for" +
41 " STOP_COMMAND: 1) \"FAIL\" means that OBMC Boot Test should stop" +
42 " whenever a boot test fails. 2) \"ALL\" means that OBMC Boot Test" +
43 " should stop after any boot test. If environment variable" +
44 " STOP_REST_FAIL is set, OBMC Boot Test will stop if REST commands are" +
45 " no longer working.",
46 formatter_class=argparse.RawTextHelpFormatter,
47 prefix_chars='-+')
48
49# The stock_list will be passed to gen_get_options. We populate it with the
50# names of stock parm options we want. These stock parms are pre-defined by
51# gen_get_options.
52stock_list = [("test_mode", 0),
53 ("quiet", get_plug_default("quiet", 0)),
54 ("debug", get_plug_default("debug", 0))]
55
56
57def exit_function(signal_number=0,
58 frame=None):
Michael Walsh3ba8ecd2018-04-24 11:33:25 -050059 r"""
60 Execute whenever the program ends normally or with the signals that we
61 catch (i.e. TERM, INT).
62 """
63
64 dprint_executing()
65 dprint_var(signal_number)
66
Michael Walsha0ce75a2018-07-31 13:54:29 -050067 process_robot_output_files()
68
Michael Walsh3ba8ecd2018-04-24 11:33:25 -050069 qprint_pgm_footer()
70
71
72def signal_handler(signal_number,
73 frame):
Michael Walsh3ba8ecd2018-04-24 11:33:25 -050074 r"""
75 Handle signals. Without a function to catch a SIGTERM or SIGINT, our
76 program would terminate immediately with return code 143 and without
77 calling our exit_function.
78 """
79
80 # Our convention is to set up exit_function with atexit.register() so
81 # there is no need to explicitly call exit_function from here.
82
83 dprint_executing()
84
85 # Calling exit prevents us from returning to the code that was running
86 # when we received the signal.
87 exit(0)
88
89
90def validate_parms():
Michael Walsh3ba8ecd2018-04-24 11:33:25 -050091 r"""
92 Validate program parameters, etc. Return True or False (i.e. pass/fail)
93 accordingly.
94 """
95
96 get_plug_vars()
97
98 if not valid_value(AUTOBOOT_OPENBMC_HOST, ["", None]):
99 return False
100
101 gen_post_validation(exit_function, signal_handler)
102
103 return True
104
105
106def rest_fail():
Michael Walsh3ba8ecd2018-04-24 11:33:25 -0500107 r"""
108 If STOP_REST_FAIL, then this function will determine whether REST commands
109 to the target are working. If not, this function will stop the program by
110 returning stop_test_rc.
111 """
112
113 if STOP_REST_FAIL != '1':
114 return
115
116 print_timen("Checking to see whether REST commands are working.")
117 init_robot_out_parms(get_plug_in_package_name() + ".")
118 lib_file_path = init_robot_file_path("lib/state.py")
119 set_mod_global(lib_file_path)
120 timeout = '0 seconds'
121 interval = '1 second'
122 keyword_string = "${match_state}= Create Dictionary rest=1 ;" +\
123 " ${state}= Wait State ${match_state} " + timeout + " " +\
124 interval + " quiet=${1} ; Rpvar state"
125 set_mod_global(keyword_string)
126
127 cmd_buf = create_robot_cmd_string("extended/run_keyword.robot",
128 OPENBMC_HOST, keyword_string,
129 lib_file_path, quiet, test_mode, debug,
130 outputdir, output, log, report, loglevel)
131 if not robot_cmd_fnc(cmd_buf):
132 print_timen("The caller wishes to stop test execution if REST" +
133 " commands are failing.")
134 exit(stop_test_rc)
135 print_timen("REST commands are working so no reason as of yet to stop" +
136 " the test.")
137
138
139def esel_stop_check():
Michael Walsh3ba8ecd2018-04-24 11:33:25 -0500140 r"""
141 Run the esel_stop_check program to determine whether any eSEL entries
Gunnar Mills7732c7e2018-08-14 11:54:24 -0500142 found warrant stopping the test run. See esel_stop_check help text for
Michael Walsh3ba8ecd2018-04-24 11:33:25 -0500143 details.
144 """
145
146 if STOP_ESEL_STOP_FILE_PATH == "":
147 return
148
149 cmd_buf = "esel_stop_check --esel_stop_file_path=" +\
150 STOP_ESEL_STOP_FILE_PATH
151 shell_rc, out_buf = cmd_fnc_u(cmd_buf, show_err=0)
152 if shell_rc == stop_test_rc:
153 print_timen("The caller wishes to stop test execution based on the" +
154 " presence of certain esel entries.")
155 exit(stop_test_rc)
156
157
158def main():
159
160 if not gen_get_options(parser, stock_list):
161 return False
162
163 if not validate_parms():
164 return False
165
166 qprint_pgm_header()
167
168 if not debug:
169 qprint_vars(STOP_REST_FAIL, STOP_COMMAND, AUTOBOOT_BOOT_SUCCESS)
170
171 dprint_plug_vars()
172
173 rest_fail()
174
175 esel_stop_check()
176
177 if STOP_COMMAND.upper() == "FAIL":
178 if AUTOBOOT_BOOT_SUCCESS == "0":
179 print_timen("The caller wishes to stop after each boot failure.")
180 exit(stop_test_rc)
181 elif STOP_COMMAND.upper() == "ALL":
182 print_timen("The caller wishes to stop after each boot test.")
183 exit(stop_test_rc)
184 elif len(STOP_COMMAND) > 0:
185 shell_rc, out_buf = cmd_fnc_u(STOP_COMMAND, quiet=quiet, show_err=0)
186 if shell_rc != 0:
187 print_timen("The caller wishes to stop test execution.")
188 exit(stop_test_rc)
189
190 qprint_timen("The caller does not wish to stop the test run.")
191 return True
192
193# Main
194
Joy Onyerikwu004ad3c2018-06-11 16:29:56 -0500195
Michael Walsh3ba8ecd2018-04-24 11:33:25 -0500196if not main():
197 exit(1)