52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
#! /usr/bin/python3
|
|
|
|
import unittest
|
|
|
|
from iwd import IWD
|
|
from config import ctx
|
|
from validation import validate, client_connect
|
|
|
|
class Test(unittest.TestCase):
|
|
def test_connection_success(self):
|
|
wd = IWD(True)
|
|
|
|
dev1, dev2 = wd.list_devices(2)
|
|
|
|
client_connect(wd, dev1, 'TestAP1')
|
|
|
|
dev1.start_ap('TestAP2', 'Password2')
|
|
|
|
validate(wd, dev2, dev1, 'TestAP2', 'Password2')
|
|
|
|
# Finally test dev1 can go to client mode and connect again
|
|
client_connect(wd, dev1, 'TestAP1')
|
|
|
|
def test_client_start_ap(self):
|
|
wd = IWD(True)
|
|
|
|
dev1, dev2 = wd.list_devices(2)
|
|
|
|
ctx.start_process(['iwctl', 'device', dev1.name, 'set-property', 'Mode', 'ap'], check=True)
|
|
ctx.start_process(['iwctl', 'ap', dev1.name, 'start', 'TestAP2', 'Password2'], check=True)
|
|
|
|
iwctl = ctx.start_process(['iwctl', 'ap', 'list'], check=True)
|
|
|
|
self.assertIn(dev1.name, iwctl.out)
|
|
|
|
iwctl = ctx.start_process(['iwctl', 'ap', dev1.name, 'show'], check=True)
|
|
|
|
self.assertIn('TestAP2', iwctl.out)
|
|
|
|
validate(wd, dev2, dev1, 'TestAP2', 'Password2')
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
IWD.copy_to_storage('TestAP1.psk')
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
IWD.clear_storage()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(exit=True)
|