| # Copyright (c) 2015 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import logging |
| import time |
| |
| from autotest_lib.client.common_lib import error |
| from autotest_lib.client.common_lib.cros.network import xmlrpc_datatypes |
| from autotest_lib.server import site_linux_system |
| from autotest_lib.server.cros.network import hostap_config |
| from autotest_lib.server.cros.network import wifi_cell_test_base |
| from autotest_lib.server.cros.network import wifi_client |
| |
| |
| class network_WiFi_DisconnectReason(wifi_cell_test_base.WiFiCellTestBase): |
| """Verify the client disconnects from an AP and read (but not verify) |
| the supplicant DisconnectReason for various scenarios.""" |
| version = 1 |
| |
| INITIAL_CHANNEL = 64 |
| ALT_CHANNEL = 6 |
| CHANNEL_SWITCH_ATTEMPTS = 5 |
| CHANNEL_SWITCH_WAIT_TIME_SEC = 3 |
| |
| def run_once(self, disconnect_trigger, req_caps=None): |
| """Sets up a router, connects to it, pings it and disables it to trigger |
| disconnect.""" |
| configuration = hostap_config.HostapConfig( |
| channel=self.INITIAL_CHANNEL, |
| mode=hostap_config.HostapConfig.MODE_11A, |
| spectrum_mgmt_required=True) |
| if req_caps is None: |
| req_caps = [] |
| self.context.router.require_capabilities(req_caps) |
| self.context.configure(configuration) |
| |
| if site_linux_system.LinuxSystem.CAPABILITY_MULTI_AP in req_caps: |
| # prep alternate Access Point |
| alt_ap_config = hostap_config.HostapConfig( |
| channel=self.ALT_CHANNEL, |
| mode=hostap_config.HostapConfig.MODE_11N_MIXED) |
| self.context.configure(alt_ap_config, multi_interface=True) |
| alt_assoc_params = xmlrpc_datatypes.AssociationParameters() |
| alt_assoc_params.ssid = self.context.router.get_ssid(instance=1) |
| |
| assoc_params = xmlrpc_datatypes.AssociationParameters() |
| assoc_params.ssid = self.context.router.get_ssid(instance=0) |
| self.context.assert_connect_wifi(assoc_params) |
| self.context.assert_ping_from_dut() |
| |
| with self.context.client.assert_disconnect_event(): |
| if disconnect_trigger == 'AP gone': |
| self.context.router.deconfig() |
| elif disconnect_trigger == 'deauth client': |
| self.context.router.deauth_client(self.context.client.wifi_mac) |
| elif disconnect_trigger == 'AP send channel switch': |
| for _ in range(self.CHANNEL_SWITCH_ATTEMPTS): |
| self.context.router.send_management_frame_on_ap( |
| 'channel_switch', |
| self.ALT_CHANNEL) |
| time.sleep(self.CHANNEL_SWITCH_WAIT_TIME_SEC) |
| elif disconnect_trigger == 'switch AP': |
| self.context.assert_connect_wifi(alt_assoc_params) |
| elif disconnect_trigger == 'disable client wifi': |
| self.context.client.set_device_enabled( |
| self.context.client.wifi_if, False) |
| else: |
| raise error.TestError('unknown test mode: %s' % |
| disconnect_trigger) |
| time.sleep(wifi_client.DISCONNECT_WAIT_TIME_SECONDS) |
| |
| disconnect_reasons = self.context.client.get_disconnect_reasons() |
| if disconnect_reasons is None or len(disconnect_reasons) == 0: |
| raise error.TestFail('supplicant DisconnectReason not logged') |
| for entry in disconnect_reasons: |
| logging.info("DisconnectReason: %s", entry); |