tests: iftype parameter with GET_CAPABILITY key_mgmt

Signed-off-by: Jouni Malinen <jouni@codeaurora.org>

diff --git a/tests/hwsim/test_wpas_ctrl.py b/tests/hwsim/test_wpas_ctrl.py
index cc3d5e98b97c1bd8ae47273ec41f69a96cde2255..04418f09367f5fc88f1c7726e83bcb9fae7790a3 100644 (file)
--- a/tests/hwsim/test_wpas_ctrl.py
+++ b/tests/hwsim/test_wpas_ctrl.py
@@ -987,6 +987,16 @@ def test_wpas_ctrl_get_capability(dev):
     if "WPA-PSK" not in res or "WPA-EAP" not in res:
         raise Exception("Unexpected GET_CAPABILITY key_mgmt response: " + str(res))
 
+    res = dev[0].get_capability("key_mgmt iftype=STATION")
+    if "WPA-PSK" not in res or "WPA-EAP" not in res:
+        raise Exception("Unexpected GET_CAPABILITY key_mgmt iftype=STATION response: " + str(res))
+
+    iftypes = [ "STATION", "AP_VLAN", "AP", "P2P_GO", "P2P_CLIENT",
+                "P2P_DEVICE", "MESH", "IBSS", "NAN", "UNKNOWN" ]
+    for i in iftypes:
+        res = dev[0].get_capability("key_mgmt iftype=" + i)
+        logger.info("GET_CAPABILITY key_mgmt iftype=%s: %s" % (i, res))
+
     res = dev[0].get_capability("proto")
     if "WPA" not in res or "RSN" not in res:
         raise Exception("Unexpected GET_CAPABILITY proto response: " + str(res))