mirror of
https://git.kernel.org/pub/scm/network/wireless/iwd.git
synced 2024-11-09 21:49:23 +01:00
c73575585b
Don't use del wd to dereference the IWD instance at the end of the function where it has been defined in the first place as at this point wd is about to have its reference count decreased anyway (the variable's scope is ending) so it's pointless (but didn't hurt). Relying on the __del__ destructor to kill the IWD process in those tests it has been started in the constructor is a bit of a hack in the first place, because the destructor is called on garbage collection and even through CPython does this on the refcount reaching 0, this is not documented and there's no guideline on when it should happen or if it should happen at all. So it could be argued that we should keep the del wd statemenets to be able to easily replace all of them with a call to a new method. But most of them are not placed so that they're guaranteed to happen on test success or failure. It would probably be easier to do this and other housekeeping in a base class and make the tests its subclasses. Also some of these tests don't really need to launch iwd themselves, since IWD now tracks changes in the known network files I think IWD only really needs to be killed between tests when main.conf changes.
103 lines
2.9 KiB
Python
103 lines
2.9 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
|
|
|
|
class TestMFP(unittest.TestCase):
|
|
'''
|
|
The bellow test cases excesise the following MFP option setting scenarios:
|
|
|
|
IWD_MFP: AP_MFP: Result:
|
|
0 0 No MFP, connection succeeds
|
|
0 1 No MFP, connection succeeds
|
|
0 2 Not capable error
|
|
1 0 No MFP, connection succeeds
|
|
1 1 MFP enabled, connection succeeds
|
|
1 2 MFP enabled, connection succeeds
|
|
2 0 Not capable error
|
|
2 1 MFP enabled, connection succeeds
|
|
2 2 MFP enabled, connection succeeds
|
|
|
|
where:
|
|
0 - MFP is disabled
|
|
1 - MFP is optional
|
|
2 - MFP is required
|
|
'''
|
|
|
|
def check_mfp_connection(self, wd, device, ssid, throws_exception):
|
|
ordered_network = device.get_ordered_network(ssid)
|
|
|
|
condition = 'not obj.connected'
|
|
wd.wait_for_object_condition(ordered_network.network_object, condition)
|
|
|
|
if throws_exception:
|
|
with self.assertRaises(iwd.NotSupportedEx):
|
|
ordered_network.network_object.connect()
|
|
return
|
|
else:
|
|
ordered_network.network_object.connect()
|
|
|
|
condition = 'obj.connected'
|
|
wd.wait_for_object_condition(ordered_network.network_object, condition)
|
|
|
|
device.disconnect()
|
|
|
|
condition = 'not obj.connected'
|
|
wd.wait_for_object_condition(ordered_network.network_object, condition)
|
|
|
|
IWD.clear_storage()
|
|
|
|
def stage_iwd(self, wd, config_dir):
|
|
psk_agent = PSKAgent( ['secret123', 'secret123', 'secret123'] )
|
|
wd.register_psk_agent(psk_agent)
|
|
|
|
devices = wd.list_devices(1)
|
|
self.assertIsNotNone(devices)
|
|
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)
|
|
|
|
if config_dir == '/tmp/IWD-MFP2':
|
|
self.check_mfp_connection(wd, device, 'ssidMFP0', True)
|
|
else:
|
|
self.check_mfp_connection(wd, device, 'ssidMFP0', False)
|
|
|
|
self.check_mfp_connection(wd, device, 'ssidMFP1', False)
|
|
|
|
if config_dir == '/tmp/IWD-MFP0':
|
|
self.check_mfp_connection(wd, device, 'ssidMFP2', True)
|
|
else:
|
|
self.check_mfp_connection(wd, device, 'ssidMFP2', False)
|
|
|
|
wd.unregister_psk_agent(psk_agent)
|
|
|
|
def test_iwd_mfp0(self):
|
|
wd = IWD(True, '/tmp/IWD-MFP0')
|
|
|
|
self.stage_iwd(wd, '/tmp/IWD-MFP0')
|
|
|
|
def test_iwd_mfp1(self):
|
|
wd = IWD(True, '/tmp/IWD-MFP1')
|
|
|
|
self.stage_iwd(wd, '/tmp/IWD-MFP1')
|
|
|
|
def test_iwd_mfp2(self):
|
|
wd = IWD(True, '/tmp/IWD-MFP2')
|
|
|
|
self.stage_iwd(wd, '/tmp/IWD-MFP2')
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(exit=True)
|