tests: Use a helper function for waiting session termination

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

diff --git a/tests/hwsim/test_p2p_autogo.py b/tests/hwsim/test_p2p_autogo.py
index d62da59724e5e3d68cf41ae95db892aa4b41ec31..b10c05f109b515bffc48d9059cccd5933ed3e0d4 100644 (file)
--- a/tests/hwsim/test_p2p_autogo.py
+++ b/tests/hwsim/test_p2p_autogo.py
@@ -32,11 +32,7 @@ def test_autogo(dev):
     autogo(dev[0])
     connect_cli(dev[0], dev[1])
     dev[0].remove_group()
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=2)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()
 
 def test_autogo_2cli(dev):
     """P2P autonomous GO and two clients joining group"""
@@ -45,17 +41,9 @@ def test_autogo_2cli(dev):
     connect_cli(dev[0], dev[2])
     hwsim_utils.test_connectivity_p2p(dev[1], dev[2])
     dev[0].global_request("P2P_REMOVE_CLIENT " + dev[1].p2p_dev_addr())
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=2)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()
     dev[0].remove_group()
-    ev = dev[2].wait_event(["P2P-GROUP-REMOVED"], timeout=2)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[2].wait_go_ending_session()
 
 def test_autogo_tdls(dev):
     """P2P autonomous GO and two clients using TDLS"""

diff --git a/tests/hwsim/test_p2p_grpform.py b/tests/hwsim/test_p2p_grpform.py
index 3780eb0f67873560ae2da6a07030ae4691cac25b..1b276e40f5b9ccae7c7fc1ec7f02c5631561ea5e 100644 (file)
--- a/tests/hwsim/test_p2p_grpform.py
+++ b/tests/hwsim/test_p2p_grpform.py
@@ -177,16 +177,8 @@ def test_grpform_per_sta_psk(dev):
     hwsim_utils.test_connectivity_p2p(dev[1], dev[2])
 
     dev[0].remove_group()
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=2)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
-    ev = dev[2].wait_event(["P2P-GROUP-REMOVED"], timeout=2)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()
+    dev[2].wait_go_ending_session()
 
 def test_grpform_per_sta_psk_wps(dev):
     """P2P group formation with per-STA PSKs with non-P2P WPS STA"""
@@ -204,8 +196,4 @@ def test_grpform_per_sta_psk_wps(dev):
 
     dev[0].remove_group()
     dev[2].request("DISCONNECT")
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=2)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()

diff --git a/tests/hwsim/test_p2p_persistent.py b/tests/hwsim/test_p2p_persistent.py
index 67e096b7907e2925e0b7038f33edc422d711be33..d5ff95e5846932148a41ad3354bb7d9031eb3537 100644 (file)
--- a/tests/hwsim/test_p2p_persistent.py
+++ b/tests/hwsim/test_p2p_persistent.py
@@ -42,11 +42,7 @@ def test_persistent_group(dev):
 
     logger.info("Terminate persistent group")
     dev[0].remove_group()
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=3)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()
 
     logger.info("Re-invoke persistent group from client")
     dev[0].request("SET persistent_reconnect 1")
@@ -76,11 +72,7 @@ def test_persistent_group(dev):
 
     logger.info("Terminate persistent group")
     dev[0].remove_group()
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=3)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()
 
     logger.info("Re-invoke persistent group from GO")
     dev[1].request("SET persistent_reconnect 1")
@@ -110,11 +102,7 @@ def test_persistent_group(dev):
 
     logger.info("Terminate persistent group")
     dev[0].remove_group()
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=3)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()
 
 def test_persistent_group_per_sta_psk(dev):
     """P2P persistent group formation and re-invocation using per-client PSK"""
@@ -158,11 +146,7 @@ def test_persistent_group_per_sta_psk(dev):
 
     logger.info("Remove one of the clients from the group")
     dev[0].global_request("P2P_REMOVE_CLIENT " + addr2)
-    ev = dev[2].wait_event(["P2P-GROUP-REMOVED"], timeout=3)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[2].wait_go_ending_session()
 
     logger.info("Try to reconnect after having been removed from group")
     if not dev[2].discover_peer(addr0, social=True):
@@ -178,11 +162,7 @@ def test_persistent_group_per_sta_psk(dev):
 
     logger.info("Remove the remaining client from the group")
     dev[0].global_request("P2P_REMOVE_CLIENT " + addr1)
-    ev = dev[1].wait_event(["P2P-GROUP-REMOVED"], timeout=3)
-    if ev is None:
-        raise Exception("Group removal event timed out")
-    if "reason=GO_ENDING_SESSION" not in ev:
-        raise Exception("Unexpected group removal reason")
+    dev[1].wait_go_ending_session()
 
     logger.info("Terminate persistent group")
     dev[0].remove_group()

diff --git a/tests/hwsim/wpasupplicant.py b/tests/hwsim/wpasupplicant.py
index 0f376cfa33266a8f4ea0d2285062ab5ea145199a..6080b67e7004f44d97a7260bb533c3676b08190b 100644 (file)
--- a/tests/hwsim/wpasupplicant.py
+++ b/tests/hwsim/wpasupplicant.py
@@ -320,6 +320,13 @@ class WpaSupplicant:
                             return ev
         return None
 
+    def wait_go_ending_session(self):
+        ev = self.wait_event(["P2P-GROUP-REMOVED"], timeout=3)
+        if ev is None:
+            raise Exception("Group removal event timed out")
+        if "reason=GO_ENDING_SESSION" not in ev:
+            raise Exception("Unexpected group removal reason")
+
     def dump_monitor(self):
         while self.mon.pending():
             ev = self.mon.recv()