3
0
mirror of https://git.kernel.org/pub/scm/network/wireless/iwd.git synced 2024-11-22 06:29:23 +01:00

auto-t: add client test to testEAP-WPS

Tests iwctl functionality with wsc commands
This commit is contained in:
James Prestwood 2022-06-30 11:03:27 -07:00 committed by Denis Kenzior
parent d127c5b997
commit ac2aee288f
2 changed files with 23 additions and 10 deletions

View File

@ -4,22 +4,23 @@ import unittest
import sys
sys.path.append('../util')
import iwd
from iwd import IWD
from iwd import DeviceState
from hostapd import HostapdCLI
from config import ctx
class Test(unittest.TestCase):
def four_digit_pin_success(self, wd):
def four_digit_pin_success(self, wd, client=False):
devices = wd.list_devices(1)
device = devices[0]
pin = '1234'
self.hostapd.wps_pin(pin)
if not client:
device.wps_start_pin(pin)
else:
ctx.start_process(['iwctl', 'wsc', device.name, 'start-user-pin', pin], check=True)
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(device, condition)
@ -36,6 +37,11 @@ class Test(unittest.TestCase):
self.four_digit_pin_success(wd)
def test_client_four_digit_pin(self):
wd = IWD(True)
self.four_digit_pin_success(wd, client=True)
@classmethod
def setUpClass(cls):
cls.hostapd = HostapdCLI(config='ssidWPS.conf')

View File

@ -4,20 +4,22 @@ import unittest
import sys
sys.path.append('../util')
import iwd
from iwd import IWD
from iwd import DeviceState
from hostapd import HostapdCLI
from config import ctx
class Test(unittest.TestCase):
def push_button_success(self, wd):
def push_button_success(self, wd, client=False):
self.hostapd.wps_push_button()
devices = wd.list_devices(1)
device = devices[0]
if not client:
device.wps_push_button()
else:
ctx.start_process(['iwctl', 'wsc', device.name, 'push-button'], check=True)
condition = 'obj.state == DeviceState.connected'
wd.wait_for_object_condition(device, condition)
@ -34,6 +36,11 @@ class Test(unittest.TestCase):
self.push_button_success(wd)
def test_client_push_button(self):
wd = IWD(True)
self.push_button_success(wd, client=True)
@classmethod
def setUpClass(cls):
cls.hostapd = HostapdCLI(config='ssidWPS.conf')