Handle NULL return from os_zalloc() in sta_track_add()

[mech_eap.git] / tests / hwsim / test_p2p_wifi_display.py

diff --git a/tests/hwsim/test_p2p_wifi_display.py b/tests/hwsim/test_p2p_wifi_display.py
index 38eb16a..94c7da9 100644 (file)
--- a/tests/hwsim/test_p2p_wifi_display.py
+++ b/tests/hwsim/test_p2p_wifi_display.py
@@ -4,6 +4,7 @@
 # This software may be distributed under the terms of the BSD license.
 # See README for more details.
 
+from remotehost import remote_compatible
 import logging
 logger = logging.getLogger()
 import time
@@ -12,8 +13,7 @@ import Queue
 
 import hwsim_utils
 import utils
-from test_p2p_autogo import connect_cli
-from test_p2p_persistent import form, invite, invite_from_cli, invite_from_go
+from p2p_utils import *
 
 def test_wifi_display(dev):
     """Wi-Fi Display extensions to P2P"""
@@ -299,6 +299,7 @@ def test_wifi_display_persistent_group(dev):
         dev[1].request("SET wifi_display 0")
         dev[2].request("SET wifi_display 0")
 
+@remote_compatible
 def test_wifi_display_invalid_subelem(dev):
     """Wi-Fi Display and invalid subelement parsing"""
     addr1 = dev[1].p2p_dev_addr()
@@ -328,7 +329,7 @@ def test_wifi_display_parsing(dev):
         dev[1].request("VENDOR_ELEM_REMOVE 11 *")
         dev[0].request("SET wifi_display 0")
 
-def test_wifi_display_parsing(dev):
+def _test_wifi_display_parsing(dev):
     wfd_devinfo = "00411c440028"
     dev[0].request("SET wifi_display 1")
     dev[0].request("WFD_SUBELEM_SET 0 0006" + wfd_devinfo)