diff --git a/tests/hwsim/test_p2p_discovery.py b/tests/hwsim/test_p2p_discovery.py index c159aad24..803f830b9 100644 --- a/tests/hwsim/test_p2p_discovery.py +++ b/tests/hwsim/test_p2p_discovery.py @@ -100,8 +100,10 @@ def test_discovery_group_client(dev): logger.info("Client connected") hwsim_utils.test_connectivity_p2p(dev[0], dev[1]) logger.info("Try to discover a P2P client in a group") - if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False): - raise Exception("Could not discover group client") + if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False, timeout=10): + if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False, timeout=10): + if not dev[2].discover_peer(dev[1].p2p_dev_addr(), social=False, timeout=10): + raise Exception("Could not discover group client") # This is not really perfect, but something to get a bit more testing # coverage.. For proper discoverability mechanism validation, the P2P diff --git a/tests/hwsim/wpasupplicant.py b/tests/hwsim/wpasupplicant.py index a1a32fb5f..fb77e9b79 100644 --- a/tests/hwsim/wpasupplicant.py +++ b/tests/hwsim/wpasupplicant.py @@ -613,7 +613,8 @@ class WpaSupplicant: def p2p_connect_group(self, go_addr, pin, timeout=0, social=False): self.dump_monitor() if not self.discover_peer(go_addr, social=social): - raise Exception("GO " + go_addr + " not found") + if social or not self.discover_peer(go_addr, social=social): + raise Exception("GO " + go_addr + " not found") self.dump_monitor() cmd = "P2P_CONNECT " + go_addr + " " + pin + " join" if "OK" in self.global_request(cmd):