diff options
| -rw-r--r-- | acts_tests/tests/google/wifi/WifiEnterpriseRoamingTest.py | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/acts_tests/tests/google/wifi/WifiEnterpriseRoamingTest.py b/acts_tests/tests/google/wifi/WifiEnterpriseRoamingTest.py index 0cebd42e28..b9b7efd207 100644 --- a/acts_tests/tests/google/wifi/WifiEnterpriseRoamingTest.py +++ b/acts_tests/tests/google/wifi/WifiEnterpriseRoamingTest.py @@ -59,13 +59,22 @@ class WifiEnterpriseRoamingTest(WifiBaseTest): ap_count=2, radius_conf_2g=self.radius_conf_2g, radius_conf_5g=self.radius_conf_5g,) + elif "OpenWrtAP" in self.user_params: + self.configure_openwrt_ap_and_start( + mirror_ap=True, + ent_network=True, + ap_count=2, + radius_conf_2g=self.radius_conf_2g, + radius_conf_5g=self.radius_conf_5g,) self.ent_network_2g_a = self.ent_networks[0]["2g"] self.ent_network_2g_b = self.ent_networks[1]["2g"] - self.bssid_2g_a = self.ent_network_2g_a[WifiEnums.BSSID_KEY.lower()] - self.bssid_2g_b = self.ent_network_2g_b[WifiEnums.BSSID_KEY.lower()] self.ent_roaming_ssid = self.ent_network_2g_a[WifiEnums.SSID_KEY] - self.bssid_a = self.bssid_2g_a - self.bssid_b = self.bssid_2g_b + if "AccessPoint" in self.user_params: + self.bssid_a = self.ent_network_2g_a[WifiEnums.BSSID_KEY.lower()] + self.bssid_b = self.ent_network_2g_b[WifiEnums.BSSID_KEY.lower()] + elif "OpenWrtAP" in self.user_params: + self.bssid_a = self.bssid_map[0]["2g"][self.ent_roaming_ssid] + self.bssid_b = self.bssid_map[1]["2g"][self.ent_roaming_ssid] self.config_peap = { Ent.EAP: int(EAP.PEAP), @@ -97,6 +106,8 @@ class WifiEnterpriseRoamingTest(WifiBaseTest): } self.attn_a = self.attenuators[0] self.attn_b = self.attenuators[2] + if "OpenWrtAP" in self.user_params: + self.attn_b = self.attenuators[1] # Set screen lock password so ConfigStore is unlocked. self.dut.droid.setDevicePassword(self.device_password) self.set_attns("default") |