3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-12-24 23:42:42 +01:00
iwd/autotests/testWPA2/password_test.py
2018-12-03 14:21:30 -06:00

74 lines
2.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
class Test(unittest.TestCase):
def test_connection_success(self):
wd = IWD()
devices = wd.list_devices(1);
device = devices[0]
device.disconnect()
condition = 'obj.state == DeviceState.disconnected'
wd.wait_for_object_condition(device, condition)
condition = 'not obj.scanning'
wd.wait_for_object_condition(device, condition)
if not device.get_ordered_networks():
device.scan()
condition = 'obj.scanning'
wd.wait_for_object_condition(device, condition)
condition = 'not obj.scanning'
wd.wait_for_object_condition(device, condition)
ordered_networks = device.get_ordered_networks()
self.assertEqual(len(ordered_networks), 1)
ordered_network = ordered_networks[0]
self.assertEqual(ordered_network.name, "ssidCCMP")
self.assertEqual(ordered_network.type, NetworkType.psk)
network = ordered_network.network_object
# 0 chars
psk_agent = PSKAgent("")
wd.register_psk_agent(psk_agent)
self.assertRaises(iwd.InvalidFormatEx, network.connect)
wd.unregister_psk_agent(psk_agent)
# 7 chars
psk_agent = PSKAgent("a" * 7)
wd.register_psk_agent(psk_agent)
self.assertRaises(iwd.InvalidFormatEx, network.connect)
wd.unregister_psk_agent(psk_agent)
# 64 chars
psk_agent = PSKAgent("a" * 64)
wd.register_psk_agent(psk_agent)
self.assertRaises(iwd.InvalidFormatEx, network.connect)
wd.unregister_psk_agent(psk_agent)
# 64k chars
psk_agent = PSKAgent("a" * 65536)
wd.register_psk_agent(psk_agent)
self.assertRaises(iwd.InvalidFormatEx, network.connect)
wd.unregister_psk_agent(psk_agent)
del wd
@classmethod
def tearDownClass(cls):
IWD.clear_storage()
if __name__ == '__main__':
unittest.main(exit=True)