mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2024-11-12 15:09:23 +01:00
a15b781feb
This changes all tests to use the default get_ordered_network behavior rather than some custom or incorrect logic. Any use of scan_if_needed=True has been removed since this is now the default. Also any explicit scanning has been removed for tests which do not require it (where the default behavior is good enough).
98 lines
3.0 KiB
Python
98 lines
3.0 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 hostapd import HostapdCLI
|
|
import testutil
|
|
from config import ctx
|
|
import os
|
|
|
|
class Test(unittest.TestCase):
|
|
|
|
def test_connection_success(self):
|
|
wd = IWD(True, iwd_storage_dir='/tmp/storage')
|
|
|
|
ns0 = ctx.get_namespace('ns0')
|
|
|
|
wd_ns0 = IWD(True, namespace=ns0)
|
|
|
|
psk_agent = PSKAgent("secret123")
|
|
psk_agent_ns0 = PSKAgent("secret123", namespace=ns0)
|
|
wd.register_psk_agent(psk_agent)
|
|
wd_ns0.register_psk_agent(psk_agent_ns0)
|
|
|
|
dev1 = wd.list_devices(1)[0]
|
|
dev2 = wd_ns0.list_devices(1)[0]
|
|
|
|
ordered_network = dev1.get_ordered_network('ssidTKIP')
|
|
|
|
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(dev1, condition)
|
|
|
|
testutil.test_iface_operstate()
|
|
testutil.test_ifaces_connected()
|
|
|
|
testutil.test_ip_address_match(dev1.name, '192.168.1.10')
|
|
|
|
dev2.scan()
|
|
|
|
condition = 'not obj.scanning'
|
|
wd_ns0.wait_for_object_condition(dev2, condition)
|
|
|
|
ordered_network = dev2.get_ordered_network('ssidTKIP', scan_if_needed=True)
|
|
|
|
condition = 'not obj.connected'
|
|
wd_ns0.wait_for_object_condition(ordered_network.network_object, condition)
|
|
|
|
ordered_network.network_object.connect()
|
|
|
|
condition = 'obj.state == DeviceState.connected'
|
|
wd_ns0.wait_for_object_condition(dev2, condition)
|
|
|
|
wd.wait(1)
|
|
testutil.test_ip_address_match(dev1.name, None)
|
|
|
|
dev1.disconnect()
|
|
dev2.disconnect()
|
|
|
|
condition = 'not obj.connected'
|
|
wd.wait_for_object_condition(ordered_network.network_object, condition)
|
|
|
|
wd.unregister_psk_agent(psk_agent)
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
hapd = HostapdCLI()
|
|
# TODO: This could be moved into test-runner itself if other tests ever
|
|
# require this functionality (p2p, FILS, etc.). Since its simple
|
|
# enough it can stay here for now.
|
|
ctx.start_process(['ifconfig', hapd.ifname, '192.168.1.1', 'netmask', '255.255.255.0'],
|
|
wait=True)
|
|
ctx.start_process(['touch', '/tmp/dhcpd.leases'], wait=True)
|
|
cls.dhcpd_pid = ctx.start_process(['dhcpd', '-f', '-cf', '/tmp/dhcpd.conf',
|
|
'-lf', '/tmp/dhcpd.leases',
|
|
hapd.ifname])
|
|
IWD.copy_to_storage('ssidTKIP.psk', '/tmp/storage')
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
IWD.clear_storage()
|
|
cls.dhcpd_pid.kill()
|
|
os.system('rm -rf /tmp/dhcpd.leases')
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(exit=True)
|