1 # Copyright 2016 The Chromium OS Authors. All rights reserved. 2 # Use of this source code is governed by a BSD-style license that can be 3 # found in the LICENSE file. 4 5 AUTHOR = 'krisr, tienchang, bmahadev' 6 NAME = 'network_WiFi_SuspendStress.stress_WPA2' 7 TIME = 'LONG' 8 TEST_TYPE = 'Server' 9 DEPENDENCIES = 'servo, wificell' 10 ATTRIBUTES = 'suite:wifi_stress' 11 12 DOC = """ 13 This test uses servo to simulate lid close and open events and checks that the 14 wifi adapter is brought back up and connects to a WPA2 network. 15 """ 16 17 from autotest_lib.client.common_lib.cros.network import xmlrpc_datatypes 18 from autotest_lib.client.common_lib.cros.network import xmlrpc_datatypes 19 from autotest_lib.client.common_lib.cros.network import xmlrpc_security_types 20 from autotest_lib.server.cros.network import hostap_config 21 from autotest_lib.server import utils 22 23 args_dict = utils.args_to_dict(args) 24 servo_args = hosts.CrosHost.get_servo_arguments(args_dict) 25 26 def run(machine): 27 try: 28 host = hosts.create_host(machine, servo_args=servo_args) 29 except IOError: 30 # We probably failed to connect to the servo. That's probably 31 # because the cell doesn't have a servo. 32 # 33 # Ideally, we'd raise TestNAError here, and be done. But the 34 # test isn't actually running yet. So we instantiate a host 35 # without servo, and let the network_WiFi_SuspendStress 36 # instance report the TestNAError. 37 host = hosts.create_host(machine) 38 39 wpa_config = xmlrpc_security_types.WPAConfig( 40 psk='chromeos', 41 wpa_mode=xmlrpc_security_types.WPAConfig.MODE_PURE_WPA2, 42 wpa2_ciphers=[xmlrpc_security_types.WPAConfig.CIPHER_CCMP]) 43 ap_config = hostap_config.HostapConfig( 44 frequency=2412, 45 mode=hostap_config.HostapConfig.MODE_11N_PURE, 46 security_config=wpa_config) 47 assoc_params = xmlrpc_datatypes.AssociationParameters() 48 assoc_params.security_config = wpa_config 49 configurations = [(ap_config, assoc_params)] 50 job.run_test('network_WiFi_SuspendStress', 51 host=host, 52 tag=NAME.split('.')[1], 53 suspends=690, 54 raw_cmdline_args=args, 55 additional_params=configurations) 56 57 parallel_simple(run, machines) 58