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 <quic_jouni@quicinc.com>
This commit is contained in:
Jouni Malinen 2024-08-29 16:00:10 +03:00 committed by Jouni Malinen
parent 2b7277d3f0
commit 3935d78e7b

View file

@ -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'])