3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-12-23 14:12:47 +01:00
iwd/autotests/testKnownNetworks/frequency_test.py
James Prestwood a15b781feb auto-t: update all tests to remove bad scanning logic
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).
2021-08-12 16:59:03 -05:00

139 lines
4.1 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
import testutil
import os
from configparser import ConfigParser
class Test(unittest.TestCase):
def connect_network(self, wd, device, network):
ordered_network = device.get_ordered_network(network)
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)
device.disconnect()
condition = 'not obj.connected'
wd.wait_for_object_condition(ordered_network.network_object, condition)
def test_connection_success(self):
wd = IWD(True, '/tmp')
psk_agent = PSKAgent("secret123")
wd.register_psk_agent(psk_agent)
devices = wd.list_devices(1)
device = devices[0]
#
# Connect to the PSK network, then Hotspot so IWD creates 2 entries in
# the known frequency file.
#
self.connect_network(wd, device, 'ssidCCMP')
wd.unregister_psk_agent(psk_agent)
psk_agent = PSKAgent('abc', ('domain\\user', 'testpasswd'))
wd.register_psk_agent(psk_agent)
self.connect_network(wd, device, 'Hotspot')
wd.unregister_psk_agent(psk_agent)
psk_freqs = None
psk_uuid = None
hs20_freqs = None
hs20_uuid = None
config = ConfigParser()
config.read('/tmp/iwd/.known_network.freq')
for s in config.sections():
if os.path.basename(config[s]['name']) == 'ssidCCMP.psk':
psk_freqs = config[s]['list']
psk_freqs = psk_freqs.split(' ')
psk_uuid = s
elif os.path.basename(config[s]['name']) == 'example.conf':
hs20_freqs = config[s]['list']
hs20_freqs = hs20_freqs.split(' ')
hs20_uuid = s
#
# Verify the frequencies are what we expect
#
self.assertIsNotNone(psk_freqs)
self.assertIsNotNone(psk_uuid)
self.assertIn('5180', psk_freqs)
self.assertIn('2412', psk_freqs)
self.assertIsNotNone(hs20_freqs)
self.assertIsNotNone(hs20_uuid)
self.assertIn('2412', hs20_freqs)
#
# Forget all know networks, this should remove all entries in the
# known frequencies file.
#
for n in wd.list_known_networks():
n.forget()
psk_agent = PSKAgent("secret123")
wd.register_psk_agent(psk_agent)
#
# Reconnect, this should generate a completely new UUID since we
# previously forgot the network.
#
self.connect_network(wd, device, 'ssidCCMP')
wd.unregister_psk_agent(psk_agent)
#
# Ensure that a new UUID was created and that we still have the same
# frequencies listed.
#
psk_freqs = None
psk_uuid2 = None
hs20_freqs = None
config = ConfigParser()
config.read('/tmp/iwd/.known_network.freq')
for s in config.sections():
self.assertNotEqual(os.path.basename(config[s]['name']),
'example.conf')
if os.path.basename(config[s]['name']) == 'ssidCCMP.psk':
psk_freqs = config[s]['list']
psk_freqs = psk_freqs.split(' ')
psk_uuid2 = s
self.assertIsNotNone(psk_freqs)
self.assertIsNotNone(psk_uuid2)
self.assertNotEqual(psk_uuid, psk_uuid2)
self.assertIn('5180', psk_freqs)
self.assertIn('2412', psk_freqs)
@classmethod
def setUpClass(cls):
IWD.copy_to_hotspot('example.conf')
conf = '[General]\nDisableANQP=0\n'
os.system('echo "%s" > /tmp/main.conf' % conf)
@classmethod
def tearDownClass(cls):
IWD.clear_storage()
os.remove('/tmp/main.conf')
if __name__ == '__main__':
unittest.main(exit=True)