jprdonnelly/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-30 16:00:32 -04:00
Code Issues Packages Projects Releases Wiki Activity

test: add tests for template transform

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN-
2026-03-20 17:30:36 +08:00
parent 98815b55d0
commit cc0a7a19f3
1 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
api/tests/unit_tests/core/workflow/nodes/template_transform/template_transform_node_spec.py
View File

@@ -348,6 +348,37 @@ class TestTemplateTransformNode:
assert mapping == {"node_123.var1": ["sys", "input1"]}
def test_extract_variable_selector_to_variable_mapping_returns_empty_mapping_without_variables(self):
node_data = {
"title": "Test",
"template": "{{ missing }}",
}
mapping = TemplateTransformNode._extract_variable_selector_to_variable_mapping(
graph_config={}, node_id="node_123", node_data=node_data
)
assert mapping == {}
def test_extract_variable_selector_to_variable_mapping_accepts_sequence_value_selectors(self):
node_data = {
"title": "Test",
"variables": [
{"variable": "var1", "value_selector": ("sys", "input1")},
{"variable": "empty_selector", "value_selector": ()},
],
"template": "{{ var1 }}",
}
mapping = TemplateTransformNode._extract_variable_selector_to_variable_mapping(
graph_config={}, node_id="node_123", node_data=node_data
)
assert mapping == {
"node_123.var1": ["sys", "input1"],
"node_123.empty_selector": [],
}
def test_extract_variable_selector_to_variable_mapping_ignores_invalid_entries(self):
node_data = {
"title": "Test",