| # |
| # Copyright 2017 - The Android Open Source Project |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| import time |
| from acts import asserts |
| from acts import utils |
| from acts import signals |
| from acts.keys import Config |
| from acts.test_decorators import test_tracker_info |
| from acts_contrib.test_utils.net import connectivity_const as cconsts |
| from acts_contrib.test_utils.wifi import wifi_test_utils as wutils |
| from acts_contrib.test_utils.wifi.WifiBaseTest import WifiBaseTest |
| from acts_contrib.test_utils.wifi.aware import aware_test_utils as autils |
| import os |
| |
| WifiEnums = wutils.WifiEnums |
| |
| |
| class WifiRoamingTest(WifiBaseTest): |
| |
| def setup_class(self): |
| """Configure the required networks for testing roaming.""" |
| super().setup_class() |
| |
| self.dut = self.android_devices[0] |
| self.dut_client = self.android_devices[1] |
| req_params = ["roaming_attn"] |
| self.unpack_userparams(req_param_names=req_params,) |
| self.country_code = wutils.WifiEnums.CountryCode.US |
| if hasattr(self, "country_code_file"): |
| if isinstance(self.country_code_file, list): |
| self.country_code_file = self.country_code_file[0] |
| if not os.path.isfile(self.country_code_file): |
| self.country_code_file = os.path.join( |
| self.user_params[Config.key_config_path.value], |
| self.country_code_file) |
| self.country_code = utils.load_config( |
| self.country_code_file)["country"] |
| |
| if "AccessPoint" in self.user_params: |
| self.legacy_configure_ap_and_start(ap_count=2) |
| |
| def teardown_class(self): |
| self.dut.ed.clear_all_events() |
| if "AccessPoint" in self.user_params: |
| del self.user_params["reference_networks"] |
| del self.user_params["open_network"] |
| |
| def setup_test(self): |
| super().setup_test() |
| self.dut.ed.clear_all_events() |
| self.dut.droid.wakeLockAcquireBright() |
| self.dut.droid.wakeUpNow() |
| |
| def teardown_test(self): |
| super().teardown_test() |
| self.dut.droid.wakeLockRelease() |
| self.dut.droid.goToSleepNow() |
| wutils.reset_wifi(self.dut) |
| wutils.set_attns(self.attenuators, "default") |
| for ad in self.android_devices: |
| wutils.set_wifi_country_code(ad, self.country_code) |
| if "OpenWrtAP" in self.user_params: |
| for ap in self.access_points: |
| ap.close() |
| |
| ### Helper Methods ### |
| |
| def register_network_callback_for_internet(self): |
| self.dut.log.debug("Registering network callback for wifi internet" |
| "connectivity") |
| network_request = { |
| cconsts.NETWORK_CAP_TRANSPORT_TYPE_KEY : |
| cconsts.NETWORK_CAP_TRANSPORT_WIFI, |
| cconsts.NETWORK_CAP_CAPABILITY_KEY : |
| [cconsts.NETWORK_CAP_CAPABILITY_INTERNET] |
| } |
| key = self.dut.droid.connectivityRegisterNetworkCallback(network_request) |
| return key |
| |
| def generate_wifi_info(self, network): |
| return { |
| WifiEnums.SSID_KEY : network[WifiEnums.SSID_KEY], |
| # We need to use "BSSID" in WifiInfo map, also need to use lower |
| # chars for bssid. |
| WifiEnums.BSSID_KEY : network["bssid"].lower() |
| } |
| |
| def roaming_from_AP1_and_AP2(self, AP1_network, AP2_network): |
| """Test roaming between two APs. |
| |
| Args: |
| AP1_network: AP-1's network information. |
| AP2_network: AP-2's network information. |
| |
| Steps: |
| 1. Make AP1 visible, AP2 not visible. |
| 2. Connect to AP1's ssid. |
| 3. Make AP1 not visible, AP2 visible. |
| 4. Expect DUT to roam to AP2. |
| 5. Validate connection information and ping. |
| """ |
| network_cb_key = None |
| if self.dut.droid.isSdkAtLeastS(): |
| network_cb_key = self.register_network_callback_for_internet() |
| wutils.set_attns(self.attenuators, "AP1_on_AP2_off", self.roaming_attn) |
| wifi_config = AP1_network.copy() |
| wifi_config.pop("bssid") |
| wutils.connect_to_wifi_network(self.dut, wifi_config) |
| if network_cb_key is not None: |
| self.dut.log.info("Waiting for onAvailable and " |
| "onCapabilitiesChanged after connection") |
| # Ensure that the connection completed and we got the ON_AVAILABLE |
| # callback. |
| autils.wait_for_event_with_keys( |
| self.dut, |
| cconsts.EVENT_NETWORK_CALLBACK, |
| 20, |
| (cconsts.NETWORK_CB_KEY_ID, network_cb_key), |
| (cconsts.NETWORK_CB_KEY_EVENT, cconsts.NETWORK_CB_AVAILABLE)) |
| autils.wait_for_event_with_keys( |
| self.dut, cconsts.EVENT_NETWORK_CALLBACK, 10, |
| (cconsts.NETWORK_CB_KEY_ID, network_cb_key), |
| (cconsts.NETWORK_CB_KEY_EVENT, |
| cconsts.NETWORK_CB_CAPABILITIES_CHANGED), |
| (cconsts.NETWORK_CB_KEY_TRANSPORT_INFO, |
| self.generate_wifi_info(AP1_network))) |
| self.log.info("Roaming from %s to %s", AP1_network, AP2_network) |
| wutils.trigger_roaming_and_validate( |
| self.dut, self.attenuators, "AP1_off_AP2_on", AP2_network, |
| self.roaming_attn) |
| if network_cb_key is not None: |
| self.dut.log.info("Waiting for onCapabilitiesChanged after" |
| " roaming") |
| # Ensure that the roaming complete triggered a capabilities change |
| # with the new bssid. |
| autils.wait_for_event_with_keys( |
| self.dut, cconsts.EVENT_NETWORK_CALLBACK, 10, |
| (cconsts.NETWORK_CB_KEY_ID, network_cb_key), |
| (cconsts.NETWORK_CB_KEY_EVENT, |
| cconsts.NETWORK_CB_CAPABILITIES_CHANGED), |
| (cconsts.NETWORK_CB_KEY_TRANSPORT_INFO, |
| self.generate_wifi_info(AP2_network))) |
| |
| ### Test Cases ### |
| |
| @test_tracker_info(uuid="db8a46f9-713f-4b98-8d9f-d36319905b0a") |
| def test_roaming_between_AP1_to_AP2_open_2g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(open_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| self.log.info("BSSID map: %s" % self.bssid_map) |
| ap1_network = self.open_network[0]["2g"] |
| ap2_network = self.open_network[1]["2g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["2g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["2g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="0db67d9b-6ea9-4f40-acf2-155c4ecf9dc5") |
| def test_roaming_between_AP1_to_AP2_open_5g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(open_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.open_network[0]["5g"] |
| ap2_network = self.open_network[1]["5g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["5g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["5g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="eabc7319-d962-4bef-b679-725e9ff00420") |
| def test_roaming_between_AP1_to_AP2_psk_2g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(wpa_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.reference_networks[0]["2g"] |
| ap2_network = self.reference_networks[1]["2g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["2g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["2g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="1cf9c681-4ff0-45c1-9719-f01629f6a7f7") |
| def test_roaming_between_AP1_to_AP2_psk_5g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(wpa_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.reference_networks[0]["5g"] |
| ap2_network = self.reference_networks[1]["5g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["5g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["5g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="a28f7d2e-fae4-4e66-b633-7ee59f8b46e0") |
| def test_roaming_between_AP1_to_AP2_owe_2g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(owe_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.owe_networks[0]["2g"] |
| ap2_network = self.owe_networks[1]["2g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["2g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["2g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="3c39110a-9336-4abd-b885-acbba85dc10d") |
| def test_roaming_between_AP1_to_AP2_owe_5g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(owe_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.owe_networks[0]["5g"] |
| ap2_network = self.owe_networks[1]["5g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["5g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["5g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="68b2baf6-162a-44f2-a00d-4973e5ac9471") |
| def test_roaming_between_AP1_to_AP2_sae_2g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(sae_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.sae_networks[0]["2g"] |
| ap2_network = self.sae_networks[1]["2g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["2g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["2g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="20e24ed3-0cd1-46dd-bd26-2183ffb443e6") |
| def test_roaming_between_AP1_to_AP2_sae_5g(self): |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(sae_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.sae_networks[0]["5g"] |
| ap2_network = self.sae_networks[1]["5g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["5g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["5g"][ap2_network["SSID"]] |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| |
| @test_tracker_info(uuid="521269cb-5d2c-46e6-bc01-a03bd148ce28") |
| def test_soft_2g_ap_channel_when_roam_to_chan_13(self): |
| """Verify softAp 2G channel when after roaming to network on channel 13. |
| |
| Steps: |
| 1. Configure 2 APs - 1 on channel 6, the other on channel 13. |
| 2. Connect DUT to AP on channel 6. |
| 3. Start softAp on DUT on 2G band. |
| 4. Verify softAp is started on channel 6. |
| 5. Roam to AP2 with channel 13. |
| 6. Verify SoftAp on DUT changed to channel 13. |
| """ |
| for ad in self.android_devices: |
| wutils.set_wifi_country_code( |
| ad, wutils.WifiEnums.CountryCode.AUSTRALIA) |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(open_network=True, |
| ap_count=2, |
| channel_2g_ap2=13, |
| mirror_ap=True) |
| ap1_network = self.open_network[0]["2g"] |
| ap2_network = self.open_network[1]["2g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["2g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["2g"][ap2_network["SSID"]] |
| wutils.set_attns(self.attenuators, "AP1_on_AP2_off", self.roaming_attn) |
| wutils.connect_to_wifi_network(self.dut, ap1_network) |
| |
| # start softap on 2G and verify the channel is 6. |
| sap_config = { |
| WifiEnums.SSID_KEY: "hotspot_%s" % utils.rand_ascii_str(6), |
| WifiEnums.PWD_KEY: "pass_%s" % utils.rand_ascii_str(6), |
| WifiEnums.AP_BAND_KEY: WifiEnums.WIFI_CONFIG_SOFTAP_BAND_2G} |
| asserts.assert_true( |
| self.dut.droid.wifiSetWifiApConfiguration(sap_config), |
| "Failed to set WifiAp Configuration") |
| wutils.start_wifi_tethering_saved_config(self.dut) |
| softap_conf = self.dut.droid.wifiGetApConfiguration() |
| self.log.info("softap conf: %s" % softap_conf) |
| wutils.connect_to_wifi_network(self.dut_client, sap_config) |
| conn_info = self.dut_client.droid.wifiGetConnectionInfo() |
| self.log.info("Wifi connection info on dut_client: %s" % conn_info) |
| softap_channel = wutils.WifiEnums.freq_to_channel[conn_info["frequency"]] |
| asserts.assert_true(softap_channel == 6, |
| "Dut client did not connect to softAp on channel 6") |
| |
| # trigger roaming to AP2 with channel 13 |
| self.log.info("Roaming from %s to %s", ap1_network, ap2_network) |
| wutils.trigger_roaming_and_validate( |
| self.dut, self.attenuators, "AP1_off_AP2_on", ap2_network, |
| self.roaming_attn) |
| |
| # verify softap is now moved to channel 13 |
| conn_info = self.dut_client.droid.wifiGetConnectionInfo() |
| self.log.info("Wifi connection info on dut_client: %s" % conn_info) |
| softap_channel = wutils.WifiEnums.freq_to_channel[conn_info["frequency"]] |
| asserts.assert_true(softap_channel == 13, |
| "Dut client did not connect to softAp on channel 13") |
| |
| @test_tracker_info(uuid="3114d625-5cdd-4205-bb46-5a9d057dc80d") |
| def test_roaming_fail_psk_2g(self): |
| """Verify roaming fail with mismatch passwords. |
| |
| Steps: |
| DUT connect to AP1. |
| Change AP2's password. |
| DUT try roaming from AP1 to AP2 with mismatched password. |
| Change AP2's password back to original one. |
| DUT try roaming from AP1 to AP2 with matched passwords. |
| """ |
| # Use OpenWrt as Wi-Fi AP when it's available in testbed. |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(wpa_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| self.openwrt1 = self.access_points[0] |
| self.openwrt2 = self.access_points[1] |
| ap1_network = self.reference_networks[0]["2g"] |
| ap2_network = self.reference_networks[1]["2g"] |
| # Get APs' BSSIDs. |
| ap1_bssid = self.bssid_map[0]["2g"][ap1_network["SSID"]] |
| ap2_bssid = self.bssid_map[1]["2g"][ap2_network["SSID"]] |
| # Make AP's configs. |
| ap1_network["bssid"] = ap1_bssid |
| ap2_network["bssid"] = ap2_bssid |
| |
| # Change AP2 2G to password. |
| self.openwrt2.set_password(pwd_2g=utils.rand_ascii_str(8)) |
| |
| try: |
| # DUT roaming from AP1 to AP2 with mismatched passwords. |
| self.dut.log.info("Roaming via mismatched passwords to AP2 [{}]" |
| .format(ap2_bssid)) |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| except: |
| self.dut.log.info("Failed roaming to AP2") |
| self.dut.log.info("Roaming via matched passwords to AP2 [{}]" |
| .format(ap1_bssid)) |
| wutils.set_attns_steps(self.attenuators, "AP1_on_AP2_off", |
| self.roaming_attn) |
| self.openwrt2.set_password(pwd_2g=ap2_network["password"]) |
| self.dut.log.info("Toggling wifi OFF.") |
| wutils.wifi_toggle_state(self.dut, False) |
| self.dut.log.info("Toggling wifi ON.") |
| wutils.wifi_toggle_state(self.dut, True) |
| self.roaming_from_AP1_and_AP2(ap1_network, ap2_network) |
| else: |
| raise signals.TestFailure("DUT unexpectedly connect to Wi-Fi.") |
| |
| # Use Google OnHub as Wi-Fi AP to test when OpenWrt is no available. |
| elif "AccessPoint" in self.user_params: |
| network = {'SSID':'test_roaming_fail', 'password':'roam123456@'} |
| # AP2 network with incorrect password. |
| network_fail = {'SSID':'test_roaming_fail', 'password':'roam123456@#$%^'} |
| # Setup AP1 with the correct password. |
| wutils.ap_setup(self, 0, self.access_points[0], network) |
| network_bssid = self.access_points[0].get_bssid_from_ssid( |
| network["SSID"], '2g') |
| # Setup AP2 with the incorrect password. |
| wutils.ap_setup(self, 1, self.access_points[1], network_fail) |
| network_fail_bssid = self.access_points[1].get_bssid_from_ssid( |
| network_fail["SSID"], '2g') |
| network['bssid'] = network_bssid |
| network_fail['bssid'] = network_fail_bssid |
| try: |
| # Initiate roaming with AP2 configured with incorrect password. |
| self.roaming_from_AP1_and_AP2(network, network_fail) |
| except: |
| self.log.info("Roaming failed to AP2 with incorrect password.") |
| # Re-configure AP2 after roaming failed, with correct password. |
| self.log.info("Re-configuring AP2 with correct password.") |
| wutils.ap_setup(self, 1, self.access_points[1], network) |
| self.roaming_from_AP1_and_AP2(network, network_fail) |
| |
| @test_tracker_info(uuid="b6d73094-22bc-4460-9d55-ce34a0a6a8c9") |
| def test_roaming_fail_different_bssid(self): |
| """Verify devices is disconnect with difference bssid after roaming |
| |
| Steps: |
| 1. Configure 2 APs |
| 2. Connect DUT to AP 1 |
| 3. Roam to AP2 |
| 4. Verify the bssid is difference and the device can't connect |
| 5. Verify device is disconnect after roaming. |
| """ |
| if "OpenWrtAP" in self.user_params: |
| self.configure_openwrt_ap_and_start(wpa_network=True, |
| ap_count=2, |
| mirror_ap=True) |
| ap1_network = self.reference_networks[0]["2g"] |
| ap2_network = self.reference_networks[1]["2g"] |
| if "OpenWrtAP" in self.user_params: |
| ap1_network["bssid"] = self.bssid_map[0]["2g"][ap1_network["SSID"]] |
| ap2_network["bssid"] = self.bssid_map[1]["2g"][ap2_network["SSID"]] |
| wutils.set_attns(self.attenuators, "AP1_on_AP2_off") |
| wutils.connect_to_wifi_network(self.dut, ap1_network) |
| |
| # Initiate roaming with AP2 |
| wutils.set_attns(self.attenuators, "AP1_off_AP2_on") |
| time.sleep(10) |
| try: |
| wutils.verify_wifi_connection_info(self.dut, ap2_network) |
| except: |
| self.log.info("Roaming failed to AP2 with incorrect BSSID") |
| wutils.wait_for_disconnect(self.dut) |
| self.log.info("Device is disconnect") |
