1 files changed, 15 insertions, 11 deletions
diff --git a/scripts/cli/test_interfaces_wireless.py b/scripts/cli/test_interfaces_wireless.py
index 5d04a1c44..fae233244 100755
--- a/scripts/cli/test_interfaces_wireless.py
+++ b/scripts/cli/test_interfaces_wireless.py
@@ -18,6 +18,8 @@ import os
 import unittest
 
 from base_interfaces_test import BasicInterfaceTest
+from psutil import process_iter
+from vyos.util import check_kmod
 
 class WirelessInterfaceTest(BasicInterfaceTest.BaseTest):
     def setUp(self):
@@ -37,18 +39,20 @@ class WirelessInterfaceTest(BasicInterfaceTest.BaseTest):
         self._interfaces = list(self._options)
         self.session.set(['system', 'wifi-regulatory-domain', 'SE'])
 
-    def test_wifi_client(self):
-        """ test creation of a wireless station """
-        for intf in self._interfaces:
-            # prepare interfaces
-            for option in self._options.get(intf, []):
-                self.session.set(self._base_path + [intf] + option.split())
-
-            # commit changes
-            self.session.commit()
-
+    def test_add_address_single(self):
+        """ derived method to check if member interfaces are enslaved properly """
+        super().test_add_address_single()
 
+        for option, option_value in self._options.items():
+            if 'type access-point' in option_value:
+                # Check for running process
+                self.assertIn('hostapd', (p.name() for p in process_iter()))
+            elif 'type station' in option_value:
+                # Check for running process
+                self.assertIn('wpa_supplicant', (p.name() for p in process_iter()))
+            else:
+                self.assertTrue(False)
 
 if __name__ == '__main__':
-    os.system('sudo modprobe mac80211_hwsim')
+    check_kmod('mac80211_hwsim')
     unittest.main()