From 49d8866dd394ca99d77d4f98ac48bafc1208ed1f Mon Sep 17 00:00:00 2001
From: Erik <erik@montnemery.com>
Date: Tue, 23 Jan 2024 15:53:17 +0100
Subject: [PATCH] Update tests after rebase

---
 tests/helpers/test_script.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/tests/helpers/test_script.py b/tests/helpers/test_script.py
index ea17eda05ee842..7677d55ca5a392 100644
--- a/tests/helpers/test_script.py
+++ b/tests/helpers/test_script.py
@@ -5549,7 +5549,12 @@ async def test_conversation_response_subscript_if(
     assert result.conversation_response == response
 
     expected_trace = {
-        "0": [{"result": {"conversation_response": "Testing 123"}}],
+        "0": [
+            {
+                "result": {"conversation_response": "Testing 123"},
+                "variables": {"var": var},
+            }
+        ],
         "1": [{"result": {"choice": choice}}],
         "1/if": [{"result": {"result": if_result}}],
         "1/if/condition/0": [{"result": {"result": var == 1, "entities": []}}],
@@ -5588,7 +5593,12 @@ async def test_conversation_response_not_set_subscript_if(
     assert result.conversation_response == "Testing 123"
 
     expected_trace = {
-        "0": [{"result": {"conversation_response": "Testing 123"}}],
+        "0": [
+            {
+                "result": {"conversation_response": "Testing 123"},
+                "variables": {"var": var},
+            }
+        ],
         "1": [{"result": {"choice": choice}}],
         "1/if": [{"result": {"result": if_result}}],
         "1/if/condition/0": [{"result": {"result": var == 1, "entities": []}}],