diff --git a/autotests/testWPA2/connection_test.py b/autotests/testWPA2/connection_test.py index bf8831ea..2cbdf8e1 100644 --- a/autotests/testWPA2/connection_test.py +++ b/autotests/testWPA2/connection_test.py @@ -8,20 +8,19 @@ import iwd from iwd import IWD from iwd import PSKAgent from iwd import NetworkType +from hostapd import HostapdCLI import testutil class Test(unittest.TestCase): - def test_connection_success(self): - wd = IWD() - + def validate_connection_success(self, wd): psk_agent = PSKAgent("secret123") wd.register_psk_agent(psk_agent) devices = wd.list_devices(1) device = devices[0] - ordered_network = device.get_ordered_network('ssidCCMP') + ordered_network = device.get_ordered_network('ssidWPA2') self.assertEqual(ordered_network.type, NetworkType.psk) @@ -43,13 +42,32 @@ class Test(unittest.TestCase): wd.unregister_psk_agent(psk_agent) + def test_ccmp(self): + self.hostapd.set_value('rsn_pairwise', 'CCMP') + self.hostapd.reload() + self.hostapd.wait_for_event("AP-ENABLED") + self.validate_connection_success(self.wd) + + def test_gcmp(self): + self.hostapd.set_value('rsn_pairwise', 'GCMP') + self.hostapd.reload() + self.hostapd.wait_for_event("AP-ENABLED") + self.validate_connection_success(self.wd) + + def setUp(self): + self.wd = IWD(True) + + def tearDown(self): + self.wd.clear_storage() + self.wd = None + @classmethod def setUpClass(cls): - pass + cls.hostapd = HostapdCLI(config='ssidWPA2.conf') @classmethod def tearDownClass(cls): - IWD.clear_storage() + pass if __name__ == '__main__': unittest.main(exit=True) diff --git a/autotests/testWPA2/failure_test.py b/autotests/testWPA2/failure_test.py index 605ee5dd..61cf6ade 100644 --- a/autotests/testWPA2/failure_test.py +++ b/autotests/testWPA2/failure_test.py @@ -13,7 +13,7 @@ import testutil class Test(unittest.TestCase): def test_connection_success(self): - wd = IWD() + wd = IWD(True) psk_agent = PSKAgent("InvalidPassword") wd.register_psk_agent(psk_agent) @@ -22,7 +22,7 @@ class Test(unittest.TestCase): self.assertIsNotNone(devices) device = devices[0] - ordered_network = device.get_ordered_network('ssidCCMP') + ordered_network = device.get_ordered_network('ssidWPA2') self.assertEqual(ordered_network.type, NetworkType.psk) diff --git a/autotests/testWPA2/hw.conf b/autotests/testWPA2/hw.conf index fad15b88..51350f98 100644 --- a/autotests/testWPA2/hw.conf +++ b/autotests/testWPA2/hw.conf @@ -1,5 +1,6 @@ [SETUP] num_radios=2 +start_iwd=0 [HOSTAPD] -rad0=ssidCCMP.conf +rad0=ssidWPA2.conf diff --git a/autotests/testWPA2/password_test.py b/autotests/testWPA2/password_test.py index d937d5c0..1dbfce00 100644 --- a/autotests/testWPA2/password_test.py +++ b/autotests/testWPA2/password_test.py @@ -13,7 +13,7 @@ import testutil class Test(unittest.TestCase): def test_connection_success(self): - wd = IWD() + wd = IWD(True) devices = wd.list_devices(1) device = devices[0] @@ -22,7 +22,7 @@ class Test(unittest.TestCase): condition = 'obj.state == DeviceState.disconnected' wd.wait_for_object_condition(device, condition) - ordered_network = device.get_ordered_network("ssidCCMP") + ordered_network = device.get_ordered_network("ssidWPA2") self.assertEqual(ordered_network.type, NetworkType.psk) network = ordered_network.network_object diff --git a/autotests/testWPA2/ssidCCMP.conf b/autotests/testWPA2/ssidWPA2.conf similarity index 83% rename from autotests/testWPA2/ssidCCMP.conf rename to autotests/testWPA2/ssidWPA2.conf index 074e8228..64fd7796 100644 --- a/autotests/testWPA2/ssidCCMP.conf +++ b/autotests/testWPA2/ssidWPA2.conf @@ -1,6 +1,6 @@ hw_mode=g channel=1 -ssid=ssidCCMP +ssid=ssidWPA2 wpa=2 wpa_pairwise=CCMP