3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-11-05 11:39:24 +01:00
iwd/autotests/testHT-VHT/connection_test.py
James Prestwood d232edde72 auto-t: replace waiting for network with device
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.
2020-09-14 16:03:04 -05:00

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)