tests: Make ap_wpa2_gtk_rekey_fail_1_sta more robust

Slight differences in timing might make the final data connectivity
check fail, so disable the incorrect behavior first and wait for a
successfully completed GTK rekeying at the end before the final test.

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

diff --git a/tests/hwsim/test_ap_psk.py b/tests/hwsim/test_ap_psk.py
index 52fab8f02600d3e72e69f06af9410e489529a388..157b863d799a69aaa54cc1706e9fc629bb43309b 100644 (file)
--- a/tests/hwsim/test_ap_psk.py
+++ b/tests/hwsim/test_ap_psk.py
@@ -579,7 +579,11 @@ def test_ap_wpa2_gtk_rekey_fail_1_sta(dev, apdev):
             raise Exception("Unexpected disconnection [%d]" % i)
         hwsim_utils.test_connectivity(dev[i], hapd)
 
+    dev[1].set("disable_eapol_g2_tx", "0")
     dev[1].wait_connected()
+    ev = dev[1].wait_event(["RSN: Group rekeying completed"], timeout=10)
+    if ev is None:
+        raise Exception("GTK rekey timed out [1b]")
     hwsim_utils.test_connectivity(dev[1], hapd)
 
 @remote_compatible