diff --git a/autotests/testEAP-AKA/connection_test.py b/autotests/testEAP-AKA/connection_test.py index 5717ecbc..a80431a8 100644 --- a/autotests/testEAP-AKA/connection_test.py +++ b/autotests/testEAP-AKA/connection_test.py @@ -9,9 +9,22 @@ from iwd import IWD from iwd import NetworkType from hlrauc import AuthCenter +from hostapd import HostapdCLI +from hostapd import hostapd_map + class Test(unittest.TestCase): def test_connection_success(self): + hostapd = None + + for hostapd_if in list(hostapd_map.values()): + hpd = HostapdCLI(hostapd_if) + if hpd.get_config_value('ssid') == 'ssidEAP-AKA': + hostapd = hpd + break + + self.assertIsNotNone(hostapd) + auth = AuthCenter('/tmp/hlrauc.sock', '/tmp/sim.db') wd = IWD() @@ -46,6 +59,13 @@ class Test(unittest.TestCase): condition = 'obj.connected' wd.wait_for_object_condition(ordered_network.network_object, condition) + hostapd.eapol_reauth(device.address) + + wd.wait(10) + + condition = 'obj.connected' + wd.wait_for_object_condition(ordered_network.network_object, condition) + device.disconnect() condition = 'not obj.connected' diff --git a/autotests/testEAP-SIM/connection_test.py b/autotests/testEAP-SIM/connection_test.py index 678e88b4..8037bd20 100644 --- a/autotests/testEAP-SIM/connection_test.py +++ b/autotests/testEAP-SIM/connection_test.py @@ -9,9 +9,20 @@ from iwd import IWD from iwd import NetworkType from hlrauc import AuthCenter +from hostapd import HostapdCLI +from hostapd import hostapd_map + class Test(unittest.TestCase): def test_connection_success(self): + hostapd = None + + for hostapd_if in list(hostapd_map.values()): + hpd = HostapdCLI(hostapd_if) + if hpd.get_config_value('ssid') == 'ssidEAP-SIM': + hostapd = hpd + break + auth = AuthCenter('/tmp/hlrauc.sock', '/tmp/sim.db') wd = IWD() @@ -46,6 +57,13 @@ class Test(unittest.TestCase): condition = 'obj.connected' wd.wait_for_object_condition(ordered_network.network_object, condition) + hostapd.eapol_reauth(device.address) + + wd.wait(10) + + condition = 'obj.connected' + wd.wait_for_object_condition(ordered_network.network_object, condition) + device.disconnect() condition = 'not obj.connected'