mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2024-12-20 19:12:33 +01:00
d232edde72
Many tests waited on the network object 'connected' property after issuing a Connect command. This is not correct as 'connected' is set quite early in the connection process. The correct way of doing this is waiting for the device state to change to connected. This common code was replaced, hopefully putting to rest any random failures that happen occasionally.
107 lines
2.9 KiB
Python
107 lines
2.9 KiB
Python
#! /usr/bin/python3
|
|
|
|
import unittest
|
|
import sys
|
|
|
|
sys.path.append('../util')
|
|
import iwd
|
|
from iwd import IWD
|
|
from iwd import PSKAgent
|
|
from iwd import NetworkType
|
|
from hwsim import Hwsim
|
|
import testutil
|
|
from hostapd import HostapdCLI
|
|
|
|
class Test(unittest.TestCase):
|
|
def do_connect(self, wd, device, hostapd):
|
|
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('testSSID')
|
|
|
|
self.assertEqual(ordered_network.type, NetworkType.psk)
|
|
|
|
condition = 'not obj.connected'
|
|
wd.wait_for_object_condition(ordered_network.network_object, condition)
|
|
|
|
ordered_network.network_object.connect()
|
|
|
|
condition = 'obj.state == DeviceState.connected'
|
|
wd.wait_for_object_condition(device, condition)
|
|
|
|
testutil.test_iface_operstate()
|
|
testutil.test_ifaces_connected(device.name, hostapd.ifname)
|
|
|
|
device.disconnect()
|
|
|
|
condition = 'not obj.connected'
|
|
wd.wait_for_object_condition(ordered_network.network_object, condition)
|
|
|
|
def test_connection_success(self):
|
|
hwsim = Hwsim()
|
|
non_ht_hostapd = HostapdCLI(config='non-ht-vht.conf')
|
|
ht_hostapd = HostapdCLI(config='ht.conf')
|
|
vht_hostapd = HostapdCLI(config='vht.conf')
|
|
non_ht_radio = hwsim.get_radio('rad0')
|
|
ht_radio = hwsim.get_radio('rad1')
|
|
vht_radio = hwsim.get_radio('rad2')
|
|
|
|
self.assertIsNotNone(non_ht_hostapd)
|
|
self.assertIsNotNone(ht_hostapd)
|
|
self.assertIsNotNone(vht_hostapd)
|
|
|
|
rule0 = hwsim.rules.create()
|
|
rule0.source = vht_radio.addresses[0]
|
|
rule0.bidirectional = True
|
|
rule0.signal = -2000
|
|
|
|
rule1 = hwsim.rules.create()
|
|
rule1.source = ht_radio.addresses[0]
|
|
rule1.bidirectional = True
|
|
rule1.signal = -2000
|
|
|
|
rule2 = hwsim.rules.create()
|
|
rule2.source = non_ht_radio.addresses[0]
|
|
rule2.bidirectional = True
|
|
rule2.signal = -2000
|
|
|
|
wd = IWD()
|
|
|
|
psk_agent = PSKAgent("secret123")
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
device = wd.list_devices(1)[0]
|
|
|
|
self.do_connect(wd, device, vht_hostapd)
|
|
|
|
# lower VHT BSS signal, HT should now be preferred
|
|
rule0.signal = -6000
|
|
|
|
self.do_connect(wd, device, ht_hostapd)
|
|
|
|
# lower HT BSS signal, basic rate BSS should now be preferred
|
|
rule1.signal = -6000
|
|
|
|
self.do_connect(wd, device, non_ht_hostapd)
|
|
|
|
wd.unregister_psk_agent(psk_agent)
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
pass
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
IWD.clear_storage()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(exit=True)
|