tests: SAE and PMKSA caching (PMKID in AssocReq after SAE)

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

diff --git a/tests/hwsim/test_sae.py b/tests/hwsim/test_sae.py
index 8ebf7d16386d45a1152eeabfaad31a50bc5873c0..8824ad4c6f550a5f4d232d508c54f9b5c3074deb 100644 (file)
--- a/tests/hwsim/test_sae.py
+++ b/tests/hwsim/test_sae.py
@@ -92,6 +92,18 @@ def test_sae_password_ffc(dev, apdev):
 @remote_compatible
 def test_sae_pmksa_caching(dev, apdev):
     """SAE and PMKSA caching"""
+    run_sae_pmksa_caching(dev, apdev)
+
+@remote_compatible
+def test_sae_pmksa_caching_pmkid(dev, apdev):
+    """SAE and PMKSA caching (PMKID in AssocReq after SAE)"""
+    try:
+        dev[0].set("sae_pmkid_in_assoc", "1")
+        run_sae_pmksa_caching(dev, apdev)
+    finally:
+        dev[0].set("sae_pmkid_in_assoc", "0")
+
+def run_sae_pmksa_caching(dev, apdev):
     if "SAE" not in dev[0].get_capability("auth_alg"):
         raise HwsimSkip("SAE not supported")
     params = hostapd.wpa2_params(ssid="test-sae",