tests: Report missing erp support more nicely in wpas_ctrl_get_capability

Signed-off-by: Jouni Malinen <j@w1.fi>

diff --git a/tests/hwsim/test_wpas_ctrl.py b/tests/hwsim/test_wpas_ctrl.py
index 63307582f9c6f25a5094cadee493e54a744c4175..dbb2e8db2e5da4182dcaac85b273b0f3ebdd2266 100644 (file)
--- a/tests/hwsim/test_wpas_ctrl.py
+++ b/tests/hwsim/test_wpas_ctrl.py
@@ -800,7 +800,7 @@ def test_wpas_ctrl_get_capability(dev):
         raise Exception("Unexpected GET_CAPABILITY tdls response: " + str(res))
 
     res = dev[0].get_capability("erp")
-    if "ERP" not in res[0]:
+    if res is None or "ERP" not in res[0]:
         raise Exception("Unexpected GET_CAPABILITY erp response: " + str(res))
 
     if dev[0].get_capability("foo") is not None: