mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2024-11-20 04:19:25 +01:00
d1c5e42ae2
All the AP tests were basically doing the same validation. Refactor this code into validation.py and import that into the tests.
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
from iwd import PSKAgent
|
|
from iwd import NetworkType
|
|
from hostapd import HostapdCLI
|
|
import testutil
|
|
|
|
def validate(wd, sta_dev, ap_dev, ssid, passphrase,
|
|
sta_ip_info=None, ap_ip_info=None, ip_checks=True):
|
|
try:
|
|
network = sta_dev.get_ordered_network(ssid, full_scan=True)
|
|
|
|
if network.type != NetworkType.psk:
|
|
raise Exception("Network type mismatch")
|
|
|
|
psk_agent = PSKAgent(passphrase)
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
network.network_object.connect()
|
|
|
|
condition = 'obj.state == DeviceState.connected'
|
|
wd.wait_for_object_condition(sta_dev, condition)
|
|
|
|
testutil.test_iface_operstate(sta_dev.name)
|
|
|
|
# This implies separate namespaces so the iface names won't exist
|
|
if not sta_ip_info or not ap_ip_info:
|
|
testutil.test_ifaces_connected(ap_dev.name, sta_dev.name, group=False)
|
|
|
|
if not ip_checks:
|
|
return
|
|
|
|
if sta_ip_info:
|
|
testutil.test_ip_address_match(sta_dev.name, sta_ip_info[0])
|
|
|
|
if sta_ip_info and ap_ip_info:
|
|
testutil.test_ip_connected(sta_ip_info, ap_ip_info)
|
|
|
|
wd.unregister_psk_agent(psk_agent)
|
|
|
|
sta_dev.disconnect()
|
|
|
|
condition = 'not obj.connected'
|
|
wd.wait_for_object_condition(network.network_object, condition)
|
|
finally:
|
|
if ip_checks:
|
|
ap_dev.stop_ap()
|
|
|
|
def client_connect(wd, dev, ssid):
|
|
hostapd = HostapdCLI(config='psk-ccmp.conf')
|
|
|
|
ordered_network = dev.get_ordered_network(ssid)
|
|
|
|
if ordered_network.type != NetworkType.psk:
|
|
raise Exception("Network type mismatch")
|
|
|
|
psk_agent = PSKAgent('Password1')
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
ordered_network.network_object.connect()
|
|
|
|
condition = 'obj.state == DeviceState.connected'
|
|
wd.wait_for_object_condition(dev, condition)
|
|
|
|
wd.unregister_psk_agent(psk_agent)
|
|
|
|
testutil.test_iface_operstate(dev.name)
|
|
testutil.test_ifaces_connected(hostapd.ifname, dev.name)
|
|
|
|
dev.disconnect()
|
|
|
|
condition = 'not obj.connected'
|
|
wd.wait_for_object_condition(ordered_network.network_object, condition)
|