dify

dify/api/tests/unit_tests/models/test_conversation_variable.py

@@ -0,0 +1,26 @@
+from uuid import uuid4
+
+from core.variables import SegmentType
+from factories import variable_factory
+from models import ConversationVariable
+
+
+def test_from_variable_and_to_variable():
+    variable = variable_factory.build_conversation_variable_from_mapping(
+        {
+            "id": str(uuid4()),
+            "name": "name",
+            "value_type": SegmentType.OBJECT,
+            "value": {
+                "key": {
+                    "key": "value",
+                }
+            },
+        }
+    )
+
+    conversation_variable = ConversationVariable.from_variable(
+        app_id="app_id", conversation_id="conversation_id", variable=variable
+    )
+
+    assert conversation_variable.to_variable() == variable