tests: Make nfc_wps_handover_init more robust

Data connectivity test could have been started in the middle of 4-way
handshake. This test case needs to wait for two STA connections before
starting the connectivity test since the first one is only for the
provisioning step.

Signed-off-by: Jouni Malinen <j@w1.fi>

diff --git a/tests/hwsim/test_nfc_wps.py b/tests/hwsim/test_nfc_wps.py
index 7609639764d5a450a1ed7ce0831f30cc3fd7931b..e98eb4113d27648e17cfe4132a958ef2a98710f9 100644 (file)
--- a/tests/hwsim/test_nfc_wps.py
+++ b/tests/hwsim/test_nfc_wps.py
@@ -241,6 +241,9 @@ def _test_nfc_wps_handover_init(dev, apdev):
     if "FAIL" in res:
         raise Exception("Failed to report NFC connection handover to to wpa_supplicant")
     dev[0].wait_connected(timeout=15)
+    # WPS provisioning
+    hapd.wait_sta()
+    # data connection
     hapd.wait_sta()
     check_wpa2_connection(dev[0], apdev[0], hapd, ssid, mixed=True)