[WifiBaseTest]Correcting condition to raise AP value error.
Bug: 111500771
Bug: 111502276
Test: Local
Change-Id: I6af55fe0faffbfa7726d4439d32972ca8f92769d
diff --git a/acts/framework/acts/controllers/access_point.py b/acts/framework/acts/controllers/access_point.py
index 1b304e8..4691e9b 100755
--- a/acts/framework/acts/controllers/access_point.py
+++ b/acts/framework/acts/controllers/access_point.py
@@ -27,6 +27,7 @@
from acts.controllers.ap_lib import dhcp_server
from acts.controllers.ap_lib import hostapd
from acts.controllers.ap_lib import hostapd_config
+from acts.controllers.ap_lib import hostapd_constants
from acts.controllers.utils_lib.commands import ip
from acts.controllers.utils_lib.commands import route
from acts.controllers.utils_lib.commands import shell
diff --git a/acts/framework/acts/test_utils/wifi/WifiBaseTest.py b/acts/framework/acts/test_utils/wifi/WifiBaseTest.py
index 010e26d..1f69270 100755
--- a/acts/framework/acts/test_utils/wifi/WifiBaseTest.py
+++ b/acts/framework/acts/test_utils/wifi/WifiBaseTest.py
@@ -261,8 +261,10 @@
config_count = 1
count = 0
- if mirror_ap and ap_count == 1:
- raise ValueError("ap_count cannot be 1 if mirror_ap is True.")
+ # For example, the NetworkSelector tests use 2 APs and require that
+ # both APs are not mirrored.
+ if not mirror_ap and ap_count == 1:
+ raise ValueError("ap_count cannot be 1 if mirror_ap is False.")
if not mirror_ap:
config_count = ap_count