diff --git a/autotests/testPreauth-roam/connection_test.py b/autotests/testPreauth-roam/connection_test.py index f31b8b0a..545d4cba 100644 --- a/autotests/testPreauth-roam/connection_test.py +++ b/autotests/testPreauth-roam/connection_test.py @@ -4,69 +4,39 @@ import unittest import sys, os sys.path.append('../util') -import iwd from iwd import IWD -from iwd import PSKAgent from iwd import NetworkType -from hwsim import Hwsim from hostapd import HostapdCLI import testutil class Test(unittest.TestCase): def test_preauth_success(self): - hwsim = Hwsim() - bss_hostapd = [ HostapdCLI(config='eaptls-preauth-1.conf'), HostapdCLI(config='eaptls-preauth-2.conf') ] - bss_radio = [ hwsim.get_radio('rad0'), - hwsim.get_radio('rad1') ] - rule0 = hwsim.rules.create() - rule0.source = bss_radio[0].addresses[0] - rule0.bidirectional = True - - rule1 = hwsim.rules.create() - rule1.source = bss_radio[1].addresses[0] - rule1.bidirectional = True + bss0_addr = bss_hostapd[0].bssid + bss1_addr = bss_hostapd[1].bssid # Fill in the neighbor AP tables in both BSSes. By default each # instance knows only about current BSS, even inside one hostapd # process. # Roaming still works without the neighbor AP table but neighbor # reports have to be disabled in the .conf files - bss0_nr = ''.join(bss_radio[0].addresses[0].split(':')) + \ + bss0_nr = ''.join(bss0_addr.split(':')) + \ '8f0000005101060603000000' - bss1_nr = ''.join(bss_radio[1].addresses[0].split(':')) + \ + bss1_nr = ''.join(bss1_addr.split(':')) + \ '8f0000005102060603000000' - bss_hostapd[0].set_neighbor(bss_radio[1].addresses[0], 'TestPreauth', - bss1_nr) - bss_hostapd[1].set_neighbor(bss_radio[0].addresses[0], 'TestPreauth', - bss0_nr) - - # Check that iwd selects BSS 0 first - rule0.signal = -2500 - rule1.signal = -6900 + bss_hostapd[0].set_neighbor(bss1_addr, 'TestPreauth', bss1_nr) + bss_hostapd[1].set_neighbor(bss0_addr, 'TestPreauth', bss0_nr) wd = IWD(True) device = wd.list_devices(1)[0] - condition = 'not obj.scanning' - wd.wait_for_object_condition(device, condition) - - device.scan() - - condition = 'obj.scanning' - wd.wait_for_object_condition(device, condition) - - condition = 'not obj.scanning' - wd.wait_for_object_condition(device, condition) - - ordered_network = device.get_ordered_network('TestPreauth') + ordered_network = device.get_ordered_network('TestPreauth', scan_if_needed=True) self.assertEqual(ordered_network.type, NetworkType.eap) - self.assertEqual(ordered_network.signal_strength, -2500) condition = 'not obj.connected' wd.wait_for_object_condition(ordered_network.network_object, condition) @@ -74,7 +44,7 @@ class Test(unittest.TestCase): self.assertFalse(bss_hostapd[0].list_sta()) self.assertFalse(bss_hostapd[1].list_sta()) - ordered_network.network_object.connect() + device.connect_bssid(bss0_addr) condition = 'obj.state == DeviceState.connected' wd.wait_for_object_condition(device, condition) @@ -85,10 +55,9 @@ class Test(unittest.TestCase): testutil.test_iface_operstate(device.name) testutil.test_ifaces_connected(bss_hostapd[0].ifname, device.name) self.assertRaises(Exception, testutil.test_ifaces_connected, - bss_hostapd[1].ifname, device.name) + bss_hostapd[1].ifname, device.name, True, True) - # Check that iwd starts transition to BSS 1 in less than 15 seconds - rule0.signal = -8000 + device.roam(bss1_addr) condition = 'obj.state == DeviceState.roaming' wd.wait_for_object_condition(device, condition) @@ -106,7 +75,7 @@ class Test(unittest.TestCase): testutil.test_iface_operstate(device.name) testutil.test_ifaces_connected(bss_hostapd[1].ifname, device.name) self.assertRaises(Exception, testutil.test_ifaces_connected, - (bss_hostapd[0].ifname, device.name)) + (bss_hostapd[0].ifname, device.name, True, True)) device.disconnect()