#863 verify_response_with fix (#864)

taverntesting · Nov 18, 2023 · 7464cab · 7464cab
1 parent 722b40b
commit 7464cab
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/example/mqtt/testing_utils.py b/example/mqtt/testing_utils.py
@@ -1,6 +1,6 @@
 def message_says_hello(msg):
     """Make sure that the response was friendly"""
-    assert msg.payload.get("message") == "hello world"
+    assert msg.msg.payload.get("message") == "hello world"
 
 
 def return_hello(_=None):

diff --git a/tavern/response.py b/tavern/response.py
@@ -131,7 +131,11 @@ def check_ext_functions(verify_block):
                     "Badly formatted 'verify_response_with' block"
                 )
 
-        check_ext_functions(response_block.get("verify_response_with", None))
+        if mqtt_responses := response_block.get("mqtt_responses"):
+            for mqtt_response in mqtt_responses:
+                check_ext_functions(mqtt_response.get("verify_response_with", None))
+        else:
+            check_ext_functions(response_block.get("verify_response_with", None))
 
         def check_deprecated_validate(name):
             nfuncs = len(self.validate_functions)