#!/usr/bin/env python3.4 # # Copyright 2016 - 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 import acts.utils from acts import asserts from acts import signals from acts.test_decorators import test_tracker_info from acts_contrib.test_utils.wifi import wifi_test_utils as wutils from acts_contrib.test_utils.wifi.WifiBaseTest import WifiBaseTest WifiEnums = wutils.WifiEnums DEFAULT_TIMEOUT = 10 # EAP Macros EAP = WifiEnums.Eap EapPhase2 = WifiEnums.EapPhase2 # Enterprise Config Macros Ent = WifiEnums.Enterprise class WifiEnterpriseTest(WifiBaseTest): def setup_class(self): super().setup_class() self.enable_packet_log = True self.dut = self.android_devices[0] # If running in a setup with attenuators, set attenuation on all # channels to zero. if getattr(self, "attenuators", []): for a in self.attenuators: a.set_atten(0) required_userparam_names = ( "ca_cert", "client_cert", "client_key", "eap_identity", "eap_password", "invalid_ca_cert", "invalid_client_cert", "invalid_client_key", "device_password", "radius_conf_2g", "radius_conf_5g", "radius_conf_pwd", "wifi6_models") self.unpack_userparams(required_userparam_names, roaming_consortium_ids=None, plmn=None, ocsp=0) if "AccessPoint" in self.user_params: self.legacy_configure_ap_and_start( ent_network=True, radius_conf_2g=self.radius_conf_2g, radius_conf_5g=self.radius_conf_5g, ent_network_pwd=True, radius_conf_pwd=self.radius_conf_pwd, wpa_network=True, ap_count=2, ) elif "OpenWrtAP" in self.user_params: self.configure_openwrt_ap_and_start( ent_network=True, radius_conf_2g=self.radius_conf_2g, radius_conf_5g=self.radius_conf_5g, ent_network_pwd=True, radius_conf_pwd=self.radius_conf_pwd, wpa_network=True, ap_count=2, ) self.ent_network_2g = self.ent_networks[0]["2g"] self.ent_network_5g = self.ent_networks[0]["5g"] self.ent_network_pwd = self.ent_networks_pwd[0]["2g"] if hasattr(self, "reference_networks") and \ isinstance(self.reference_networks, list): self.wpa_psk_2g = self.reference_networks[0]["2g"] self.wpa_psk_5g = self.reference_networks[0]["5g"] # Default configs for EAP networks. self.config_peap0 = { Ent.EAP: int(EAP.PEAP), Ent.CA_CERT: self.ca_cert, Ent.IDENTITY: self.eap_identity, Ent.PASSWORD: self.eap_password, Ent.PHASE2: int(EapPhase2.MSCHAPV2), WifiEnums.SSID_KEY: self.ent_network_5g[WifiEnums.SSID_KEY], Ent.OCSP: self.ocsp, } self.config_peap1 = dict(self.config_peap0) self.config_peap1[WifiEnums.SSID_KEY] = \ self.ent_network_2g[WifiEnums.SSID_KEY] self.config_tls = { Ent.EAP: int(EAP.TLS), Ent.CA_CERT: self.ca_cert, WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY], Ent.CLIENT_CERT: self.client_cert, Ent.PRIVATE_KEY_ID: self.client_key, Ent.IDENTITY: self.eap_identity, Ent.OCSP: self.ocsp, } self.config_ttls = { Ent.EAP: int(EAP.TTLS), Ent.CA_CERT: self.ca_cert, Ent.IDENTITY: self.eap_identity, Ent.PASSWORD: self.eap_password, Ent.PHASE2: int(EapPhase2.MSCHAPV2), WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY], Ent.OCSP: self.ocsp, } self.config_pwd = { Ent.EAP: int(EAP.PWD), Ent.IDENTITY: self.eap_identity, Ent.PASSWORD: self.eap_password, WifiEnums.SSID_KEY: self.ent_network_pwd[WifiEnums.SSID_KEY], } self.config_sim = { Ent.EAP: int(EAP.SIM), WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY], } self.config_aka = { Ent.EAP: int(EAP.AKA), WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY], } self.config_aka_prime = { Ent.EAP: int(EAP.AKA_PRIME), WifiEnums.SSID_KEY: self.ent_network_2g[WifiEnums.SSID_KEY], } # Set screen lock password so ConfigStore is unlocked. self.dut.droid.setDevicePassword(self.device_password) def teardown_class(self): wutils.reset_wifi(self.dut) self.dut.droid.disableDevicePassword(self.device_password) self.dut.ed.clear_all_events() def setup_test(self): super().setup_test() self.dut.droid.wifiStartTrackingStateChange() self.dut.droid.wakeLockAcquireBright() self.dut.droid.wakeUpNow() wutils.reset_wifi(self.dut) self.dut.ed.clear_all_events() def teardown_test(self): super().teardown_test() self.dut.droid.wakeLockRelease() self.dut.droid.goToSleepNow() self.dut.droid.wifiStopTrackingStateChange() # Turn off airplane mode acts.utils.force_airplane_mode(self.dut, False) wutils.set_attns(self.attenuators, "default") """Helper Functions""" def eap_negative_connect_logic(self, config, ad): """Tries to connect to an enterprise network with invalid credentials and expect a failure. Args: config: A dict representing an invalid EAP credential. Returns: True if connection failed as expected, False otherwise. """ with asserts.assert_raises(signals.TestFailure, extras=config): verdict = wutils.wifi_connect(ad, config) asserts.explicit_pass("Connection failed as expected.") def gen_negative_configs(self, config, neg_params): """Generic function used to generate negative configs. For all the valid configurations, if a param in the neg_params also exists in a config, a copy of the config is made with an invalid value of the param. Args: config: A valid configuration. neg_params: A dict that has all the invalid values. Returns: An invalid configurations generated based on the valid configuration. Each invalid configuration has a different invalid field. """ negative_config = dict(config) if negative_config in [ self.config_sim, self.config_aka, self.config_aka_prime ]: negative_config[WifiEnums.SSID_KEY] = 'wrong_hostapd_ssid' for k, v in neg_params.items(): # Skip negative test for TLS's identity field since it's not # used for auth. if config[Ent.EAP] == EAP.TLS and k == Ent.IDENTITY: continue if k in config: negative_config[k] = v negative_config["invalid_field"] = k return negative_config def gen_negative_eap_configs(self, config): """Generates invalid configurations for different EAP authentication types. For all the valid EAP configurations, if a param that is part of the authentication info exists in a config, a copy of the config is made with an invalid value of the param. Args: A valid network configration Returns: An invalid EAP configuration. """ neg_params = { Ent.CLIENT_CERT: self.invalid_client_cert, Ent.CA_CERT: self.invalid_ca_cert, Ent.PRIVATE_KEY_ID: self.invalid_client_key, Ent.IDENTITY: "fake_identity", Ent.PASSWORD: "wrong_password" } return self.gen_negative_configs(config, neg_params) def eap_connect_toggle_wifi(self, config, *args): """Connects to an enterprise network, toggles wifi state and ensures that the device reconnects. This logic expect the enterprise network to have Internet access. Args: config: A dict representing a wifi enterprise configuration. args: args to be passed to |wutils.eap_connect|. Returns: True if the connection is successful and Internet access works. """ ad = args[0] wutils.wifi_connect(ad, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) wutils.toggle_wifi_and_wait_for_reconnection(ad, config, num_of_tries=5) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) def toggle_out_of_range_stress(self, stress_count=3): """toggle_out_of_range_stress.""" current_network = self.dut.droid.wifiGetConnectionInfo() self.log.info("Current network: {}".format(current_network)) for count in range(stress_count): # move the DUT out of range self.attenuators[0].set_atten(95) self.attenuators[1].set_atten(95) self.attenuators[2].set_atten(95) self.attenuators[3].set_atten(95) time.sleep(20) try: wutils.start_wifi_connection_scan(self.dut) wifi_results = self.dut.droid.wifiGetScanResults() self.log.debug("Scan result {}".format(wifi_results)) time.sleep(20) current_network = self.dut.droid.wifiGetConnectionInfo() self.log.info("Current network: {}".format(current_network)) asserts.assert_true( ('network_id' in current_network and current_network['network_id'] == -1), "Device is connected to network {}".format(current_network)) time.sleep(DEFAULT_TIMEOUT) finally: self.dut.droid.wifiLockRelease() # move the DUT back in range wutils.set_attns(self.attenuators, "default") time.sleep(30) try: wutils.start_wifi_connection_scan(self.dut) wifi_results = self.dut.droid.wifiGetScanResults() self.log.debug("Scan result {}".format(wifi_results)) time.sleep(20) current_network = self.dut.droid.wifiGetConnectionInfo() self.log.info("Current network: {}".format(current_network)) asserts.assert_true( ('network_id' in current_network and current_network['network_id'] != -1), "Device is disconnected to network {}".format(current_network)) time.sleep(DEFAULT_TIMEOUT) finally: self.dut.droid.wifiLockRelease() def check_connection(self, network_bssid): """Check current wifi connection networks. Args: network_bssid: Network bssid to which connection. Returns: True if connection to given network happen, else return False. """ time.sleep(10) #time for connection state to be updated self.log.info("Check network for {}".format(network_bssid)) current_network = self.dut.droid.wifiGetConnectionInfo() self.log.debug("Current network: {}".format(current_network)) if WifiEnums.BSSID_KEY in current_network: return current_network[WifiEnums.BSSID_KEY] == network_bssid return False """ Tests """ # EAP connect tests """ Test connecting to enterprise networks of different authentication types. The authentication types tested are: EAP-TLS EAP-PEAP with different phase2 types. EAP-TTLS with different phase2 types. Procedures: For each enterprise wifi network 1. Connect to the network. 2. Send a GET request to a website and check response. Expect: Successful connection and Internet access through the enterprise networks. """ @test_tracker_info(uuid="4e720cac-ea17-4de7-a540-8dc7c49f9713") def test_eap_connect_with_config_tls(self): wutils.wifi_connect(self.dut, self.config_tls) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="10e3a5e9-0018-4162-a9fa-b41500f13340") def test_eap_connect_with_config_pwd(self): wutils.wifi_connect(self.dut, self.config_pwd) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="b4513f78-a1c4-427f-bfc7-2a6b3da714b5") def test_eap_connect_with_config_sim(self): wutils.wifi_connect(self.dut, self.config_sim) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="7d390e30-cb67-4b55-bf00-567adad2d9b0") def test_eap_connect_with_config_aka(self): wutils.wifi_connect(self.dut, self.config_aka) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="742f921b-27c3-4b68-a3ca-88e64fe79c1d") def test_eap_connect_with_config_aka_prime(self): wutils.wifi_connect(self.dut, self.config_aka_prime) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="d34e30f3-6ef6-459f-b47a-e78ed90ce4c6") def test_eap_connect_with_config_ttls_none(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.NONE.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="0dca3a15-472e-427c-8e06-4e38088ee973") def test_eap_connect_with_config_ttls_pap(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.PAP.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="47c4b459-2cb1-4fc7-b4e7-82534e8e090e") def test_eap_connect_with_config_ttls_mschap(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAP.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="fdb286c7-8069-481d-baf0-c5dd7a31ff03") def test_eap_connect_with_config_ttls_mschapv2(self): config = dict(self.config_ttls) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="d9315962-7987-4ee7-905d-6972c78ce8a1") def test_eap_connect_with_config_ttls_gtc(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="90a67bd3-30da-4daf-8ab0-d964d7ad19be") def test_eap_connect_with_config_peap0_mschapv2(self): config = dict(self.config_peap0) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="3c451ba4-0c83-4eef-bc95-db4c21893008") def test_eap_connect_with_config_peap0_gtc(self): config = dict(self.config_peap0) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="6b45157d-0325-417a-af18-11af5d240d79") def test_eap_connect_with_config_peap1_mschapv2(self): config = dict(self.config_peap1) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) @test_tracker_info(uuid="1663decc-71ae-4f95-a027-8a6dbf9c337f") def test_eap_connect_with_config_peap1_gtc(self): config = dict(self.config_peap1) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value wutils.wifi_connect(self.dut, config) wutils.verify_11ax_wifi_connection( self.dut, self.wifi6_models, "wifi6_ap" in self.user_params) # EAP connect negative tests """ Test connecting to enterprise networks. Procedures: For each enterprise wifi network 1. Connect to the network with invalid credentials. Expect: Fail to establish connection. """ @test_tracker_info(uuid="b2a91f1f-ccd7-4bd1-ab81-19aab3d8ee38") def test_eap_connect_negative_with_config_tls(self): config = self.gen_negative_eap_configs(self.config_tls) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="b7fb8517-5d52-468e-890a-40ea24129bf1") def test_network_selection_status_wpa2_eap_tls_invalid_cert(self): config = self.gen_negative_eap_configs(self.config_tls) try: wutils.connect_to_wifi_network(self.dut, config) asserts.fail( "WPA2 EAP TLS worked with invalid cert. Expected to fail.") except: asserts.assert_true( self.dut.droid.wifiIsNetworkTemporaryDisabledForNetwork(config), "WiFi network is not temporary disabled.") asserts.explicit_pass( "Connection failed with correct network selection status.") @test_tracker_info(uuid="6466abde-1d16-4168-9dd8-1e7a0a19889b") def test_eap_connect_negative_with_config_pwd(self): config = self.gen_negative_eap_configs(self.config_pwd) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="d7742a2a-85b0-409a-99d8-47711ddc5612") def test_eap_connect_negative_with_config_sim(self): config = self.gen_negative_eap_configs(self.config_sim) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="0ec0de93-cab3-4f41-960b-c0af64ff48c4") def test_eap_connect_negative_with_config_aka(self): config = self.gen_negative_eap_configs(self.config_aka) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="bb640ea4-32a6-48ea-87c9-f7128fffbbf6") def test_eap_connect_negative_with_config_aka_prime(self): config = self.gen_negative_eap_configs(self.config_aka_prime) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="86336ada-0ced-45a4-8a22-c4aa23c81111") def test_eap_connect_negative_with_config_ttls_none(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.NONE.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="71e0498d-9973-4958-94bd-79051c328920") def test_eap_connect_negative_with_config_ttls_pap(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.PAP.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="c04142a8-b204-4d2d-98dc-150b16c8397e") def test_eap_connect_negative_with_config_ttls_mschap(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAP.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="625e7aa5-e3e6-4bbe-98c0-5aad8ca1555b") def test_eap_connect_negative_with_config_ttls_mschapv2(self): config = dict(self.config_ttls) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="24ea0d80-0a3f-41c2-8e05-d6387e589058") def test_eap_connect_negative_with_config_ttls_gtc(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="b7c1f0f8-6338-4501-8e1d-c9b136aaba88") def test_eap_connect_negative_with_config_peap0_mschapv2(self): config = dict(self.config_peap0) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="9cf83dcb-38ad-4f75-9ea9-98de1cfaf7f3") def test_eap_connect_negative_with_config_peap0_gtc(self): config = dict(self.config_peap0) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="89bb2b6b-d073-402a-bdc1-68ac5f8752a3") def test_eap_connect_negative_with_config_peap1_mschapv2(self): config = dict(self.config_peap1) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) @test_tracker_info(uuid="2252a864-9ff7-43b5-82d9-afe57d1f5e5f") def test_eap_connect_negative_with_config_peap1_gtc(self): config = dict(self.config_peap1) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value config = self.gen_negative_eap_configs(config) self.eap_negative_connect_logic(config, self.dut) # EAP connect config store tests """ Test connecting to enterprise networks of different authentication types after wifi toggle. The authentication types tested are: EAP-TLS EAP-PEAP with different phase2 types. EAP-TTLS with different phase2 types. Procedures: For each enterprise wifi network 1. Connect to the network. 2. Send a GET request to a website and check response. 3. Toggle wifi. 4. Ensure that the device reconnects to the same network. Expect: Successful connection and Internet access through the enterprise networks. """ @test_tracker_info(uuid="2a933b7f-27d7-4201-a34f-25b9d8072a8c") def test_eap_connect_config_store_with_config_tls(self): self.eap_connect_toggle_wifi(self.config_tls, self.dut) @test_tracker_info(uuid="08dc071b-9fea-408a-a3f6-d3493869f6d4") def test_eap_connect_config_store_with_config_pwd(self): self.eap_connect_toggle_wifi(self.config_pwd, self.dut) @test_tracker_info(uuid="230cb03e-58bc-41cb-b9b3-7215c2ab2325") def test_eap_connect_config_store_with_config_sim(self): self.eap_connect_toggle_wifi(self.config_sim, self.dut) @test_tracker_info(uuid="dfc3e59c-2309-4598-8c23-bb3fe95ef89f") def test_eap_connect_config_store_with_config_aka(self): self.eap_connect_toggle_wifi(self.config_aka, self.dut) @test_tracker_info(uuid="6050a1d1-4f3a-476d-bf93-638abd066790") def test_eap_connect_config_store_with_config_aka_prime(self): self.eap_connect_toggle_wifi(self.config_aka_prime, self.dut) @test_tracker_info(uuid="03108057-cc44-4a80-8331-77c93694099c") def test_eap_connect_config_store_with_config_ttls_none(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.NONE.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="53dd8195-e272-4589-a261-b8fa3607ad8d") def test_eap_connect_config_store_with_config_ttls_pap(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.PAP.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="640f697b-9c62-4b19-bd76-53b236a152e0") def test_eap_connect_config_store_with_config_ttls_mschap(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAP.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="f0243684-fae0-46f3-afbd-bf525fc712e2") def test_eap_connect_config_store_with_config_ttls_mschapv2(self): config = dict(self.config_ttls) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="49ec7202-3b00-49c3-970a-201360888c74") def test_eap_connect_config_store_with_config_ttls_gtc(self): config = dict(self.config_ttls) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="1c6abfa3-f344-4e28-b891-5481ab79efcf") def test_eap_connect_config_store_with_config_peap0_mschapv2(self): config = dict(self.config_peap0) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="2815bc76-49fa-43a5-a4b6-84788f9809d5") def test_eap_connect_config_store_with_config_peap0_gtc(self): config = dict(self.config_peap0) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="e93f7472-6895-4e36-bff2-9b2dcfd07ad0") def test_eap_connect_config_store_with_config_peap1_mschapv2(self): config = dict(self.config_peap1) config[ WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.MSCHAPV2.value self.eap_connect_toggle_wifi(config, self.dut) @test_tracker_info(uuid="6da72fa0-b858-4475-9559-46fe052d0d64") def test_eap_connect_config_store_with_config_peap1_gtc(self): config = dict(self.config_peap1) config[WifiEnums.Enterprise.PHASE2] = WifiEnums.EapPhase2.GTC.value self.eap_connect_toggle_wifi(config, self.dut) # Airplane mode on with wifi connect tests """ Test connecting to enterprise networks of different authentication types after airplane mode on. The authentication types tested are: EAP-SIM EAP-AKA EAP-AKA_PRIME Procedures: For each enterprise wifi network 1. Turn on Airplane mode 2. Toggle wifi.. 3. Ensure that the device connects to the enterprise network. Expect: Successful connection and Internet access through the enterprise networks with Airplane mode on. """ @test_tracker_info(uuid="54b96a6c-f366-421c-9a72-80d7ee8fac8f") def test_eap_connect_with_config_sim_airplane_on(self): self.log.info("Toggling Airplane mode ON") asserts.assert_true( acts.utils.force_airplane_mode(self.dut, True), "Can not turn on airplane mode on: %s" % self.dut.serial) time.sleep(DEFAULT_TIMEOUT) wutils.wifi_toggle_state(self.dut, True) wutils.wifi_connect(self.dut, self.config_sim) self.log.info("Toggling Airplane mode OFF") asserts.assert_true( acts.utils.force_airplane_mode(self.dut, False), "Can not turn OFF airplane mode: %s" % self.dut.serial) wutils.start_wifi_connection_scan_and_ensure_network_found( self.dut, self.config_sim[WifiEnums.SSID_KEY]) @test_tracker_info(uuid="344f63f6-7c99-4507-8036-757f9f911d20") def test_eap_connect_with_config_aka_airplane_on(self): self.log.info("Toggling Airplane mode ON") asserts.assert_true( acts.utils.force_airplane_mode(self.dut, True), "Can not turn on airplane mode on: %s" % self.dut.serial) time.sleep(DEFAULT_TIMEOUT) wutils.wifi_toggle_state(self.dut, True) wutils.wifi_connect(self.dut, self.config_aka) self.log.info("Toggling Airplane mode OFF") asserts.assert_true( acts.utils.force_airplane_mode(self.dut, False), "Can not turn OFF airplane mode: %s" % self.dut.serial) wutils.start_wifi_connection_scan_and_ensure_network_found( self.dut, self.config_aka[WifiEnums.SSID_KEY]) @test_tracker_info(uuid="5502b8c8-89d7-4ce9-afee-cae50e71f5f4") def test_eap_connect_with_config_aka_prime_airplane_on(self): self.log.info("Toggling Airplane mode ON") asserts.assert_true( acts.utils.force_airplane_mode(self.dut, True), "Can not turn on airplane mode on: %s" % self.dut.serial) time.sleep(DEFAULT_TIMEOUT) wutils.wifi_toggle_state(self.dut, True) wutils.wifi_connect(self.dut, self.config_aka_prime) self.log.info("Toggling Airplane mode OFF") asserts.assert_true( acts.utils.force_airplane_mode(self.dut, False), "Can not turn OFF airplane mode: %s" % self.dut.serial) wutils.start_wifi_connection_scan_and_ensure_network_found( self.dut, self.config_aka_prime[WifiEnums.SSID_KEY]) @test_tracker_info(uuid="360a6fec-f4ee-4ecd-9b15-e836c977d6db") def test_connect_eap_sim_network_out_of_range_back(self): """Test connecting to enterprise networks to do out of range then back in range 3 times 1. Connecting EAP-SIM network 2. Move DUT out of range then back in range 3 times 3. Check that device is connected to network. """ wutils.wifi_connect(self.dut, self.config_sim) self.toggle_out_of_range_stress() @test_tracker_info(uuid="9fb71afb-5599-4ca1-b458-09752c40bb0d") def test_eap_sim_network_out_of_range_back_airplane(self): """Test connecting to enterprise networks with airplne mode on to do out of range then back in range 3 times 1. Turn on airplane mode 2. Connecting EAP-SIM network 3. Move DUT out of range then back in range 3 times 4. Check that device is connected to network. """ self.log.debug("Toggling Airplane mode ON") asserts.assert_true( acts.utils.force_airplane_mode(self.dut, True), "Can not turn on airplane mode on: %s" % self.dut.serial) time.sleep(DEFAULT_TIMEOUT) self.log.debug("Toggling WiFi mode ON") wutils.wifi_toggle_state(self.dut, True) time.sleep(DEFAULT_TIMEOUT) wutils.wifi_connect(self.dut, self.config_sim) self.toggle_out_of_range_stress() @test_tracker_info(uuid="9e899c55-1a62-498c-bbf1-e9472e42e84f") def test_eap_sim_network_reboot(self): """Test connecting to enterprise networks with airplne mode on to do out of range then back in range 3 times 1. Connecting EAP-SIM network 3. Check that device is connected to network after reboot. """ self.dut.droid.disableDevicePassword(self.device_password) wutils.wifi_connect(self.dut, self.config_sim) current_network = self.dut.droid.wifiGetConnectionInfo() self.log.info("Current network: {}".format(current_network)) self.dut.reboot() time.sleep(DEFAULT_TIMEOUT) self.check_connection(self.config_sim) @test_tracker_info(uuid="8e7465fb-5b16-4abb-92d8-a2c79355e377") def test_connect_to_EAP_SIM_network_switch_to_WPA2(self): """Test connecting PSK's AP1 and one EAP AP2 network switch test 1. Connect to a PSK's AP1 before connect to EAP-SIM AP2 network. 2. Out of PSK's AP1 range. 3. Connect to EAP-SIM network, then in AP1 range to switch WPA2-PSK network. """ attn1 = self.attenuators[0] attn2 = self.attenuators[2] if "OpenWrtAP" in self.user_params: attn2 = self.attenuators[1] ap1_network = self.config_sim ap2_network = self.reference_networks[1]["2g"] attn1.set_atten(0) attn2.set_atten(0) wutils.connect_to_wifi_network(self.dut, ap2_network) #Enable EAP network signal attn2.set_atten(95) time.sleep(DEFAULT_TIMEOUT) wutils.connect_to_wifi_network(self.dut, ap1_network) current_network = self.dut.droid.wifiGetConnectionInfo() self.log.info("Current network: {}".format(current_network)) #Enable SSID1 network signal attn1.set_atten(95) attn2.set_atten(0) time.sleep(20) self.check_connection(ap2_network)