76     host = hosts.create_host(machine_dict)
78             **servo_host.get_servo_args_for_host(host))
80     host.set_servo_host(servohost)
81     host.servo.uart_logs_dir = logs_dir
83         yield host
85         host.close()
102 def download_image_to_servo_usb(host, build):  argument
108     _, update_url = host.stage_image_for_servo(build)
109     host.servo.image_to_servo_usb(update_url)
112 def try_reset_by_servo(host):  argument
121     host.servo.get_power_state_controller().reset()
124     if not host.wait_up(timeout=host.BOOT_TIMEOUT):
126             'DUT failed to come back after %d seconds' % host.BOOT_TIMEOUT)
129 def power_cycle_via_servo(host, recover_src=False):  argument
137         logging.info('Shutting down %s from via ssh.', host.hostname)
138         host.halt()
143         host.servo.set_servo_v4_role('src')
146     host.servo.get_power_state_controller().power_off()
147     host.servo.switch_usbkey('off')
148     time.sleep(host.SHUTDOWN_TIMEOUT)
157     time.sleep(host.SHUTDOWN_TIMEOUT)
158     host.servo.get_power_state_controller().power_on()
161     if not host.wait_up(timeout=host.BOOT_TIMEOUT):
163                                   host.BOOT_TIMEOUT)
166 def verify_battery_status(host):  argument
175     host_info = host.host_info_store.get()
179     power_info = host.get_power_supply_info()
182     status = host.run(cmd, timeout=30, ignore_status=True).stdout.strip()
194         power_info = host.get_power_supply_info()
229 def verify_servo(host):  argument
235     host_info = host.host_info_store.get()
239     servo_host = host._servo_host
243     if host._servo_host.is_servo_topology_supported():
244         servo_topology = host._servo_host.get_topology()
249     servo_type = host.servo.get_servo_type()
255     if not host.is_servo_in_working_state():
260     host._set_servo_topology()
264 def verify_ccd_testlab_enable(host):  argument
274     host_info = host.host_info_store.get()
280     if host.servo and host.servo.get_main_servo_device() == 'ccd_cr50':
281         if not host.servo.has_control('cr50_testlab'):
286         status = host.servo.get('cr50_testlab')
295 def verify_labstation_RPM_config_unsafe(host):  argument
310     host_info = host.host_info_store.get()
320         logging.info("intentionally skipping labstation %s", host.hostname)
330     host.rpm_power_off_and_wait()
331     host.rpm_power_on_and_wait()
335 def verify_boot_into_rec_mode(host):  argument
348         logging.info('Shutting down %s from via ssh.', host.hostname)
349         host.halt()
353     host.servo.get_power_state_controller().power_off()
354     time.sleep(host.SHUTDOWN_TIMEOUT)
356     need_snk = host.require_snk_mode_in_recovery()
357     host.servo.boot_in_recovery_mode(snk_mode=need_snk)
359         if not host.wait_up(timeout=host.USB_BOOT_TIMEOUT):
364             host.run('chromeos-tpm-recovery')
372             host.servo.set_servo_v4_role('src')
376     power_cycle_via_servo(host, recover_src=need_snk)
380 def install_test_image(host):  argument
389     servo = host.servo
394     time.sleep(host.SHUTDOWN_TIMEOUT)
399     boot_deadline = time.time() + host.BOOT_TIMEOUT
403         if host.ping_wait_up(timeout=5):
408     host.run('chromeos-install --yes', timeout=host.INSTALL_TIMEOUT)
412         host.reboot()
415         try_reset_by_servo(host)
419 def reinstall_test_image(host):  argument
427     host.servo_install()
430 def flash_firmware_using_servo(host, build):  argument
437     host.firmware_install(build)
440 def install_firmware(host):  argument
459         host.run('flashrom -p %s --wp-disable' % fprom,
463     pid = _start_firmware_update(host, fw_update_log)
464     _wait_firmware_update_process(host, pid)
465     _check_firmware_update_result(host, fw_update_log)
468         host.reboot()
471         try_reset_by_servo(host)
477     host.run('/usr/share/vboot/bin/set_gbb_flags.sh 0',
479     host.run('crossystem disable_dev_request=1',
484         host.reboot()
489         try_reset_by_servo(host)
493 def _start_firmware_update(host, result_file):  argument
512     return host.run_background(';'.join(cmd))
515 def _wait_firmware_update_process(host, pid, timeout=_FIRMWARE_UPDATE_TIMEOUT):  argument
524             lambda: host.run('ps -f -p %s' % pid, timeout=20).exit_status,
533         if not host.wait_up(timeout=host.USB_BOOT_TIMEOUT):
538 def _check_firmware_update_result(host, result_file):  argument
546     result = host.run('cat %s' % result_file)
572 def setup_hwid_and_serialnumber(host):  argument
577     if not hasattr(host, 'host_info_store'):
578         raise Exception('%s does not have host_info_store' % host.hostname)
580     info = host.host_info_store.get()
581     hwid = host.run('crossystem hwid', ignore_status=True).stdout
582     serial_number = host.run('vpd -g serial_number', ignore_status=True).stdout
587                 host.hostname)
590                         host.hostname)
592         raise Exception('Failed to retrieve HWID from host %s' % host.hostname)
596     if info != host.host_info_store.get():
597         host.host_info_store.commit(info)