1 # Copyright 2014 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 from autotest_lib.client.common_lib import error 6 from autotest_lib.client.common_lib.cros.network import apmanager_constants 7 from autotest_lib.client.common_lib.cros.network import xmlrpc_datatypes 8 from autotest_lib.server.cros.network import apmanager_service_provider 9 from autotest_lib.server.cros.network import wifi_cell_test_base 10 11 12 class apmanager_SimpleConnect(wifi_cell_test_base.WiFiCellTestBase): 13 """Test that the DUT can connect to an AP created by apmanager.""" 14 version = 1 15 16 XMLRPC_BRINGUP_TIMEOUT_SECONDS = 60 17 18 def parse_additional_arguments(self, commandline_args, additional_params): 19 """Hook into super class to take control files parameters. 20 21 @param commandline_args dict of parsed parameters from the autotest. 22 @param additional_params dict of AP configuration parameters. 23 24 """ 25 self._configurations = additional_params 26 27 28 def run_once(self): 29 """Sets up a router, connects to it, pings it.""" 30 # Setup bridge mode test environments if AP is configured to operate in 31 # bridge mode. 32 if (apmanager_constants.CONFIG_OPERATION_MODE in self._configurations 33 and self._configurations[apmanager_constants.CONFIG_OPERATION_MODE] 34 == apmanager_constants.OPERATION_MODE_BRIDGE): 35 # Setup DHCP server on the other side of the bridge. 36 self.context.router.setup_bridge_mode_dhcp_server() 37 self._configurations[apmanager_constants.CONFIG_BRIDGE_INTERFACE] =\ 38 self.context.router.get_bridge_interface() 39 40 ssid = self.context.router.build_unique_ssid() 41 self._configurations[apmanager_constants.CONFIG_SSID] = ssid 42 with apmanager_service_provider.ApmanagerServiceProvider( 43 self.context.router, self._configurations): 44 assoc_params = xmlrpc_datatypes.AssociationParameters() 45 assoc_params.ssid = ssid 46 self.context.assert_connect_wifi(assoc_params) 47 self.context.assert_ping_from_server() 48 # AP is terminated, wait for client to become disconnected. 49 success, state, elapsed_seconds = \ 50 self.context.client.wait_for_service_states( 51 ssid, ( 'idle', ), 30) 52 if not success: 53 raise error.TestFail('Failed to disconnect from %s after AP was ' 54 'terminated for %f seconds (state=%s)' % 55 (ssid, elapsed_seconds, state)) 56
