mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2024-11-05 03:29:24 +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.
127 lines
3.5 KiB
Python
127 lines
3.5 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 time import sleep
|
|
|
|
# Table beacon with wildcard BSSID
|
|
basic_beacon = '51000000000002ffffffffffff020100'
|
|
# Table beacon with wildcard BSSID and SSID filter
|
|
beacon_with_ssid = '51000000000002ffffffffffff02010000077373696452524d'
|
|
# Passive beacon with wildcard BSSID
|
|
beacon_passive = '510b0000000000ffffffffffff020100'
|
|
# Active beacon with wildcard BSSID
|
|
beacon_active = '510b0000000001ffffffffffff020100'
|
|
# Passive beacon with duration set
|
|
beacon_passive_duration = '510b0000c80000ffffffffffff020100'
|
|
|
|
class Test(unittest.TestCase):
|
|
|
|
def test_connection_success(self):
|
|
hapd = HostapdCLI(config='ssidRRM.conf')
|
|
wd = IWD()
|
|
|
|
psk_agent = PSKAgent("secret123")
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
devices = wd.list_devices(1)
|
|
device = devices[0]
|
|
|
|
condition = 'not obj.scanning'
|
|
wd.wait_for_object_condition(device, condition)
|
|
|
|
device.scan()
|
|
|
|
condition = 'not obj.scanning'
|
|
wd.wait_for_object_condition(device, condition)
|
|
|
|
ordered_network = device.get_ordered_network('ssidRRM')
|
|
|
|
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()
|
|
|
|
hapd.wait_for_event('AP-STA-CONNECTED')
|
|
|
|
# This should return both APs
|
|
hapd.req_beacon(device.address, basic_beacon)
|
|
|
|
for e in ['BEACON-RESP-RX', 'BEACON-RESP-RX']:
|
|
event = hapd.wait_for_event(e)
|
|
if event:
|
|
print(event)
|
|
|
|
sleep(0.5)
|
|
|
|
# This should return just ssidRRM
|
|
hapd.req_beacon(device.address, beacon_with_ssid)
|
|
event = hapd.wait_for_event('BEACON-RESP-RX')
|
|
if event:
|
|
print(event)
|
|
|
|
sleep(0.5)
|
|
|
|
# This should passive scan on channel 11, returning otherSSID
|
|
hapd.req_beacon(device.address, beacon_passive)
|
|
# TODO: See if we are scanning here (scan not initiated from station)
|
|
|
|
event = hapd.wait_for_event('BEACON-RESP-RX')
|
|
if event:
|
|
print(event)
|
|
|
|
sleep(0.5)
|
|
|
|
# This should active scan on channel 11, returning otherSSID
|
|
hapd.req_beacon(device.address, beacon_active)
|
|
# TODO: See if we are scanning here (scan not initiated from station)
|
|
|
|
event = hapd.wait_for_event('BEACON-RESP-RX')
|
|
if event:
|
|
print(event)
|
|
|
|
sleep(0.5)
|
|
|
|
# This should passive scan on channel 11, returning otherSSID
|
|
hapd.req_beacon(device.address, beacon_passive_duration)
|
|
# TODO: See if we are scanning here (scan not initiated from station)
|
|
|
|
event = hapd.wait_for_event('BEACON-RESP-RX')
|
|
if event:
|
|
print(event)
|
|
|
|
device.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):
|
|
pass
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
IWD.clear_storage()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(exit=True)
|