From: Jouni Malinen Date: Thu, 29 Aug 2024 13:00:10 +0000 (+0300) Subject: tests: More robust error handling for wifi_display_parsing X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=3935d78e7b40a02a3665ac75c62e9b73ecb72469;p=thirdparty%2Fhostap.git tests: More robust error handling for wifi_display_parsing Check explicitly that wfd_subelems is present in the dict instead of failing on a key error when trying to fetch it. Signed-off-by: Jouni Malinen --- diff --git a/tests/hwsim/test_p2p_wifi_display.py b/tests/hwsim/test_p2p_wifi_display.py index 29110bca7..c235181a6 100644 --- a/tests/hwsim/test_p2p_wifi_display.py +++ b/tests/hwsim/test_p2p_wifi_display.py @@ -430,6 +430,8 @@ def _test_wifi_display_parsing(dev): bssid = dev[0].get_group_status_field('bssid') dev[2].scan_for_bss(bssid, freq=2412, force_scan=True) bss = dev[2].get_bss(bssid) + if 'wfd_subelems' not in bss: + raise Exception("Missing WFD elements in scan results") if bss['wfd_subelems'] != "000006" + wfd_devinfo: raise Exception("Unexpected WFD elements in scan results: " + bss['wfd_subelems'])