tests: Fix wpas_ctrl_network for builds without Suite B support

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 1f2daae0ad53324ad55957d3ceb46034c3bc989d..ba1cc63f71c95bdec024678e308c50795fd0f82a 100644 (file)
--- a/tests/hwsim/test_wpas_ctrl.py
+++ b/tests/hwsim/test_wpas_ctrl.py
@@ -62,7 +62,7 @@ def test_wpas_ctrl_network(dev):
             raise Exception("Unexpected success for invalid SET_NETWORK: " + t)
 
     tests = (("key_mgmt", "WPA-PSK WPA-EAP IEEE8021X NONE WPA-NONE FT-PSK FT-EAP WPA-PSK-SHA256 WPA-EAP-SHA256"),
-             ("key_mgmt", "WPS SAE FT-SAE OSEN WPA-EAP-SUITE-B WPA-EAP-SUITE-B-192"),
+             ("key_mgmt", "WPS SAE FT-SAE OSEN"),
              ("pairwise", "CCMP-256 GCMP-256 CCMP GCMP TKIP"),
              ("group", "CCMP-256 GCMP-256 CCMP GCMP TKIP"),
              ("auth_alg", "OPEN SHARED LEAP"),
@@ -82,6 +82,18 @@ def test_wpas_ctrl_network(dev):
         if res != value:
             raise Exception("Unexpected response for '" + field + "': '" + res + "'")
 
+    try:
+        value = "WPA-EAP-SUITE-B WPA-EAP-SUITE-B-192"
+        dev[0].set_network(id, "key_mgmt", value)
+        res = dev[0].get_network(id, "key_mgmt")
+        if res != value:
+            raise Exception("Unexpected response for key_mgmt")
+    except Exception, e:
+        if str(e).startswith("Unexpected"):
+            raise
+        else:
+            pass
+
     q_tests = (("identity", "hello"),
                ("anonymous_identity", "foo@nowhere.com"))
     for field, value in q_tests: