tests: pmksa_cache: Wait for STA on AP side before query

Before querying the PMKSA cache, wait for the STA to have appeared on
the AP side, otherwise scheduling differences may have us asking when
the STA thinks it's connected but the AP hasn't fully processed that.

Signed-off-by: Johannes Berg <johannes.berg@intel.com>

diff --git a/tests/hwsim/test_pmksa_cache.py b/tests/hwsim/test_pmksa_cache.py
index 65ef5d8162e8ee38e43ef598fc3eb4380f7c544a..17f478f1ce60d6091e4b11639374141baa66d279 100644 (file)
--- a/tests/hwsim/test_pmksa_cache.py
+++ b/tests/hwsim/test_pmksa_cache.py
@@ -980,6 +980,7 @@ def test_pmksa_cache_ctrl(dev, apdev):
     pmksa_sta = dev[0].get_pmksa(bssid)
     if pmksa_sta is None:
         raise Exception("No PMKSA cache entry created on STA")
+    hapd.wait_sta()
     pmksa_ap = hapd.get_pmksa(addr)
     if pmksa_ap is None:
         raise Exception("No PMKSA cache entry created on AP")
@@ -1000,6 +1001,7 @@ def test_pmksa_cache_ctrl(dev, apdev):
     pmksa_sta2 = dev[0].get_pmksa(bssid)
     if pmksa_sta2 is None:
         raise Exception("No PMKSA cache entry created on STA after reconnect")
+    hapd.wait_sta()
     pmksa_ap2 = hapd.get_pmksa(addr)
     if pmksa_ap2 is None:
         raise Exception("No PMKSA cache entry created on AP after reconnect")