tests: Replace hapd_connected() with hapd.wait_sta()

[thirdparty/hostap.git] / tests / hwsim / test_ocv.py

diff --git a/tests/hwsim/test_ocv.py b/tests/hwsim/test_ocv.py
index 9573ecb9e8286f5bf53de7872bae22ece0edc831..df84e7124f0259bc0d5d0260f8283d7873878a53 100644 (file)
--- a/tests/hwsim/test_ocv.py
+++ b/tests/hwsim/test_ocv.py
@@ -15,7 +15,7 @@ import hwsim_utils
 from utils import HwsimSkip
 
 from test_ap_ht import set_world_reg
-from test_ap_psk import parse_eapol, build_eapol, pmk_to_ptk, eapol_key_mic, recv_eapol, send_eapol, reply_eapol, hapd_connected, build_eapol_key_3_4, aes_wrap, pad_key_data
+from test_ap_psk import parse_eapol, build_eapol, pmk_to_ptk, eapol_key_mic, recv_eapol, send_eapol, reply_eapol, build_eapol_key_3_4, aes_wrap, pad_key_data
 
 #TODO: Refuse setting up AP with OCV but without MFP support
 #TODO: Refuse to connect to AP that advertises OCV but not MFP
@@ -424,7 +424,7 @@ class APConnection:
 
         reply_eapol("4/4", self.hapd, self.addr, self.msg, 0x030a, None, None,
                     self.kck)
-        hapd_connected(self.hapd)
+        self.hapd.wait_sta(timeout=15)
 
 @remote_compatible
 def test_wpa2_ocv_ap_mismatch(dev, apdev):