tests: Make dpp_controller_init_through_relay* more robust

Avoid a race condition that could result in the relay (hostapd)
maintaining STA association with PMF and discarding of the GAS frames.

Signed-off-by: Jouni Malinen <quic_jouni@quicinc.com>

diff --git a/tests/hwsim/test_dpp.py b/tests/hwsim/test_dpp.py
index 49da060ad64fe06a207962dc3d48b89412afba4e..e595beea7b079b31e9352c41ba2748e00e96d988 100644 (file)
--- a/tests/hwsim/test_dpp.py
+++ b/tests/hwsim/test_dpp.py
@@ -5977,9 +5977,11 @@ def run_dpp_controller_init_through_relay(dev, apdev, params, dynamic=False,
         raise Exception("DPP network id not reported")
     network = int(ev.split(' ')[1])
     dev[0].wait_connected()
+    relay.wait_sta()
     dev[0].dump_monitor()
     dev[0].request("DISCONNECT")
     dev[0].wait_disconnected()
+    relay.wait_sta_disconnect()
     dev[0].dump_monitor()
 
     if add: