Add assertion to verify state_variable_id matches state_variables

devin-ai-integration[bot] · vincent0426 · devin-ai-integration[bot] · commit 01967b6621ba · 2026-01-07T05:07:07.000Z
Co-Authored-By: vincent@vellum.ai &lt;0426vincent@gmail.com&gt;
diff --git a/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_chat_message_trigger_custom_state_serialization.py b/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_chat_message_trigger_custom_state_serialization.py
@@ -62,3 +62,8 @@ def test_custom_state_workflow_serialization():
             },
         }
     ]
+
+    # AND the state_variable_id should match the corresponding state variable
+    state_variable_id = serialized_workflow["triggers"][0]["exec_config"]["state"]["state_variable_id"]
+    state_variable_ids = [sv["id"] for sv in serialized_workflow["state_variables"]]
+    assert state_variable_id in state_variable_ids

Original file line number	Diff line number	Diff line change
`@@ -62,3 +62,8 @@ def test_custom_state_workflow_serialization():`
`62`	`62`	`},`
`63`	`63`	`}`
`64`	`64`	`]`
	`65`	`+`
	`66`	`+ # AND the state_variable_id should match the corresponding state variable`
	`67`	`+ state_variable_id = serialized_workflow["triggers"][0]["exec_config"]["state"]["state_variable_id"]`
	`68`	`+ state_variable_ids = [sv["id"] for sv in serialized_workflow["state_variables"]]`
	`69`	`+ assert state_variable_id in state_variable_ids`