3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-11-22 14:49:24 +01:00
iwd/autotests/testHT-VHT/connection_test.py
James Prestwood 9594b03a35 auto-t: prepare autotests for test-runner re-write
The tests basically remained the same with a few minor changes.
The wiphy_map and in turn hostapd_map are no longer used. This
was already partially converted a long time ago when the 'config'
parameter was added to HostapdCLI. This patch fully converts all
autotests to use 'config' rather than looking up by interface.

Some test scripts were named 'test.py' which was fine before but
the new rewrite actually loads each python test as a module. The
name 'test' is too ambiguous and causes issues due to a native
python module with the same name. All of these files were
renamed to 'connection_test.py'.
2020-09-10 17:59:49 -05:00

110 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 time import sleep
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.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
sleep(1)
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)