tests: add wifi AP tests

aparcar · aparcar · commit 3e044131fc8a · 2025-07-12T15:16:47.000+02:00
Signed-off-by: Paul Spooren &lt;mail@aparcar.org&gt;
diff --git a/targets/cznic_turris-omnia.yaml b/targets/cznic_turris-omnia.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/enterasys_ws-ap3710i.yaml b/targets/enterasys_ws-ap3710i.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/genexis_pulse-ex400.yaml b/targets/genexis_pulse-ex400.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/hpe_msm460.yaml b/targets/hpe_msm460.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/linksys_e8450.yaml b/targets/linksys_e8450.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/openwrt_one.yaml b/targets/openwrt_one.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/tplink_tl-wdr3600-v1.yaml b/targets/tplink_tl-wdr3600-v1.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/tplink_tl-wdr4300-v1.yaml b/targets/tplink_tl-wdr4300-v1.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/tplink_tl-wr1043nd-v3.yaml b/targets/tplink_tl-wr1043nd-v3.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/targets/tplink_tl-wr842n-v3.yaml b/targets/tplink_tl-wr842n-v3.yaml
@@ -1,5 +1,7 @@
 targets:
   main:
+    features:
+      - wifi
     resources:
       RemotePlace:
         name: !template "$LG_PLACE"
diff --git a/tests/test_wifi.py b/tests/test_wifi.py
@@ -1,6 +1,92 @@
+import time
+
 import pytest
 
 
+def restart_wifi_and_wait(ssh_command, timeout=5):
+    """
+    Helper function to restart wifi via ubus and wait for it to settle.
+
+    Args:
+        ssh_command: SSH command fixture for executing commands
+        timeout: Maximum time to wait for wifi to settle (default: 5 seconds)
+
+    Returns:
+        bool: True if wifi restarted successfully, False if timed out
+    """
+    # Restart wifi
+    ssh_command.run("wifi down")
+    time.sleep(2)
+    ssh_command.run("wifi up")
+    time.sleep(2)
+
+    # Wait till network reload finished
+    result = ssh_command.run(f"ubus -t {timeout} wait_for hostapd.phy0-ap0")[0]
+    return "timed out" not in "\n".join(result)
+
+
+@pytest.mark.lg_feature("wifi")
+def test_wifi_wpa3(ssh_command):
+    """
+    Test enabling a wifi network with WPA3 encryption.
+
+    This test configures a wifi network with WPA3 encryption and password 'openwrt4all'.
+    """
+    ssh_command.run("uci delete wireless.radio0.disabled")
+    ssh_command.run("uci set wireless.default_radio0.encryption=sae")
+    ssh_command.run("uci set wireless.default_radio0.key=openwrt4all")
+
+    ssh_command.run("uci commit")
+
+    restart_wifi_and_wait(ssh_command)
+
+    iwinfo_output = "\n".join(ssh_command.run("iwinfo")[0])
+
+    assert "Mode: Master" in iwinfo_output
+    assert "Encryption: WPA3 SAE (CCMP)" in iwinfo_output
+
+
+@pytest.mark.lg_feature("wifi")
+def test_wifi_wpa2(ssh_command):
+    """
+    Test enabling a wifi network with WPA2 encryption.
+
+    This test configures a wifi network with WPA2 encryption and password 'openwrt4all'.
+    """
+    ssh_command.run("uci delete wireless.radio0.disabled")
+    ssh_command.run("uci set wireless.default_radio0.encryption=psk2")
+    ssh_command.run("uci set wireless.default_radio0.key=openwrt4all")
+
+    ssh_command.run("uci commit")
+
+    restart_wifi_and_wait(ssh_command)
+
+    iwinfo_output = "\n".join(ssh_command.run("iwinfo")[0])
+
+    assert "Mode: Master" in iwinfo_output
+    assert "Encryption: WPA2 PSK (CCMP)" in iwinfo_output
+
+
+@pytest.mark.lg_feature("wifi")
+def test_wifi_scan(ssh_command):
+    """
+    Test wifi scanning functionality.
+
+    This test performs a wifi scan and verifies that at least one network is found.
+    """
+    ssh_command.run("uci delete wireless.radio0.disabled")
+    ssh_command.run("uci commit")
+
+    restart_wifi_and_wait(ssh_command)
+
+    # Perform wifi scan
+    scan_output = ssh_command.run("iwinfo phy0 scan")
+    scan_results = "\n".join(scan_output[0])
+
+    # Check that at least one network was found
+    assert "ESSID:" in scan_results
+
+
 @pytest.mark.lg_feature("hwsim")
 def test_wifi_hwsim_sae_mixed(ssh_command):
     """
