refactor(api/core/app/segments): Support more kinds of Segments. (#6706)

Signed-off-by: -LAN- <laipz8200@outlook.com>
2024-07-26 15:03:56 +08:00 · 2024-07-26 15:03:56 +08:00 · c6996a48a4
commit c6996a48a4
parent 6b50bb0fe6
11 changed files with 147 additions and 96 deletions
--- a/api/tests/unit_tests/core/app/test_segment.py
+++ b/api/tests/unit_tests/core/app/test_segment.py
--- a/api/tests/unit_tests/core/app/test_variables.py
+++ b/api/tests/unit_tests/core/app/test_variables.py
@ -5,7 +5,8 @@ from core.app.segments import (
    ArrayVariable,
    FloatVariable,
    IntegerVariable,
-    NoneVariable,
+    NoneSegment,
+    ObjectSegment,
    ObjectVariable,
    SecretVariable,
    SegmentType,
@ -139,10 +140,10 @@ def test_variable_to_object():


 def test_build_a_object_variable_with_none_value():
-    var = factory.build_anonymous_variable(
+    var = factory.build_segment(
        {
            'key1': None,
        }
    )
-    assert isinstance(var, ObjectVariable)
-    assert isinstance(var.value['key1'], NoneVariable)
+    assert isinstance(var, ObjectSegment)
+    assert isinstance(var.value['key1'], NoneSegment)