diff --git a/tests/hwsim/test_ap_tdls.py b/tests/hwsim/test_ap_tdls.py index 3afa4a886..2d1fc2cf5 100644 --- a/tests/hwsim/test_ap_tdls.py +++ b/tests/hwsim/test_ap_tdls.py @@ -359,6 +359,22 @@ def test_ap_wpa2_tdls_responder_teardown(dev, apdev): setup_tdls(dev[0], dev[1], hapd) teardown_tdls(dev[0], dev[1], hapd, responder=True) +def tdls_clear_reg(hapd, dev): + if hapd: + hapd.request("DISABLE") + dev[0].request("DISCONNECT") + dev[0].request("ABORT_SCAN") + dev[1].request("DISCONNECT") + dev[1].request("ABORT_SCAN") + dev[0].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=0.5) + dev[0].dump_monitor() + dev[1].wait_event(["CTRL-EVENT-DISCONNECTED"], timeout=0.5) + dev[1].dump_monitor() + subprocess.call(['iw', 'reg', 'set', '00']) + dev[0].wait_event(["CTRL-EVENT-REGDOM-CHANGE"], timeout=0.5) + dev[0].flush_scan_cache() + dev[1].flush_scan_cache() + def test_ap_open_tdls_vht(dev, apdev): """Open AP and two stations using TDLS""" params = { "ssid": "test-open", @@ -381,13 +397,7 @@ def test_ap_open_tdls_vht(dev, apdev): setup_tdls(dev[1], dev[0], hapd) teardown_tdls(dev[1], dev[0], hapd, wildcard=True) finally: - dev[0].request("DISCONNECT") - dev[1].request("DISCONNECT") - if hapd: - hapd.request("DISABLE") - subprocess.call(['iw', 'reg', 'set', '00']) - dev[0].flush_scan_cache() - dev[1].flush_scan_cache() + tdls_clear_reg(hapd, dev) def test_ap_open_tdls_vht80(dev, apdev): """Open AP and two stations using TDLS with VHT 80""" @@ -424,13 +434,7 @@ def test_ap_open_tdls_vht80(dev, apdev): raise HwsimSkip("80/160 MHz channel not supported in regulatory information") raise finally: - dev[0].request("DISCONNECT") - dev[1].request("DISCONNECT") - if hapd: - hapd.request("DISABLE") - subprocess.call(['iw', 'reg', 'set', '00']) - dev[0].flush_scan_cache() - dev[1].flush_scan_cache() + tdls_clear_reg(hapd, dev) def test_ap_open_tdls_vht80plus80(dev, apdev): """Open AP and two stations using TDLS with VHT 80+80""" @@ -474,13 +478,7 @@ def test_ap_open_tdls_vht80plus80(dev, apdev): raise HwsimSkip("80/160 MHz channel not supported in regulatory information") raise finally: - dev[0].request("DISCONNECT") - dev[1].request("DISCONNECT") - if hapd: - hapd.request("DISABLE") - subprocess.call(['iw', 'reg', 'set', '00']) - dev[0].flush_scan_cache() - dev[1].flush_scan_cache() + tdls_clear_reg(hapd, dev) def test_ap_open_tdls_vht160(dev, apdev): """Open AP and two stations using TDLS with VHT 160""" @@ -524,13 +522,7 @@ def test_ap_open_tdls_vht160(dev, apdev): raise HwsimSkip("80/160 MHz channel not supported in regulatory information") raise finally: - dev[0].request("DISCONNECT") - dev[1].request("DISCONNECT") - if hapd: - hapd.request("DISABLE") - subprocess.call(['iw', 'reg', 'set', '00']) - dev[0].flush_scan_cache() - dev[1].flush_scan_cache() + tdls_clear_reg(hapd, dev) def test_tdls_chan_switch(dev, apdev): """Open AP and two stations using TDLS"""