3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-12-27 19:22:34 +01:00
iwd/autotests/testHotspot/anqp_delay_test.py
James Prestwood cb4f1d2a99 auto-t: turn off scan address randomization in testHotspot
This (hopefully) will make this test pass better on slower machines.
In addition the mechanism of copying over separate main.conf files
was changed (rather than echo'ing the option into /tmp/main.conf)
2021-08-13 20:39:13 -05:00

99 lines
2.9 KiB
Python

#!/usr/bin/python3
import unittest
import sys
import os
sys.path.append('../util')
from iwd import IWD
from iwd import IWD_CONFIG_DIR
from iwd import PSKAgent
from iwd import NetworkType
from hostapd import HostapdCLI
from hwsim import Hwsim
import testutil
from time import sleep
class Test(unittest.TestCase):
def test_connection_success(self):
hwsim = Hwsim()
bss_radio = hwsim.get_radio('rad0')
rule0 = hwsim.rules.create()
rule0.source = bss_radio.addresses[0]
rule0.bidirectional = True
wd = IWD(True)
hapd = HostapdCLI(config='ssidHotspot.conf')
psk_agent = PSKAgent('abc', ('domain\\user', 'testpasswd'))
wd.register_psk_agent(psk_agent)
devices = wd.list_devices(1)
device = devices[0]
device.autoconnect = True
# We are dependent on a periodic scan here. We want to wait for this
# because this is the first opportunity IWD has to do ANQP. Once ANQP
# has been done once the network is set up and we cannot simulate the
# 'Connect() before ANQP' race condition anymore.
condition = 'obj.scanning'
wd.wait_for_object_condition(device, condition)
condition = 'not obj.scanning'
wd.wait_for_object_condition(device, condition)
# If no networks were found we likely had a quick scan. Try again to
# allow the full autoconnect scan to happen.
try:
ordered_network = device.get_ordered_network('Hotspot')
except:
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('Hotspot')
self.assertEqual(ordered_network.type, NetworkType.eap)
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
# Force the case where ANQP does not finish before Connect() comes in
rule0.delay = 100
rule0.prefix = '0d'
ordered_network.network_object.connect(wait=False)
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(device, condition)
hapd.wait_for_event('AP-STA-CONNECTED')
testutil.test_iface_operstate()
testutil.test_ifaces_connected(device.name, hapd.ifname)
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):
IWD.copy_to_hotspot('example.conf')
IWD.copy_to_storage('anqp_enabled.conf', storage_dir=IWD_CONFIG_DIR, name='main.conf')
@classmethod
def tearDownClass(cls):
IWD.clear_storage()
os.remove('/tmp/main.conf')
if __name__ == '__main__':
unittest.main(exit=True)