tests: Check SAE capability for couple of forgotten sigma_dut cases

Signed-off-by: Jouni Malinen <j@w1.fi>

diff --git a/tests/hwsim/test_sigma_dut.py b/tests/hwsim/test_sigma_dut.py
index c1b4c86d8ba17b51b131d6b90a004b2356224f98..5450c337e6a361068254266241fbd70620aa979b 100644 (file)
--- a/tests/hwsim/test_sigma_dut.py
+++ b/tests/hwsim/test_sigma_dut.py
@@ -4662,6 +4662,7 @@ def test_sigma_dut_ap_beacon_prot(dev, apdev, params):
 
 def test_sigma_dut_ap_transition_disable(dev, apdev, params):
     """sigma_dut controlled AP and transition disabled indication"""
+    check_sae_capab(dev[0])
     logdir = params['prefix'] + ".sigma-hostapd"
 
     with HWSimRadio() as (radio, iface):
@@ -4687,6 +4688,7 @@ def test_sigma_dut_ap_transition_disable(dev, apdev, params):
 
 def test_sigma_dut_ap_transition_disable_change(dev, apdev, params):
     """sigma_dut controlled AP and transition disabled indication change"""
+    check_sae_capab(dev[0])
     logdir = params['prefix'] + ".sigma-hostapd"
 
     with HWSimRadio() as (radio, iface):