tests: Update dpp_proto_after_wrapped_data_pkex_cr_req for PKEXv2

Check for both PKEX v1 and v2 variants of the message.

Signed-off-by: Jouni Malinen <quic_jouni@quicinc.com>

diff --git a/tests/hwsim/test_dpp.py b/tests/hwsim/test_dpp.py
index 339d7297f7d74451d05f1ae742b3ecf0e1b969ba..9a82263d903469131d86b7fd248c33f14662c986 100644 (file)
--- a/tests/hwsim/test_dpp.py
+++ b/tests/hwsim/test_dpp.py
@@ -3797,7 +3797,7 @@ def test_dpp_proto_after_wrapped_data_pkex_cr_req(dev, apdev):
     """DPP protocol testing - attribute after Wrapped Data in PKEX CR Req"""
     run_dpp_proto_init_pkex(dev, 1, 4)
     ev = dev[0].wait_event(["DPP-RX"], timeout=5)
-    if ev is None or "type=7" not in ev:
+    if ev is None or ("type=7" not in ev and "type=18" not in ev):
         raise Exception("PKEX Exchange Request not seen")
     ev = dev[0].wait_event(["DPP-RX"], timeout=5)
     if ev is None or "type=9" not in ev: