121 lines
3.1 KiB
Python
121 lines
3.1 KiB
Python
![]() |
#!/usr/bin/python3
|
||
|
|
||
|
import unittest
|
||
|
import sys
|
||
|
|
||
|
sys.path.append('../util')
|
||
|
import iwd
|
||
|
from iwd import IWD
|
||
|
import testutil
|
||
|
import subprocess
|
||
|
|
||
|
class Test(unittest.TestCase):
|
||
|
|
||
|
def check_connection_success(self, ssid):
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[0]
|
||
|
|
||
|
condition = 'not obj.scanning'
|
||
|
wd.wait_for_object_condition(device, condition)
|
||
|
|
||
|
ordered_network = device.get_ordered_network(ssid)
|
||
|
|
||
|
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)
|
||
|
|
||
|
def test_connection_with_passphrase(self):
|
||
|
ssid = 'ssidPassphrase'
|
||
|
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[0]
|
||
|
|
||
|
subprocess.check_call(['iwctl', '-P', 'passphrase',
|
||
|
'station', device.name, 'connect', ssid])
|
||
|
|
||
|
self.check_connection_success(ssid)
|
||
|
|
||
|
def test_connection_with_username_and_password(self):
|
||
|
ssid = 'ssidUNameAndPWord'
|
||
|
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[0]
|
||
|
|
||
|
subprocess.check_call(['iwctl', '-u', 'user', '-p', 'password',
|
||
|
'station', device.name, 'connect', ssid])
|
||
|
|
||
|
self.check_connection_success(ssid)
|
||
|
|
||
|
def test_connection_with_password(self):
|
||
|
ssid = 'ssidPWord'
|
||
|
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[0]
|
||
|
|
||
|
subprocess.check_call(['iwctl', '-p', 'password',
|
||
|
'station', device.name, 'connect', ssid])
|
||
|
|
||
|
self.check_connection_success(ssid)
|
||
|
|
||
|
def test_connection_failure(self):
|
||
|
ssid = 'ssidPassphrase'
|
||
|
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[0]
|
||
|
|
||
|
with self.assertRaises(subprocess.CalledProcessError):
|
||
|
subprocess.check_call(['iwctl', '-P', 'incorrect_passphrase',
|
||
|
'station', device.name, 'connect', ssid])
|
||
|
|
||
|
def test_invalid_command_line_option(self):
|
||
|
ssid = 'ssidPassphrase'
|
||
|
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[0]
|
||
|
|
||
|
with self.assertRaises(subprocess.CalledProcessError):
|
||
|
subprocess.check_call(['iwctl', '-z',
|
||
|
'station', device.name, 'connect', ssid])
|
||
|
|
||
|
def test_invalid_command(self):
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[0]
|
||
|
|
||
|
with self.assertRaises(subprocess.CalledProcessError):
|
||
|
subprocess.check_call(['iwctl', 'inexistent', 'command'])
|
||
|
|
||
|
@classmethod
|
||
|
def setUpClass(cls):
|
||
|
IWD.copy_to_storage('ssidUNameAndPWord.8021x')
|
||
|
IWD.copy_to_storage('ssidPWord.8021x')
|
||
|
|
||
|
wd = IWD()
|
||
|
|
||
|
device = wd.list_devices(1)[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)
|
||
|
|
||
|
@classmethod
|
||
|
def tearDownClass(cls):
|
||
|
IWD.clear_storage()
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
unittest.main(exit=True)
|