mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2026-02-07 02:28:02 +01:00
5b58390b8d
testAgent had a few tests which weren't reliable, and one was not actually testing anything, or at least not what the name implied it should be testing. The first issue was using iwctl in the first place. There is not a reliable way to know when iwctl has registered its agent so relying on that with a sleep, or waiting for the service to become available isn't 100% fool proof. To fix this use the updated PSKAgent which allows multiple to be registered. This ensures the agent is ready for requests. This test was also renamed to be consistent with what its actually testing: that IWD uses the first agent registered. This removes test_connection_with_other_agent as well because this test case is covered by the client test itself. There is no need to re-test iwctl's agent functionality here.
84 lines
1.9 KiB
Python
84 lines
1.9 KiB
Python
#!/usr/bin/python3
|
|
|
|
import unittest
|
|
import sys
|
|
import dbus
|
|
|
|
sys.path.append('../util')
|
|
import iwd
|
|
from iwd import IWD
|
|
from iwd import PSKAgent
|
|
from iwd import NetworkType
|
|
import testutil
|
|
import subprocess
|
|
from config import ctx
|
|
|
|
class Test(unittest.TestCase):
|
|
|
|
def check_connection(self, wd, ssid):
|
|
|
|
device = wd.list_devices(1)[0]
|
|
ordered_network = device.get_ordered_network(ssid)
|
|
|
|
ordered_network.network_object.connect()
|
|
|
|
condition = 'obj.state == DeviceState.connected'
|
|
wd.wait_for_object_condition(device, condition)
|
|
|
|
condition = 'obj.connected_network is not None'
|
|
wd.wait_for_object_condition(device, condition)
|
|
|
|
testutil.test_iface_operstate(device.name)
|
|
device.disconnect()
|
|
|
|
condition = 'obj.state == DeviceState.disconnected'
|
|
wd.wait_for_object_condition(device, condition)
|
|
|
|
def test_connection_with_no_agent(self):
|
|
|
|
wd = IWD()
|
|
|
|
with self.assertRaises(iwd.NoAgentEx):
|
|
self.check_connection(wd, 'ssid1')
|
|
|
|
IWD.clear_storage()
|
|
|
|
def test_connection_with_own_agent(self):
|
|
|
|
wd = IWD()
|
|
|
|
psk_agent = PSKAgent("secret_ssid1")
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
self.check_connection(wd, 'ssid1')
|
|
|
|
wd.unregister_psk_agent(psk_agent)
|
|
|
|
IWD.clear_storage()
|
|
|
|
def test_connection_use_first_from_multiple_registered(self):
|
|
wd = IWD()
|
|
|
|
psk_agent = PSKAgent("secret_ssid1")
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
psk_agent = PSKAgent("secret_ssid2")
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
self.check_connection(wd, 'ssid1')
|
|
|
|
wd.unregister_psk_agent(psk_agent)
|
|
|
|
IWD.clear_storage()
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
pass
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
IWD.clear_storage()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(exit=True)
|