1 # Copyright (c) 2015 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.WPA2' 7 TIME = 'MEDIUM' 8 TEST_TYPE = 'Server' 9 DEPENDENCIES = 'servo, wificell' 10 ATTRIBUTES = 'suite:wifi_matfunc, suite:wifi_matfunc_bcm4371' 11 SUITE = 'wifi_matfunc, wifi_matfunc_bcm4371' 12 13 DOC = """ 14 This test uses servo to simulate lid close and open events and checks that the 15 wifi adapter is brought back up and connects to a WPA network. 16 """ 17 18 from autotest_lib.client.common_lib.cros.network import xmlrpc_datatypes 19 from autotest_lib.client.common_lib.cros.network import xmlrpc_datatypes 20 from autotest_lib.client.common_lib.cros.network import xmlrpc_security_types 21 from autotest_lib.server.cros.network import hostap_config 22 from autotest_lib.server import utils 23 24 args_dict = utils.args_to_dict(args) 25 servo_args = hosts.CrosHost.get_servo_arguments(args_dict) 26 27 def run(machine): 28 try: 29 host = hosts.create_host(machine, servo_args=servo_args) 30 except IOError: 31 # We probably failed to connect to the servo. That's probably 32 # because the cell doesn't have a servo. 33 # 34 # Ideally, we'd raise TestNAError here, and be done. But the 35 # test isn't actually running yet. So we instantiate a host 36 # without servo, and let the network_WiFi_SuspendStress 37 # instance report the TestNAError. 38 host = hosts.create_host(machine) 39 40 wpa_config = xmlrpc_security_types.WPAConfig( 41 psk='chromeos', 42 wpa_mode=xmlrpc_security_types.WPAConfig.MODE_PURE_WPA2, 43 wpa2_ciphers=[xmlrpc_security_types.WPAConfig.CIPHER_CCMP]) 44 ap_config = hostap_config.HostapConfig( 45 frequency=2412, 46 mode=hostap_config.HostapConfig.MODE_11G, 47 security_config=wpa_config) 48 assoc_params = xmlrpc_datatypes.AssociationParameters() 49 assoc_params.security_config = wpa_config 50 configurations = [(ap_config, assoc_params)] 51 job.run_test('network_WiFi_SuspendStress', 52 host=host, 53 tag=NAME.split('.')[1], 54 suspends=5, 55 raw_cmdline_args=args, 56 additional_params=configurations) 57 58 parallel_simple(run, machines) 59