jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add test for custom component with TemplateField

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-10 15:48:51 -03:00
commit d670d0fd93
3 changed files with 39 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test_custom_component.py
View file

@ -549,3 +549,17 @@ def test_build_langchain_template_custom_component_valid_code(test_component_cod
frontend_node = build_custom_component_template(component, update_field="param")
new_param_options = frontend_node["template"]["param"]["options"]
assert param_options != new_param_options
def test_build_langchain_template_custom_component_templatefield(test_component_with_templatefield_code):
component = create_and_validate_component(test_component_with_templatefield_code)
frontend_node = build_custom_component_template(component)
assert isinstance(frontend_node, dict)
template = frontend_node["template"]
assert isinstance(template, dict)
assert "param" in template
param_options = template["param"]["options"]
# Now run it again with an update field
frontend_node = build_custom_component_template(component, update_field="param")
new_param_options = frontend_node["template"]["param"]["options"]
assert param_options != new_param_options