tests: Make mbo_cell_capa_update_pmf more robust

Wait for hostapd to report completion of connection so that the WNM
Notification Request frame does not get sent before the AP has processed
EAPOL-Key msg 4/4 and configured the TK. This could result in a race
condition especially when testing using UML with time-travel.

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

diff --git a/tests/hwsim/test_mbo.py b/tests/hwsim/test_mbo.py
index 693ddf5873548290c9e5672a1860387b5f25f4c8..5a71cfba8b9a6060bd8c8406d371c42c51665542 100644 (file)
--- a/tests/hwsim/test_mbo.py
+++ b/tests/hwsim/test_mbo.py
@@ -321,6 +321,7 @@ def test_mbo_cell_capa_update_pmf(dev, apdev):
 
     dev[0].connect(ssid, psk=passphrase, key_mgmt="WPA-PSK-SHA256",
                    proto="WPA2", ieee80211w="2", scan_freq="2412")
+    hapd.wait_sta()
 
     addr = dev[0].own_addr()
     sta = hapd.get_sta(addr)