def main(): duts = [] refs = [] monitors = [] filter_keys = [] requested_tests = ["help"] requested_hwsim_tests = [] hwsim_tests = [] cfg_file = "cfg.py" log_dir = "./logs/" verbose = False trace = False restart = False perf = False # parse input parameters try: opts, args = getopt.getopt(sys.argv[1:], "d:r:t:l:k:c:m:h:vRPT", [ "dut=", "ref=", "tests=", "log-dir=", "cfg=", "key=", "monitor=", "hwsim=" ]) except getopt.GetoptError as err: print(err) usage() sys.exit(2) for option, argument in opts: if option == "-v": verbose = True elif option == "-R": restart = True elif option == "-T": trace = True elif option == "-P": perf = True elif option in ("-d", "--dut"): duts.append(argument) elif option in ("-r", "--ref"): refs.append(argument) elif option in ("-t", "--tests"): requested_tests = re.split('; | |, ', argument) elif option in ("-l", "--log-dir"): log_dir = argument elif option in ("-k", "--key"): filter_keys.append(argument) elif option in ("-m", "--monitor"): monitors.append(argument) elif option in ("-c", "--cfg"): cfg_file = argument elif option in ("-h", "--hwsim"): requested_hwsim_tests = re.split('; | |, ', argument) else: assert False, "unhandled option" # get env configuration setup_params = config.get_setup_params(cfg_file) devices = config.get_devices(cfg_file) # put logs in log_dir symlink = os.path.join(log_dir, "current") if os.path.exists(symlink): os.unlink(symlink) log_dir = os.path.join(log_dir, time.strftime("%Y_%m_%d_%H_%M_%S")) if not os.path.exists(log_dir): os.makedirs(log_dir) os.symlink(os.path.join("../", log_dir), symlink) # setup restart/trace/perf request setup_params['local_log_dir'] = log_dir setup_params['restart_device'] = restart setup_params['trace'] = trace setup_params['perf'] = perf # configure logger logger.setLevel(logging.DEBUG) stdout_handler = logging.StreamHandler() stdout_handler.setLevel(logging.WARNING) if verbose: stdout_handler.setLevel(logging.DEBUG) logger.addHandler(stdout_handler) formatter = logging.Formatter('%(asctime)s - %(message)s') file_name = os.path.join(log_dir, 'run-tests.log') log_handler = logging.FileHandler(file_name) log_handler.setLevel(logging.DEBUG) log_handler.setFormatter(formatter) logger.addHandler(log_handler) # import available tests tests = [] failed = [] test_modules = [] files = os.listdir(scriptsdir) for t in files: m = re.match(r'(test_.*)\.py$', t) if m: mod = __import__(m.group(1)) test_modules.append(mod.__name__.replace('test_', '', 1)) for key, val in mod.__dict__.iteritems(): if key.startswith("test_"): tests.append(val) test_names = list(set([t.__name__.replace('test_', '', 1) for t in tests])) # import test_* files = os.listdir("../hwsim/") for t in files: m = re.match(r'(test_.*)\.py$', t) if m: mod = __import__(m.group(1)) test_modules.append(mod.__name__.replace('test_', '', 1)) for key, val in mod.__dict__.iteritems(): if key.startswith("test_"): hwsim_tests.append(val) # setup hwsim tests hwsim_tests_to_run = [] if len(requested_hwsim_tests) > 0: # apply filters for filter_key in filter_keys: filtered_tests = [] for hwsim_test in hwsim_tests: if re.search(filter_key, hwsim_test.__name__): filtered_tests.append(hwsim_test) hwsim_tests = filtered_tests # setup hwsim_test we should run if requested_hwsim_tests[0] == "all": hwsim_tests_to_run = hwsim_tests else: for test in requested_hwsim_tests: t = None for tt in hwsim_tests: name = tt.__name__.replace('test_', '', 1) if name == test and tt.func_code.co_argcount <= 2: t = tt break if not t: logger.warning("hwsim test case: " + test + " NOT-FOUND") continue hwsim_tests_to_run.append(t) # sort the list test_names.sort() tests.sort() # print help if requested_tests[0] == "help" and len(requested_hwsim_tests) == 0: usage() print "\nAvailable Devices:" for device in devices: print "\t", device['name'] print "\nAvailable tests:" for test in test_names: print "\t", test print "\nAvailable hwsim tests:" for hwsim_test in hwsim_tests: print "\t", hwsim_test.__name__.replace('test_', '', 1) return # show/check devices if requested_tests[0] == "devices": show_devices(devices, setup_params) return # apply filters for filter_key in filter_keys: filtered_tests = [] for test in tests: if re.search(filter_key, test.__name__): filtered_tests.append(test) tests = filtered_tests # setup test we should run tests_to_run = [] if requested_tests[0] == "all": tests_to_run = tests if requested_tests[0] == "help": pass elif requested_tests[0] == "sanity": for test in tests: if test.__name__.startswith("test_sanity_"): tests_to_run.append(test) else: for test in requested_tests: t = None for tt in tests: name = tt.__name__.replace('test_', '', 1) if name == test: t = tt break if not t: logger.warning("test case: " + test + " NOT-FOUND") continue tests_to_run.append(t) # lock devices try: get_devices(devices, duts, refs, monitors) except Exception, e: logger.warning("get devices failed: " + str(e)) logger.info(traceback.format_exc()) put_devices(devices, duts, refs, monitors) return
def main(): duts = [] refs = [] monitors = [] filter_keys = [] requested_tests = ["help"] requested_hwsim_tests = [] hwsim_tests = [] requested_modules = [] modules_tests = [] cfg_file = "cfg.py" log_dir = "./logs/" verbose = False trace = False restart = False perf = False # parse input parameters try: opts, args = getopt.getopt(sys.argv[1:], "d:f:r:t:l:k:c:m:h:vRPT", [ "dut=", "modules=", "ref=", "tests=", "log-dir=", "cfg=", "key=", "monitor=", "hwsim=" ]) except getopt.GetoptError as err: print(err) usage() sys.exit(2) for option, argument in opts: if option == "-v": verbose = True elif option == "-R": restart = True elif option == "-T": trace = True elif option == "-P": perf = True elif option in ("-d", "--dut"): duts.append(argument) elif option in ("-r", "--ref"): refs.append(argument) elif option in ("-t", "--tests"): requested_tests = re.split('; | |, ', argument) elif option in ("-l", "--log-dir"): log_dir = argument elif option in ("-k", "--key"): filter_keys.append(argument) elif option in ("-m", "--monitor"): monitors.append(argument) elif option in ("-c", "--cfg"): cfg_file = argument elif option in ("-h", "--hwsim"): requested_hwsim_tests = re.split('; | |, ', argument) elif option in ("-f", "--modules"): requested_modules = re.split('; | |, ', argument) else: assert False, "unhandled option" # get env configuration setup_params = config.get_setup_params(cfg_file) devices = config.get_devices(cfg_file) # put logs in log_dir symlink = os.path.join(log_dir, "current") if os.path.exists(symlink): os.unlink(symlink) log_dir = os.path.join(log_dir, time.strftime("%Y_%m_%d_%H_%M_%S")) if not os.path.exists(log_dir): os.makedirs(log_dir) os.symlink(os.path.join("../", log_dir), symlink) # setup restart/trace/perf request setup_params['local_log_dir'] = log_dir setup_params['restart_device'] = restart setup_params['trace'] = trace setup_params['perf'] = perf # configure logger logger.setLevel(logging.DEBUG) stdout_handler = logging.StreamHandler() stdout_handler.setLevel(logging.WARNING) if verbose: stdout_handler.setLevel(logging.DEBUG) logger.addHandler(stdout_handler) formatter = logging.Formatter('%(asctime)s - %(message)s') file_name = os.path.join(log_dir, 'run-tests.log') log_handler = logging.FileHandler(file_name) log_handler.setLevel(logging.DEBUG) log_handler.setFormatter(formatter) logger.addHandler(log_handler) # import available tests tests = [] failed = [] test_modules = [] files = os.listdir(scriptsdir) for t in files: m = re.match(r'(test_.*)\.py$', t) if m: mod = __import__(m.group(1)) test_modules.append(mod.__name__.replace('test_', '', 1)) for key, val in mod.__dict__.items(): if key.startswith("test_"): tests.append(val) test_names = list(set([t.__name__.replace('test_', '', 1) for t in tests])) # import test_* files = os.listdir("../hwsim/") for t in files: m = re.match(r'(test_.*)\.py$', t) if m: mod = __import__(m.group(1)) test_modules.append(mod.__name__.replace('test_', '', 1)) for key, val in mod.__dict__.items(): if key.startswith("test_"): hwsim_tests.append(val) # setup hwsim tests hwsim_tests_to_run = [] if len(requested_hwsim_tests) > 0: # apply filters for filter_key in filter_keys: filtered_tests = [] for hwsim_test in hwsim_tests: if re.search(filter_key, hwsim_test.__name__): filtered_tests.append(hwsim_test) hwsim_tests = filtered_tests # setup hwsim_test we should run if requested_hwsim_tests[0] == "all": hwsim_tests_to_run = hwsim_tests elif requested_hwsim_tests[0] == "remote": hwsim_tests_to_run = [ t for t in hwsim_tests if hasattr(t, "remote_compatible") and t.remote_compatible ] else: for test in requested_hwsim_tests: t = None for tt in hwsim_tests: name = tt.__name__.replace('test_', '', 1) if name == test: t = tt break if not t: logger.warning("hwsim test case: " + test + " NOT-FOUND") continue hwsim_tests_to_run.append(t) # import test_* from modules files = os.listdir("../hwsim/") for t in files: m = re.match(r'(test_.*)\.py$', t) if m: mod = __import__(m.group(1)) if mod.__name__.replace('test_', '', 1) not in requested_modules: continue for key, val in mod.__dict__.items(): if key.startswith("test_"): modules_tests.append(val) if len(requested_modules) > 0: requested_hwsim_tests = modules_tests hwsim_tests_to_run = modules_tests # sort the list test_names.sort() tests.sort(key=lambda t: t.__name__) # print help if requested_tests[0] == "help" and len(requested_hwsim_tests) == 0: usage() print("\nAvailable Devices:") for device in devices: print("\t", device['name']) print("\nAvailable tests:") for test in test_names: print("\t", test) print("\nAvailable hwsim tests:") for hwsim_test in hwsim_tests: print("\t", hwsim_test.__name__.replace('test_', '', 1)) return # show/check devices if requested_tests[0] == "devices": show_devices(devices, setup_params) return # apply filters for filter_key in filter_keys: filtered_tests = [] for test in tests: if re.search(filter_key, test.__name__): filtered_tests.append(test) tests = filtered_tests # setup test we should run tests_to_run = [] if requested_tests[0] == "all": tests_to_run = tests if requested_tests[0] == "help": pass elif requested_tests[0] == "sanity": for test in tests: if test.__name__.startswith("test_sanity_"): tests_to_run.append(test) else: for test in requested_tests: t = None for tt in tests: name = tt.__name__.replace('test_', '', 1) if name == test: t = tt break if not t: logger.warning("test case: " + test + " NOT-FOUND") continue tests_to_run.append(t) # lock devices try: get_devices(devices, duts, refs, monitors) except Exception as e: logger.warning("get devices failed: " + str(e)) logger.info(traceback.format_exc()) put_devices(devices, duts, refs, monitors) return except: logger.warning("get devices failed") logger.info(traceback.format_exc()) put_devices(devices, duts, refs, monitors) return # now run test cases for dut in duts: if len(requested_hwsim_tests) > 0: logger.warning("DUT (apdev): " + str(dut)) else: logger.warning("DUT: " + str(dut)) for ref in refs: if len(requested_hwsim_tests) > 0: logger.warning("REF (dev): " + str(ref)) else: logger.warning("REF: " + str(ref)) for monitor in monitors: logger.warning("MON: " + str(monitor)) # run check_devices at beginning logger.warning("RUN check_devices") try: check_devices(devices, setup_params, refs, duts, monitors) except Exception as e: logger.warning("FAILED: " + str(e)) logger.info(traceback.format_exc()) put_devices(devices, duts, refs, monitors) return except: logger.warning("FAILED") logger.info(traceback.format_exc()) put_devices(devices, duts, refs, monitors) return logger.warning("PASS") test_no = 1 for test in tests_to_run: try: start = datetime.now() setup_params['tc_name'] = test.__name__.replace('test_', '', 1) logger.warning("START - " + setup_params['tc_name'] + " (" + str(test_no) + "/" + str(len(tests_to_run)) + ")") if test.__doc__: logger.info("Test: " + test.__doc__) # run tc res = test(devices, setup_params, refs, duts, monitors) end = datetime.now() logger.warning("PASS (" + res + ") - " + str((end - start).total_seconds()) + "s") except KeyboardInterrupt: put_devices(devices, duts, refs, monitors) raise except TestSkip as e: end = datetime.now() logger.warning("SKIP (" + str(e) + ") - " + str((end - start).total_seconds()) + "s") except Exception as e: end = datetime.now() logger.warning("FAILED (" + str(e) + ") - " + str((end - start).total_seconds()) + "s") logger.info(traceback.format_exc()) failed.append(test.__name__.replace('test_', '', 1)) except: end = datetime.now() logger.warning("FAILED - " + str((end - start).total_seconds()) + "s") logger.info(traceback.format_exc()) failed.append(test.__name__.replace('test_', '', 1)) test_no += 1 test_no = 1 for hwsim_test in hwsim_tests_to_run: try: start = datetime.now() setup_params['tc_name'] = hwsim_test.__name__.replace( 'test_', '', 1) logger.warning("START - " + setup_params['tc_name'] + " (" + str(test_no) + "/" + str(len(hwsim_tests_to_run)) + ")") res = run_hwsim_test(devices, setup_params, refs, duts, monitors, hwsim_test) end = datetime.now() logger.warning("PASS (" + res + ") - " + str((end - start).total_seconds()) + "s") except KeyboardInterrupt: put_devices(devices, duts, refs, monitors) raise except HwsimSkip as e: end = datetime.now() logger.warning("SKIP (" + str(e) + ") - " + str((end - start).total_seconds()) + "s") failed.append(hwsim_test.__name__.replace('test_', '', 1)) except Exception as e: end = datetime.now() logger.warning("FAILED (" + str(e) + ") - " + str((end - start).total_seconds()) + "s") logger.info(traceback.format_exc()) failed.append(hwsim_test.__name__.replace('test_', '', 1)) except: end = datetime.now() logger.warning("FAILED - " + str((end - start).total_seconds()) + "s") logger.info(traceback.format_exc()) failed.append(hwsim_test.__name__.replace('test_', '', 1)) test_no += 1 # unlock devices put_devices(devices, duts, refs, monitors) if len(failed) > 0: logger.warning("Failed test cases:") for test in failed: logger.warning("\t" + test)
def main(): duts = [] refs = [] monitors = [] filter_keys = [] requested_tests = ["help"] requested_hwsim_tests = [] hwsim_tests = [] cfg_file = "cfg.py" log_dir = "./logs/" verbose = False trace = False restart = False perf = False # parse input parameters try: opts, args = getopt.getopt(sys.argv[1:], "d:r:t:l:k:c:m:h:vRPT", ["dut=", "ref=", "tests=", "log-dir=", "cfg=", "key=", "monitor=", "hwsim="]) except getopt.GetoptError as err: print(err) usage() sys.exit(2) for option, argument in opts: if option == "-v": verbose = True elif option == "-R": restart = True elif option == "-T": trace = True elif option == "-P": perf = True elif option in ("-d", "--dut"): duts.append(argument) elif option in ("-r", "--ref"): refs.append(argument) elif option in ("-t", "--tests"): requested_tests = re.split('; | |, ', argument) elif option in ("-l", "--log-dir"): log_dir = argument elif option in ("-k", "--key"): filter_keys.append(argument) elif option in ("-m", "--monitor"): monitors.append(argument) elif option in ("-c", "--cfg"): cfg_file = argument elif option in ("-h", "--hwsim"): requested_hwsim_tests = re.split('; | |, ', argument) else: assert False, "unhandled option" # get env configuration setup_params = config.get_setup_params(cfg_file) devices = config.get_devices(cfg_file) # put logs in log_dir symlink = os.path.join(log_dir, "current"); if os.path.exists(symlink): os.unlink(symlink) log_dir = os.path.join(log_dir, time.strftime("%Y_%m_%d_%H_%M_%S")) if not os.path.exists(log_dir): os.makedirs(log_dir) os.symlink(os.path.join("../", log_dir), symlink) # setup restart/trace/perf request setup_params['local_log_dir'] = log_dir setup_params['restart_device'] = restart setup_params['trace'] = trace setup_params['perf'] = perf # configure logger logger.setLevel(logging.DEBUG) stdout_handler = logging.StreamHandler() stdout_handler.setLevel(logging.WARNING) if verbose: stdout_handler.setLevel(logging.DEBUG) logger.addHandler(stdout_handler) formatter = logging.Formatter('%(asctime)s - %(message)s') file_name = os.path.join(log_dir, 'run-tests.log') log_handler = logging.FileHandler(file_name) log_handler.setLevel(logging.DEBUG) log_handler.setFormatter(formatter) logger.addHandler(log_handler) # import available tests tests = [] failed = [] test_modules = [] files = os.listdir(scriptsdir) for t in files: m = re.match(r'(test_.*)\.py$', t) if m: mod = __import__(m.group(1)) test_modules.append(mod.__name__.replace('test_', '', 1)) for key,val in mod.__dict__.iteritems(): if key.startswith("test_"): tests.append(val) test_names = list(set([t.__name__.replace('test_', '', 1) for t in tests])) # import test_* files = os.listdir("../hwsim/") for t in files: m = re.match(r'(test_.*)\.py$', t) if m: mod = __import__(m.group(1)) test_modules.append(mod.__name__.replace('test_', '', 1)) for key,val in mod.__dict__.iteritems(): if key.startswith("test_"): hwsim_tests.append(val) # setup hwsim tests hwsim_tests_to_run = [] if len(requested_hwsim_tests) > 0: # apply filters for filter_key in filter_keys: filtered_tests = [] for hwsim_test in hwsim_tests: if re.search(filter_key, hwsim_test.__name__): filtered_tests.append(hwsim_test) hwsim_tests = filtered_tests # setup hwsim_test we should run if requested_hwsim_tests[0] == "all": hwsim_tests_to_run = hwsim_tests elif requested_hwsim_tests[0] == "remote": hwsim_tests_to_run = [t for t in hwsim_tests if hasattr(t, "remote_compatible") and t.remote_compatible] else: for test in requested_hwsim_tests: t = None for tt in hwsim_tests: name = tt.__name__.replace('test_', '', 1) if name == test and tt.func_code.co_argcount <= 2: t = tt break if not t: logger.warning("hwsim test case: " + test + " NOT-FOUND") continue hwsim_tests_to_run.append(t) # sort the list test_names.sort() tests.sort() # print help if requested_tests[0] == "help" and len(requested_hwsim_tests) == 0: usage() print "\nAvailable Devices:" for device in devices: print "\t", device['name'] print "\nAvailable tests:" for test in test_names: print "\t", test print "\nAvailable hwsim tests:" for hwsim_test in hwsim_tests: print "\t", hwsim_test.__name__.replace('test_', '', 1) return # show/check devices if requested_tests[0] == "devices": show_devices(devices, setup_params) return # apply filters for filter_key in filter_keys: filtered_tests = [] for test in tests: if re.search(filter_key, test.__name__): filtered_tests.append(test) tests = filtered_tests # setup test we should run tests_to_run = [] if requested_tests[0] == "all": tests_to_run = tests if requested_tests[0] == "help": pass elif requested_tests[0] == "sanity": for test in tests: if test.__name__.startswith("test_sanity_"): tests_to_run.append(test) else: for test in requested_tests: t = None for tt in tests: name = tt.__name__.replace('test_', '', 1) if name == test: t = tt break if not t: logger.warning("test case: " + test + " NOT-FOUND") continue tests_to_run.append(t) # lock devices try: get_devices(devices, duts, refs, monitors) except Exception, e: logger.warning("get devices failed: " + str(e)) logger.info(traceback.format_exc()) put_devices(devices, duts, refs, monitors) return