tests: bgscan_simple and BTM query

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

diff --git a/tests/hwsim/test_bgscan.py b/tests/hwsim/test_bgscan.py
index a3daa06ca9be38c4cb16aa63ace6cfdb2e2f656d..3eb6423f9c178218d6f24bc632ca664e88b64606 100644 (file)
--- a/tests/hwsim/test_bgscan.py
+++ b/tests/hwsim/test_bgscan.py
@@ -129,6 +129,37 @@ def test_bgscan_simple_same_scan_int(dev, apdev):
         if ev is None:
             raise Exception("Scanning not continued")
 
+def test_bgscan_simple_btm_query(dev, apdev):
+    """bgscan_simple and BTM query"""
+    params = {"ssid": "bgscan",
+              "bss_transition": "1"}
+    hapd = hostapd.add_ap(apdev[0], params)
+
+    dev[0].connect("bgscan", key_mgmt="NONE", scan_freq="2412",
+                   bgscan="simple:1:-20:1:2")
+    ev = dev[0].wait_event(["WNM: BSS Transition Management Request"], 10)
+    if ev is None:
+        raise Exception("BSS TM not used")
+    for i in range(2):
+        ev = dev[0].wait_event(["CTRL-EVENT-SCAN-RESULTS"], 10)
+        if ev is None:
+            raise Exception("Scanning not continued")
+
+def test_bgscan_simple_btm_query_no_ap_support(dev, apdev):
+    """bgscan_simple and BTM query and no AP support"""
+    params = {"ssid": "bgscan",
+              "bss_transition": "0"}
+    hapd = hostapd.add_ap(apdev[0], params)
+
+    dev[0].connect("bgscan", key_mgmt="NONE", scan_freq="2412",
+                   bgscan="simple:1:-20:1:2")
+    ev = dev[0].wait_event(["WNM: BSS Transition Management Request",
+                            "CTRL-EVENT-SCAN-RESULTS"], 10)
+    if ev is None:
+        raise Exception("No background scan seen")
+    if "WNM: BSS" in ev:
+        raise Exception("Unexpected use of BSS TM")
+
 def test_bgscan_simple_oom(dev, apdev):
     """bgscan_simple OOM"""
     hapd = hostapd.add_ap(apdev[0], {"ssid": "bgscan"})