1#!/usr/bin/env python
2
3r"""
4This module is the python counterpart to obmc_boot_test.
5"""
6
7import os
8import imp
9import time
10import glob
11import random
12import re
13import cPickle as pickle
14import socket
15
16from robot.utils import DotDict
17from robot.libraries.BuiltIn import BuiltIn
18
19from boot_data import *
20import gen_print as gp
21import gen_robot_print as grp
22import gen_robot_plug_in as grpi
23import gen_robot_valid as grv
24import gen_misc as gm
25import gen_cmd as gc
26import gen_robot_keyword as grk
27import state as st
28
29base_path = os.path.dirname(os.path.dirname(
30                            imp.find_module("gen_robot_print")[1])) +\
31    os.sep
32sys.path.append(base_path + "extended/")
33import run_keyword as rk
34
35# Setting master_pid correctly influences the behavior of plug-ins like
36# DB_Logging
37program_pid = os.getpid()
38master_pid = os.environ.get('AUTOBOOT_MASTER_PID', program_pid)
39
40# Set up boot data structures.
41boot_table = create_boot_table()
42valid_boot_types = create_valid_boot_list(boot_table)
43
44boot_lists = read_boot_lists()
45last_ten = []
46
47state = st.return_default_state()
48cp_setup_called = 0
49next_boot = ""
50base_tool_dir_path = os.path.normpath(os.environ.get(
51    'AUTOBOOT_BASE_TOOL_DIR_PATH', "/tmp")) + os.sep
52
53ffdc_dir_path = os.path.normpath(os.environ.get('FFDC_DIR_PATH', '')) + os.sep
54boot_success = 0
55status_dir_path = os.environ.get('STATUS_DIR_PATH', "")
56if status_dir_path != "":
57    status_dir_path = os.path.normpath(status_dir_path) + os.sep
58default_power_on = "REST Power On"
59default_power_off = "REST Power Off"
60boot_count = 0
61
62LOG_LEVEL = BuiltIn().get_variable_value("${LOG_LEVEL}")
63
64
65###############################################################################
66def process_host(host,
67                 host_var_name=""):
68
69    r"""
70    Process a host by getting the associated host name and IP address and
71    setting them in global variables.
72
73    If the caller does not pass the host_var_name, this function will try to
74    figure out the name of the variable used by the caller for the host parm.
75    Callers are advised to explicitly specify the host_var_name when calling
76    with an exec command.  In such cases, the get_arg_name cannot figure out
77    the host variable name.
78
79    This function will then create similar global variable names by
80    removing "_host" and appending "_host_name" or "_ip" to the host variable
81    name.
82
83    Example:
84
85    If a call is made like this:
86    process_host(openbmc_host)
87
88    Global variables openbmc_host_name and openbmc_ip will be set.
89
90    Description of argument(s):
91    host           A host name or IP.  The name of the variable used should
92                   have a suffix of "_host".
93    host_var_name  The name of the variable being used as the host parm.
94    """
95
96    if host_var_name == "":
97        host_var_name = gp.get_arg_name(0, 1, stack_frame_ix=2)
98
99    host_name_var_name = re.sub("host", "host_name", host_var_name)
100    ip_var_name = re.sub("host", "ip", host_var_name)
101    cmd_buf = "global " + host_name_var_name + ", " + ip_var_name + " ; " +\
102        host_name_var_name + ", " + ip_var_name + " = gm.get_host_name_ip('" +\
103        host + "')"
104    exec(cmd_buf)
105
106###############################################################################
107
108
109###############################################################################
110def process_pgm_parms():
111
112    r"""
113    Process the program parameters by assigning them all to corresponding
114    globals.  Also, set some global values that depend on program parameters.
115    """
116
117    # Program parameter processing.
118    # Assign all program parms to python variables which are global to this
119    # module.
120
121    global parm_list
122    parm_list = BuiltIn().get_variable_value("${parm_list}")
123    # The following subset of parms should be processed as integers.
124    int_list = ['max_num_tests', 'boot_pass', 'boot_fail', 'ffdc_only',
125                'boot_fail_threshold', 'quiet', 'test_mode', 'debug']
126    for parm in parm_list:
127        if parm in int_list:
128            sub_cmd = "int(BuiltIn().get_variable_value(\"${" + parm +\
129                      "}\", \"0\"))"
130        else:
131            sub_cmd = "BuiltIn().get_variable_value(\"${" + parm + "}\")"
132        cmd_buf = "global " + parm + " ; " + parm + " = " + sub_cmd
133        exec(cmd_buf)
134        if re.match(r".*_host$", parm):
135            cmd_buf = "process_host(" + parm + ", '" + parm + "')"
136            exec(cmd_buf)
137        if re.match(r".*_password$", parm):
138            # Register the value of any parm whose name ends in _password.
139            # This will cause the print functions to replace passwords with
140            # asterisks in the output.
141            cmd_buf = "gp.register_passwords(" + parm + ")"
142            exec(cmd_buf)
143
144    global ffdc_dir_path_style
145    global boot_list
146    global boot_stack
147    global boot_results_file_path
148    global boot_results
149    global ffdc_list_file_path
150    global ffdc_report_list_path
151    global ffdc_summary_list_path
152
153    if ffdc_dir_path_style == "":
154        ffdc_dir_path_style = int(os.environ.get('FFDC_DIR_PATH_STYLE', '0'))
155
156    # Convert these program parms to lists for easier processing..
157    boot_list = filter(None, boot_list.split(":"))
158    boot_stack = filter(None, boot_stack.split(":"))
159
160    boot_results_file_path = "/tmp/" + openbmc_nickname + ":pid_" +\
161                             str(master_pid) + ":boot_results"
162
163    if os.path.isfile(boot_results_file_path):
164        # We've been called before in this run so we'll load the saved
165        # boot_results object.
166        boot_results = pickle.load(open(boot_results_file_path, 'rb'))
167    else:
168        boot_results = boot_results(boot_table, boot_pass, boot_fail)
169
170    ffdc_list_file_path = base_tool_dir_path + openbmc_nickname +\
171        "/FFDC_FILE_LIST"
172    ffdc_report_list_path = base_tool_dir_path + openbmc_nickname +\
173        "/FFDC_REPORT_FILE_LIST"
174
175    ffdc_summary_list_path = base_tool_dir_path + openbmc_nickname +\
176        "/FFDC_SUMMARY_FILE_LIST"
177
178###############################################################################
179
180
181###############################################################################
182def initial_plug_in_setup():
183
184    r"""
185    Initialize all plug-in environment variables which do not change for the
186    duration of the program.
187
188    """
189
190    global LOG_LEVEL
191    BuiltIn().set_log_level("NONE")
192
193    BuiltIn().set_global_variable("${master_pid}", master_pid)
194    BuiltIn().set_global_variable("${FFDC_DIR_PATH}", ffdc_dir_path)
195    BuiltIn().set_global_variable("${STATUS_DIR_PATH}", status_dir_path)
196    BuiltIn().set_global_variable("${BASE_TOOL_DIR_PATH}", base_tool_dir_path)
197    BuiltIn().set_global_variable("${FFDC_LIST_FILE_PATH}",
198                                  ffdc_list_file_path)
199    BuiltIn().set_global_variable("${FFDC_REPORT_LIST_PATH}",
200                                  ffdc_report_list_path)
201    BuiltIn().set_global_variable("${FFDC_SUMMARY_LIST_PATH}",
202                                  ffdc_summary_list_path)
203
204    BuiltIn().set_global_variable("${FFDC_DIR_PATH_STYLE}",
205                                  ffdc_dir_path_style)
206    BuiltIn().set_global_variable("${FFDC_CHECK}",
207                                  ffdc_check)
208
209    # For each program parameter, set the corresponding AUTOBOOT_ environment
210    # variable value.  Also, set an AUTOBOOT_ environment variable for every
211    # element in additional_values.
212    additional_values = ["program_pid", "master_pid", "ffdc_dir_path",
213                         "status_dir_path", "base_tool_dir_path",
214                         "ffdc_list_file_path", "ffdc_report_list_path",
215                         "ffdc_summary_list_path"]
216
217    plug_in_vars = parm_list + additional_values
218
219    for var_name in plug_in_vars:
220        var_value = BuiltIn().get_variable_value("${" + var_name + "}")
221        var_name = var_name.upper()
222        if var_value is None:
223            var_value = ""
224        os.environ["AUTOBOOT_" + var_name] = str(var_value)
225
226    BuiltIn().set_log_level(LOG_LEVEL)
227
228    # Make sure the ffdc list directory exists.
229    ffdc_list_dir_path = os.path.dirname(ffdc_list_file_path) + os.sep
230    if not os.path.exists(ffdc_list_dir_path):
231        os.makedirs(ffdc_list_dir_path)
232
233###############################################################################
234
235
236###############################################################################
237def plug_in_setup():
238
239    r"""
240    Initialize all changing plug-in environment variables for use by the
241    plug-in programs.
242    """
243
244    global LOG_LEVEL
245    global test_really_running
246
247    BuiltIn().set_log_level("NONE")
248
249    boot_pass, boot_fail = boot_results.return_total_pass_fail()
250    if boot_pass > 1:
251        test_really_running = 1
252    else:
253        test_really_running = 0
254
255    seconds = time.time()
256    loc_time = time.localtime(seconds)
257    time_string = time.strftime("%y%m%d.%H%M%S.", loc_time)
258
259    ffdc_prefix = openbmc_nickname + "." + time_string
260
261    BuiltIn().set_global_variable("${test_really_running}",
262                                  test_really_running)
263    BuiltIn().set_global_variable("${boot_type_desc}", next_boot)
264    BuiltIn().set_global_variable("${boot_pass}", boot_pass)
265    BuiltIn().set_global_variable("${boot_fail}", boot_fail)
266    BuiltIn().set_global_variable("${boot_success}", boot_success)
267    BuiltIn().set_global_variable("${ffdc_prefix}", ffdc_prefix)
268
269    # For each program parameter, set the corresponding AUTOBOOT_ environment
270    # variable value.  Also, set an AUTOBOOT_ environment variable for every
271    # element in additional_values.
272    additional_values = ["boot_type_desc", "boot_success", "boot_pass",
273                         "boot_fail", "test_really_running", "ffdc_prefix"]
274
275    plug_in_vars = additional_values
276
277    for var_name in plug_in_vars:
278        var_value = BuiltIn().get_variable_value("${" + var_name + "}")
279        var_name = var_name.upper()
280        if var_value is None:
281            var_value = ""
282        os.environ["AUTOBOOT_" + var_name] = str(var_value)
283
284    if debug:
285        shell_rc, out_buf = \
286            gc.cmd_fnc_u("printenv | egrep AUTOBOOT_ | sort -u")
287
288    BuiltIn().set_log_level(LOG_LEVEL)
289
290###############################################################################
291
292
293###############################################################################
294def pre_boot_plug_in_setup():
295
296    # Clear the ffdc_list_file_path file.  Plug-ins may now write to it.
297    try:
298        os.remove(ffdc_list_file_path)
299    except OSError:
300        pass
301
302    # Clear the ffdc_report_list_path file.  Plug-ins may now write to it.
303    try:
304        os.remove(ffdc_report_list_path)
305    except OSError:
306        pass
307
308    # Clear the ffdc_summary_list_path file.  Plug-ins may now write to it.
309    try:
310        os.remove(ffdc_summary_list_path)
311    except OSError:
312        pass
313
314###############################################################################
315
316
317###############################################################################
318def setup():
319
320    r"""
321    Do general program setup tasks.
322    """
323
324    global cp_setup_called
325
326    gp.qprintn()
327
328    robot_pgm_dir_path = os.path.dirname(__file__) + os.sep
329    repo_bin_path = robot_pgm_dir_path.replace("/lib/", "/bin/")
330    # If we can't find process_plug_in_packages.py, ssh_pw or
331    # validate_plug_ins.py, then we don't have our repo bin in PATH.
332    shell_rc, out_buf = gc.cmd_fnc_u("which process_plug_in_packages.py" +
333                                     " ssh_pw validate_plug_ins.py", quiet=1,
334                                     print_output=0, show_err=0)
335    if shell_rc != 0:
336        os.environ['PATH'] = repo_bin_path + ":" + os.environ.get('PATH', "")
337    # Likewise, our repo lib subdir needs to be in sys.path and PYTHONPATH.
338    if robot_pgm_dir_path not in sys.path:
339        sys.path.append(robot_pgm_dir_path)
340        PYTHONPATH = os.environ.get("PYTHONPATH", "")
341        if PYTHONPATH == "":
342            os.environ['PYTHONPATH'] = robot_pgm_dir_path
343        else:
344            os.environ['PYTHONPATH'] = robot_pgm_dir_path + ":" + PYTHONPATH
345
346    validate_parms()
347
348    grp.rqprint_pgm_header()
349
350    grk.run_key("Set BMC Power Policy  RESTORE_LAST_STATE")
351
352    initial_plug_in_setup()
353
354    plug_in_setup()
355    rc, shell_rc, failed_plug_in_name = grpi.rprocess_plug_in_packages(
356        call_point='setup')
357    if rc != 0:
358        error_message = "Plug-in setup failed.\n"
359        grp.rprint_error_report(error_message)
360        BuiltIn().fail(error_message)
361    # Setting cp_setup_called lets our Teardown know that it needs to call
362    # the cleanup plug-in call point.
363    cp_setup_called = 1
364
365    # Keyword "FFDC" will fail if TEST_MESSAGE is not set.
366    BuiltIn().set_global_variable("${TEST_MESSAGE}", "${EMPTY}")
367    # FFDC_LOG_PATH is used by "FFDC" keyword.
368    BuiltIn().set_global_variable("${FFDC_LOG_PATH}", ffdc_dir_path)
369
370    # Also printed by FFDC.
371    global host_name
372    global host_ip
373    host = socket.gethostname()
374    host_name, host_ip = gm.get_host_name_ip(host)
375
376    gp.dprint_var(boot_table, 1)
377    gp.dprint_var(boot_lists)
378
379###############################################################################
380
381
382###############################################################################
383def validate_parms():
384
385    r"""
386    Validate all program parameters.
387    """
388
389    process_pgm_parms()
390
391    gp.qprintn()
392
393    global openbmc_model
394    grv.rvalid_value("openbmc_host")
395    grv.rvalid_value("openbmc_username")
396    grv.rvalid_value("openbmc_password")
397    if os_host != "":
398        grv.rvalid_value("os_username")
399        grv.rvalid_value("os_password")
400
401    if pdu_host != "":
402        grv.rvalid_value("pdu_username")
403        grv.rvalid_value("pdu_password")
404        grv.rvalid_integer("pdu_slot_no")
405    if openbmc_serial_host != "":
406        grv.rvalid_integer("openbmc_serial_port")
407    if openbmc_model == "":
408        status, ret_values =\
409            grk.run_key_u("Get BMC System Model")
410        openbmc_model = ret_values
411        BuiltIn().set_global_variable("${openbmc_model}", openbmc_model)
412    grv.rvalid_value("openbmc_model")
413    grv.rvalid_integer("max_num_tests")
414    grv.rvalid_integer("boot_pass")
415    grv.rvalid_integer("boot_fail")
416
417    plug_in_packages_list = grpi.rvalidate_plug_ins(plug_in_dir_paths)
418    BuiltIn().set_global_variable("${plug_in_packages_list}",
419                                  plug_in_packages_list)
420
421    grv.rvalid_value("stack_mode", valid_values=['normal', 'skip'])
422    if len(boot_list) == 0 and len(boot_stack) == 0 and not ffdc_only:
423        error_message = "You must provide either a value for either the" +\
424            " boot_list or the boot_stack parm.\n"
425        BuiltIn().fail(gp.sprint_error(error_message))
426
427    valid_boot_list(boot_list, valid_boot_types)
428    valid_boot_list(boot_stack, valid_boot_types)
429
430    selected_PDU_boots = list(set(boot_list + boot_stack) &
431                              set(boot_lists['PDU_reboot']))
432
433    if len(selected_PDU_boots) > 0 and pdu_host == "":
434        error_message = "You have selected the following boots which" +\
435                        " require a PDU host but no value for pdu_host:\n"
436        error_message += gp.sprint_var(selected_PDU_boots)
437        error_message += gp.sprint_var(pdu_host, 2)
438        BuiltIn().fail(gp.sprint_error(error_message))
439
440    return
441
442###############################################################################
443
444
445###############################################################################
446def my_get_state():
447
448    r"""
449    Get the system state plus a little bit of wrapping.
450    """
451
452    global state
453
454    req_states = ['epoch_seconds'] + st.default_req_states
455
456    gp.qprint_timen("Getting system state.")
457    if test_mode:
458        state['epoch_seconds'] = int(time.time())
459    else:
460        state = st.get_state(req_states=req_states, quiet=quiet)
461    gp.qprint_var(state)
462
463###############################################################################
464
465
466###############################################################################
467def select_boot():
468
469    r"""
470    Select a boot test to be run based on our current state and return the
471    chosen boot type.
472
473    Description of arguments:
474    state  The state of the machine.
475    """
476
477    global boot_stack
478
479    gp.qprint_timen("Selecting a boot test.")
480
481    my_get_state()
482
483    stack_popped = 0
484    if len(boot_stack) > 0:
485        stack_popped = 1
486        gp.qprint_dashes()
487        gp.qprint_var(boot_stack)
488        gp.qprint_dashes()
489        skip_boot_printed = 0
490        while len(boot_stack) > 0:
491            boot_candidate = boot_stack.pop()
492            if stack_mode == 'normal':
493                break
494            else:
495                if st.compare_states(state, boot_table[boot_candidate]['end']):
496                    if not skip_boot_printed:
497                        gp.print_var(stack_mode)
498                        gp.printn()
499                        gp.print_timen("Skipping the following boot tests" +
500                                       " which are unnecessary since their" +
501                                       " required end states match the" +
502                                       " current machine state:")
503                        skip_boot_printed = 1
504                    gp.print_var(boot_candidate)
505                    boot_candidate = ""
506        if boot_candidate == "":
507            gp.qprint_dashes()
508            gp.qprint_var(boot_stack)
509            gp.qprint_dashes()
510            return boot_candidate
511        if st.compare_states(state, boot_table[boot_candidate]['start']):
512            gp.qprint_timen("The machine state is valid for a '" +
513                            boot_candidate + "' boot test.")
514            gp.qprint_dashes()
515            gp.qprint_var(boot_stack)
516            gp.qprint_dashes()
517            return boot_candidate
518        else:
519            gp.qprint_timen("The machine state does not match the required" +
520                            " starting state for a '" + boot_candidate +
521                            "' boot test:")
522            gp.print_varx("boot_table[" + boot_candidate + "][start]",
523                          boot_table[boot_candidate]['start'], 1)
524            boot_stack.append(boot_candidate)
525            popped_boot = boot_candidate
526
527    # Loop through your list selecting a boot_candidates
528    boot_candidates = []
529    for boot_candidate in boot_list:
530        if st.compare_states(state, boot_table[boot_candidate]['start']):
531            if stack_popped:
532                if st.compare_states(boot_table[boot_candidate]['end'],
533                   boot_table[popped_boot]['start']):
534                    boot_candidates.append(boot_candidate)
535            else:
536                boot_candidates.append(boot_candidate)
537
538    if len(boot_candidates) == 0:
539        gp.qprint_timen("The user's boot list contained no boot tests" +
540                        " which are valid for the current machine state.")
541        boot_candidate = default_power_on
542        if not st.compare_states(state, boot_table[default_power_on]['start']):
543            boot_candidate = default_power_off
544        boot_candidates.append(boot_candidate)
545        gp.qprint_timen("Using default '" + boot_candidate +
546                        "' boot type to transition to valid state.")
547
548    gp.dprint_var(boot_candidates)
549
550    # Randomly select a boot from the candidate list.
551    boot = random.choice(boot_candidates)
552
553    return boot
554
555###############################################################################
556
557
558###############################################################################
559def print_last_boots():
560
561    r"""
562    Print the last ten boots done with their time stamps.
563    """
564
565    # indent 0, 90 chars wide, linefeed, char is "="
566    gp.qprint_dashes(0, 90)
567    gp.qprintn("Last 10 boots:\n")
568
569    for boot_entry in last_ten:
570        grp.rqprint(boot_entry)
571    gp.qprint_dashes(0, 90)
572
573###############################################################################
574
575
576###############################################################################
577def print_defect_report():
578
579    r"""
580    Print a defect report.
581    """
582
583    # Making deliberate choice to NOT run plug_in_setup().  We don't want
584    # ffdc_prefix updated.
585    rc, shell_rc, failed_plug_in_name = grpi.rprocess_plug_in_packages(
586        call_point='ffdc_report', stop_on_plug_in_failure=0)
587
588    # At some point I'd like to have the 'Call FFDC Methods' return a list
589    # of files it has collected.  In that case, the following "ls" command
590    # would no longer be needed.  For now, however, glob shows the files
591    # named in FFDC_LIST_FILE_PATH so I will refrain from printing those
592    # out (so we don't see duplicates in the list).
593
594    # Get additional header data which may have been created by ffdc plug-ins.
595    # Also, delete the individual header files to cleanup.
596    cmd_buf = "file_list=$(cat " + ffdc_report_list_path + " 2>/dev/null)" +\
597              " ; [ ! -z \"${file_list}\" ] && cat ${file_list}" +\
598              " 2>/dev/null ; rm -rf ${file_list} 2>/dev/null || :"
599    shell_rc, more_header_info = gc.cmd_fnc_u(cmd_buf, print_output=0,
600                                              show_err=0)
601
602    # Get additional header data which may have been created by ffdc plug-ins.
603    # Also, delete the individual header files to cleanup.
604    cmd_buf = "file_list=$(cat " + ffdc_summary_list_path + " 2>/dev/null)" +\
605              " ; [ ! -z \"${file_list}\" ] && cat ${file_list}" +\
606              " 2>/dev/null ; rm -rf ${file_list} 2>/dev/null || :"
607    shell_rc, ffdc_summary_info = gc.cmd_fnc_u(cmd_buf, print_output=0,
608                                               show_err=0)
609
610    LOG_PREFIX = BuiltIn().get_variable_value("${LOG_PREFIX}")
611
612    output = '\n'.join(sorted(glob.glob(LOG_PREFIX + '*')))
613    try:
614        ffdc_list = open(ffdc_list_file_path, 'r')
615    except IOError:
616        ffdc_list = ""
617
618    # Open ffdc_file_list for writing.  We will write a complete list of
619    # FFDC files to it for possible use by plug-ins like cp_stop_check.
620    ffdc_list_file = open(ffdc_list_file_path, 'w')
621
622    gp.qprintn()
623    # indent=0, width=90, linefeed=1, char="="
624    gp.qprint_dashes(0, 90, 1, "=")
625    gp.qprintn("Copy this data to the defect:\n")
626
627    if len(more_header_info) > 0:
628        gp.printn(more_header_info)
629    gp.qpvars(host_name, host_ip, openbmc_nickname, openbmc_host,
630              openbmc_host_name, openbmc_ip, openbmc_username,
631              openbmc_password, os_host, os_host_name, os_ip, os_username,
632              os_password, pdu_host, pdu_host_name, pdu_ip, pdu_username,
633              pdu_password, pdu_slot_no, openbmc_serial_host,
634              openbmc_serial_host_name, openbmc_serial_ip, openbmc_serial_port)
635
636    gp.qprintn()
637
638    print_last_boots()
639    gp.qprintn()
640    gp.qprint_var(state)
641
642    gp.qprintn()
643    gp.qprintn("FFDC data files:")
644    if status_file_path != "":
645        gp.qprintn(status_file_path)
646        ffdc_list_file.write(status_file_path + "\n")
647
648    gp.qprintn(output)
649    # gp.qprintn(ffdc_list)
650    gp.qprintn()
651
652    if len(ffdc_summary_info) > 0:
653        gp.printn(ffdc_summary_info)
654
655    gp.qprint_dashes(0, 90, 1, "=")
656
657    ffdc_list_file.write(output + "\n")
658    ffdc_list_file.close()
659
660###############################################################################
661
662
663###############################################################################
664def my_ffdc():
665
666    r"""
667    Collect FFDC data.
668    """
669
670    global state
671
672    plug_in_setup()
673    rc, shell_rc, failed_plug_in_name = grpi.rprocess_plug_in_packages(
674        call_point='ffdc', stop_on_plug_in_failure=0)
675
676    AUTOBOOT_FFDC_PREFIX = os.environ['AUTOBOOT_FFDC_PREFIX']
677    status, ret_values = grk.run_key_u("FFDC  ffdc_prefix=" +
678                                       AUTOBOOT_FFDC_PREFIX +
679                                       "  ffdc_function_list=" +
680                                       ffdc_function_list, ignore=1)
681    if status != 'PASS':
682        gp.print_error("Call to ffdc failed.\n")
683
684    my_get_state()
685
686    print_defect_report()
687
688###############################################################################
689
690
691###############################################################################
692def print_test_start_message(boot_keyword):
693
694    r"""
695    Print a message indicating what boot test is about to run.
696
697    Description of arguments:
698    boot_keyword  The name of the boot which is to be run
699                  (e.g. "BMC Power On").
700    """
701
702    global last_ten
703
704    doing_msg = gp.sprint_timen("Doing \"" + boot_keyword + "\".")
705    gp.qprint(doing_msg)
706
707    last_ten.append(doing_msg)
708
709    if len(last_ten) > 10:
710        del last_ten[0]
711
712###############################################################################
713
714
715###############################################################################
716def run_boot(boot):
717
718    r"""
719    Run the specified boot.
720
721    Description of arguments:
722    boot  The name of the boot test to be performed.
723    """
724
725    global state
726
727    print_test_start_message(boot)
728
729    plug_in_setup()
730    rc, shell_rc, failed_plug_in_name = \
731        grpi.rprocess_plug_in_packages(call_point="pre_boot")
732    if rc != 0:
733        error_message = "Plug-in failed with non-zero return code.\n" +\
734            gp.sprint_var(rc, 1)
735        BuiltIn().fail(gp.sprint_error(error_message))
736
737    if test_mode:
738        # In test mode, we'll pretend the boot worked by assigning its
739        # required end state to the default state value.
740        state = st.strip_anchor_state(boot_table[boot]['end'])
741    else:
742        # Assertion:  We trust that the state data was made fresh by the
743        # caller.
744
745        gp.qprintn()
746
747        if boot_table[boot]['method_type'] == "keyword":
748            rk.my_run_keywords(boot_table[boot].get('lib_file_path', ''),
749                               boot_table[boot]['method'],
750                               quiet=quiet)
751
752        if boot_table[boot]['bmc_reboot']:
753            st.wait_for_comm_cycle(int(state['epoch_seconds']))
754            plug_in_setup()
755            rc, shell_rc, failed_plug_in_name = \
756                grpi.rprocess_plug_in_packages(call_point="post_reboot")
757            if rc != 0:
758                error_message = "Plug-in failed with non-zero return code.\n"
759                error_message += gp.sprint_var(rc, 1)
760                BuiltIn().fail(gp.sprint_error(error_message))
761        else:
762            match_state = st.anchor_state(state)
763            del match_state['epoch_seconds']
764            # Wait for the state to change in any way.
765            st.wait_state(match_state, wait_time=state_change_timeout,
766                          interval="10 seconds", invert=1)
767
768        gp.qprintn()
769        if boot_table[boot]['end']['chassis'] == "Off":
770            boot_timeout = power_off_timeout
771        else:
772            boot_timeout = power_on_timeout
773        st.wait_state(boot_table[boot]['end'], wait_time=boot_timeout,
774                      interval="10 seconds")
775
776    plug_in_setup()
777    rc, shell_rc, failed_plug_in_name = \
778        grpi.rprocess_plug_in_packages(call_point="post_boot")
779    if rc != 0:
780        error_message = "Plug-in failed with non-zero return code.\n" +\
781            gp.sprint_var(rc, 1)
782        BuiltIn().fail(gp.sprint_error(error_message))
783
784###############################################################################
785
786
787###############################################################################
788def test_loop_body():
789
790    r"""
791    The main loop body for the loop in main_py.
792
793    Description of arguments:
794    boot_count  The iteration number (starts at 1).
795    """
796
797    global boot_count
798    global state
799    global next_boot
800    global boot_success
801
802    gp.qprintn()
803
804    next_boot = select_boot()
805    if next_boot == "":
806        return True
807
808    boot_count += 1
809    gp.qprint_timen("Starting boot " + str(boot_count) + ".")
810
811    pre_boot_plug_in_setup()
812
813    cmd_buf = ["run_boot", next_boot]
814    boot_status, msg = BuiltIn().run_keyword_and_ignore_error(*cmd_buf)
815    if boot_status == "FAIL":
816        gp.qprint(msg)
817
818    gp.qprintn()
819    if boot_status == "PASS":
820        boot_success = 1
821        gp.qprint_timen("BOOT_SUCCESS: \"" + next_boot + "\" succeeded.")
822    else:
823        boot_success = 0
824        gp.qprint_timen("BOOT_FAILED: \"" + next_boot + "\" failed.")
825
826    boot_results.update(next_boot, boot_status)
827
828    plug_in_setup()
829    # NOTE: A post_test_case call point failure is NOT counted as a boot
830    # failure.
831    rc, shell_rc, failed_plug_in_name = grpi.rprocess_plug_in_packages(
832        call_point='post_test_case', stop_on_plug_in_failure=0)
833
834    plug_in_setup()
835    rc, shell_rc, failed_plug_in_name = grpi.rprocess_plug_in_packages(
836        call_point='ffdc_check', shell_rc=0x00000200,
837        stop_on_plug_in_failure=1, stop_on_non_zero_rc=1)
838    if boot_status != "PASS" or ffdc_check == "All" or shell_rc == 0x00000200:
839        status, ret_values = grk.run_key_u("my_ffdc", ignore=1)
840        if status != 'PASS':
841            gp.print_error("Call to my_ffdc failed.\n")
842
843    # We need to purge error logs between boots or they build up.
844    grk.run_key("Delete Error logs", ignore=1)
845
846    boot_results.print_report()
847    gp.qprint_timen("Finished boot " + str(boot_count) + ".")
848
849    plug_in_setup()
850    rc, shell_rc, failed_plug_in_name = grpi.rprocess_plug_in_packages(
851        call_point='stop_check')
852    if rc != 0:
853        error_message = "Stopping as requested by user.\n"
854        grp.rprint_error_report(error_message)
855        BuiltIn().fail(error_message)
856
857    # This should help prevent ConnectionErrors.
858    grk.run_key_u("Delete All Sessions")
859
860    return True
861
862###############################################################################
863
864
865###############################################################################
866def obmc_boot_test_teardown():
867
868    r"""
869    Clean up after the Main keyword.
870    """
871
872    if cp_setup_called:
873        plug_in_setup()
874        rc, shell_rc, failed_plug_in_name = grpi.rprocess_plug_in_packages(
875            call_point='cleanup', stop_on_plug_in_failure=0)
876
877    if 'boot_results_file_path' in globals():
878        # Save boot_results object to a file in case it is needed again.
879        gp.qprint_timen("Saving boot_results to the following path.")
880        gp.qprint_var(boot_results_file_path)
881        pickle.dump(boot_results, open(boot_results_file_path, 'wb'),
882                    pickle.HIGHEST_PROTOCOL)
883
884###############################################################################
885
886
887###############################################################################
888def test_teardown():
889
890    r"""
891    Clean up after this test case.
892    """
893
894    gp.qprintn()
895    cmd_buf = ["Print Error",
896               "A keyword timeout occurred ending this program.\n"]
897    BuiltIn().run_keyword_if_timeout_occurred(*cmd_buf)
898
899    grp.rqprint_pgm_footer()
900
901###############################################################################
902
903
904###############################################################################
905def obmc_boot_test_py(alt_boot_stack=None):
906
907    r"""
908    Do main program processing.
909    """
910
911    if alt_boot_stack is not None:
912        BuiltIn().set_global_variable("${boot_stack}", alt_boot_stack)
913
914    setup()
915
916    if ffdc_only:
917        gp.qprint_timen("Caller requested ffdc_only.")
918        pre_boot_plug_in_setup()
919        grk.run_key_u("my_ffdc")
920        return
921
922    # Process caller's boot_stack.
923    while (len(boot_stack) > 0):
924        test_loop_body()
925
926    gp.qprint_timen("Finished processing stack.")
927
928    # Process caller's boot_list.
929    if len(boot_list) > 0:
930        for ix in range(1, max_num_tests + 1):
931            test_loop_body()
932
933    gp.qprint_timen("Completed all requested boot tests.")
934
935    boot_pass, boot_fail = boot_results.return_total_pass_fail()
936    if boot_fail > boot_fail_threshold:
937        error_message = "Boot failures exceed the boot failure" +\
938                        " threshold:\n" +\
939                        gp.sprint_var(boot_fail) +\
940                        gp.sprint_var(boot_fail_threshold)
941        BuiltIn().fail(gp.sprint_error(error_message))
942
943###############################################################################
944