diff --git a/tests/hwsim/test_ap_ft.py b/tests/hwsim/test_ap_ft.py index c30cb8593..a2f1a4ead 100644 --- a/tests/hwsim/test_ap_ft.py +++ b/tests/hwsim/test_ap_ft.py @@ -560,6 +560,9 @@ def test_ap_ft_pmf_required_over_ds(dev, apdev): def test_ap_ft_pmf_beacon_prot(dev, apdev): """WPA2-PSK-FT AP with PMF and beacon protection""" run_ap_ft_pmf(dev, apdev, "1", beacon_prot=True) + ev = dev[0].wait_event(["CTRL-EVENT-BEACON-LOSS"], timeout=5) + if ev is not None: + raise Exception("Beacon loss detected") def run_ap_ft_pmf(dev, apdev, ieee80211w, over_ds=False, beacon_prot=False): ssid = "test-ft" diff --git a/tests/hwsim/test_ap_pmf.py b/tests/hwsim/test_ap_pmf.py index 3a9fa7a72..7b7e5a29e 100644 --- a/tests/hwsim/test_ap_pmf.py +++ b/tests/hwsim/test_ap_pmf.py @@ -1371,6 +1371,10 @@ def run_ap_pmf_beacon_protection(dev, apdev, cipher): if valid_bip < 0 or invalid_bip > 0 or missing_bip > 0: raise Exception("Unexpected wlantest BIP counters: valid=%d invalid=%d missing=%d" % (valid_bip, invalid_bip, missing_bip)) + ev = dev[0].wait_event(["CTRL-EVENT-BEACON-LOSS"], timeout=5) + if ev is not None: + raise Exception("Beacon loss detected") + def test_ap_pmf_beacon_protection_mismatch(dev, apdev): """WPA2-PSK Beacon protection MIC mismatch""" run_ap_pmf_beacon_protection_mismatch(dev, apdev, False)