def initialize(self, host, cmdline_args): dict_args = utils.args_to_dict(cmdline_args) super(firmware_ChipFwUpdate, self).initialize(host, cmdline_args) self.new_bios_path = dict_args['bios'] if 'bios' in dict_args else None self.clear_set_gbb_flags( vboot.GBB_FLAG_DISABLE_EC_SOFTWARE_SYNC | vboot.GBB_FLAG_DISABLE_PD_SOFTWARE_SYNC, 0) self.dut_bios_path = None self.cbfs_work_dir = None # set of chip types found in CBFS self.cbfs_chip_types = set() # dict of chip FW updates from the cmd line self.req_chip_updates = {} # see if comand line specified new firmware blobs # for chips we know about for chip in chip_utils.chip_id_map.itervalues(): chip_name = chip.chip_name if chip_name not in dict_args: continue chip_file = dict_args[chip_name] if not os.path.exists(chip_file): raise error.TestError('file %s not found' % chip_file) c = chip() c.set_from_file(chip_file) if chip_name in self.req_chip_updates: raise error.TestError('multiple %s args' % chip_name) logging.info('request chip %s fw 0x%02x from command line', c.chip_name, c.fw_ver) self.req_chip_updates[chip_name] = c
def initialize(self, host, cmdline_args): super(firmware_CompareChipFwToShellBall, self).initialize(host, cmdline_args) dict_args = utils.args_to_dict(cmdline_args) self.new_bios_path = dict_args['bios'] if 'bios' in dict_args else None self.cbfs_work_dir = None self.dut_bios_path = None
def initialize(self, host, cmdline_args): dict_args = utils.args_to_dict(cmdline_args) super(firmware_FWupdate, self).initialize(host, cmdline_args) if not set(('new_ec', 'new_bios')).issubset(set(dict_args)): raise error.TestError('Missing new_ec and/or new_bios argument') self.new_ec = dict_args['new_ec'] self.new_bios = dict_args['new_bios'] if not os.path.isfile(self.new_ec) or not os.path.isfile(self.new_bios): raise error.TestError('Failed to locate ec or bios file') self.new_pd = '' if 'new_pd' in dict_args: self.new_pd = dict_args['new_pd'] if not os.path.isfile(self.new_pd): raise error.TestError('Failed to locate pd file') logging.info('EC=%s BIOS=%s PD=%s', self.new_ec, self.new_bios, self.new_pd) self.mode = 'recovery' if 'mode' in dict_args: self.mode = dict_args['mode'] if self.mode == 'recovery': self.switcher.setup_mode('normal') # Set device to normal mode elif self.mode == 'factory': self.switcher.setup_mode('dev') # Set device to dev mode else: raise error.TestError('Unknown mode:%s' % self.mode)
def initialize(self, host, cmdline_args): # Parse arguments from command line dict_args = utils.args_to_dict(cmdline_args) self.faft_iterations = int(dict_args.get('faft_iterations', 1)) super(firmware_ConsecutiveLidSwitch, self).initialize(host, cmdline_args) self.setup_usbkey(usbkey=False)
def initialize(self, host, cmdline_args): # Parse arguments from command line dict_args = utils.args_to_dict(cmdline_args) self.faft_iterations = int(dict_args.get('faft_iterations', 1)) super(firmware_DevModeStress, self).initialize(host, cmdline_args) self.switcher.setup_mode('dev') self.setup_usbkey(usbkey=False)
def initialize(self, host, cmdline_args): # This test assume the system already have the latest RW from # shellball. If you not sure, run chromeos-firmware --mode=factory. # Device should have WP disable. # Parse arguments from command line dict_args = utils.args_to_dict(cmdline_args) super(platform_Flashrom, self).initialize(host, cmdline_args)
def initialize(self, host, cmdline_args, dev_mode=False): # Parse arguments from command line dict_args = utils.args_to_dict(cmdline_args) self.faft_iterations = int(dict_args.get('faft_iterations', 1)) super(firmware_ConsecutiveBootPowerButton, self).initialize(host, cmdline_args) self.switcher.setup_mode('dev' if dev_mode else 'normal') self.setup_usbkey(usbkey=False)
def initialize(self, host, cmdline_args, dev_mode): # Parse arguments from command line dict_args = utils.args_to_dict(cmdline_args) self.faft_iterations = int(dict_args.get('faft_iterations', 1)) super(firmware_CgptStress, self).initialize(host, cmdline_args) self.backup_cgpt_attributes() self.switcher.setup_mode('dev' if dev_mode else 'normal') self.setup_usbkey(usbkey=False) self.setup_kernel('a')
def initialize(self, host, cmdline_args, dev_mode=False): # Parse arguments from command line dict_args = utils.args_to_dict(cmdline_args) self.faft_iterations = int(dict_args.get('faft_iterations', 1)) self.faft_waitup_time = int(dict_args.get('faft_waitup_time', 0)) self.faft_localrun = int(dict_args.get('faft_localrun', 0)) super(firmware_ConsecutiveBoot, self).initialize(host, cmdline_args) self.switcher.setup_mode('dev' if dev_mode else 'normal') if dev_mode: self.clear_set_gbb_flags(0, vboot.GBB_FLAG_DEV_SCREEN_SHORT_DELAY) self.setup_usbkey(usbkey=False)
def warmup(self, raw_cmdline_args): """Possibly parse the router hostname from the commandline. @param raw_cmdline_args raw input from autotest. """ cmdline_args = utils.args_to_dict(raw_cmdline_args) logging.info('Running wifi test with commandline arguments: %r', cmdline_args) self._router_hostname_from_cmdline = cmdline_args.get( wifi_test_context_manager.WiFiTestContextManager. \ CMDLINE_ROUTER_ADDR)
def test_matches(self): result = utils.args_to_dict( ['aBc:DeF', 'SyS=DEf', 'XY_Z:', 'F__o0O=', 'B8r:=:=', '_bAZ_=:=:']) self.assertEqual( result, { 'abc': 'DeF', 'sys': 'DEf', 'xy_z': '', 'f__o0o': '', 'b8r': '=:=', '_baz_': ':=:' })
def test_unmatches(self): # Temporarily shut warning messages from args_to_dict() when an argument # doesn't match its pattern. logger = logging.getLogger() saved_level = logger.level logger.setLevel(logging.ERROR) try: result = utils.args_to_dict(['ab-c:DeF', '--SyS=DEf', 'a*=b', 'a*b', ':VAL', '=VVV', 'WORD']) self.assertEqual({}, result) finally: # Restore level. logger.setLevel(saved_level)
def test_unmatches(self): # Temporarily shut warning messages from args_to_dict() when an argument # doesn't match its pattern. logger = logging.getLogger() saved_level = logger.level logger.setLevel(logging.ERROR) try: result = utils.args_to_dict([ 'ab-c:DeF', '--SyS=DEf', 'a*=b', 'a*b', ':VAL', '=VVV', 'WORD' ]) self.assertEqual({}, result) finally: # Restore level. logger.setLevel(saved_level)
def get_chart_address(host_address, args): """Get address of chart tablet from commandline args or mapping logic in test lab. @param host_address: a list of hostname strings. @param args: a dict parse from commandline args. @return: A list of strings for chart tablet addresses. """ address = utils.args_to_dict(args).get('chart') if address is not None: return address.split(',') elif utils.is_in_container(): return [utils.get_lab_chart_address(host) for host in host_address] else: return None
def initialize(self, host, cmdline_args, dev_mode=False): # Parse arguments from command line dict_args = utils.args_to_dict(cmdline_args) super(firmware_Mosys, self).initialize(host, cmdline_args) self.switcher.setup_mode('dev' if dev_mode else 'normal') # a list contain failed execution. self.failed_command = [] # Get a list of available mosys commands. lines = self.run_cmd('mosys help') self.command_list = [] cmdlist_start = False for line in lines: if cmdlist_start: cmdlst = re.split('\s+', line) if len(cmdlst) > 2: self.command_list.append(cmdlst[1]) elif 'Commands:' in line: cmdlist_start = True logging.info('Available commands: %s', ' '.join(self.command_list))
def warmup(self, host, raw_cmdline_args, additional_params=None): """ Use the additional_params argument to pass in custom test data from control file to reuse test logic. This object will be passed down via parse_additional_arguments. @param host host object representing the client DUT. @param raw_cmdline_args raw input from autotest. @param additional_params object passed in from control file. """ cmdline_args = utils.args_to_dict(raw_cmdline_args) logging.info('Running wifi test with commandline arguments: %r', cmdline_args) self._wifi_context = wifi_test_context_manager.WiFiTestContextManager( self.__class__.__name__, host, cmdline_args, self.debugdir) self._wifi_context.setup() self.parse_additional_arguments(cmdline_args, additional_params) msg = '======= WiFi autotest setup complete. Starting test... =======' self._wifi_context.client.shill_debug_log(msg)
def test_matches(self): result = utils.args_to_dict(['aBc:DeF', 'SyS=DEf', 'XY_Z:', 'F__o0O=', 'B8r:=:=', '_bAZ_=:=:']) self.assertEqual(result, {'abc':'DeF', 'sys':'DEf', 'xy_z':'', 'f__o0o':'', 'b8r':'=:=', '_baz_':':=:'})
def test_no_args(self): result = utils.args_to_dict([]) self.assertEqual({}, result)
def initialize(self, host, cmdline_args): dict_args = utils.args_to_dict(cmdline_args) self.faft_iterations = int(dict_args.get('faft_iterations', 1)) super(platform_S0ixCycle, self).initialize(host, cmdline_args) self.switcher.setup_mode('normal')