merge dev into two_edges

tests/unit/test_helper_components.py

@@ -0,0 +1,81 @@
+from langchain_core.documents import Document
+
+from langflow.components import helpers
+from langflow.custom.utils import build_custom_component_template
+from langflow.schema import Data
+
+
+def test_update_data_component():
+    # Arrange
+    update_data_component = helpers.UpdateDataComponent()
+
+    # Act
+    new_data = {"new_key": "new_value"}
+    existing_data = Data(data={"existing_key": "existing_value"})
+    result = update_data_component.build(existing_data, new_data)
+    assert result.data == {"existing_key": "existing_value", "new_key": "new_value"}
+    assert result.existing_key == "existing_value"
+    assert result.new_key == "new_value"
+
+
+def test_document_to_data_component():
+    # Arrange
+    document_to_data_component = helpers.DocumentToDataComponent()
+
+    # Act
+    # Replace with your actual test data
+    document = Document(page_content="key: value", metadata={"url": "https://example.com"})
+    result = document_to_data_component.build(document)
+
+    # Assert
+    # Replace with your actual expected result
+    assert result == [Data(data={"text": "key: value", "url": "https://example.com"})]
+
+
+def test_uuid_generator_component():
+    # Arrange
+    uuid_generator_component = helpers.UUIDGeneratorComponent()
+    uuid_generator_component.code = open(helpers.IDGenerator.__file__, "r").read()
+
+    frontend_node, _ = build_custom_component_template(uuid_generator_component)
+
+    # Act
+    build_config = frontend_node.get("template")
+    field_name = "unique_id"
+    build_config = uuid_generator_component.update_build_config(build_config, None, field_name)
+    unique_id = build_config["unique_id"]["value"]
+    result = uuid_generator_component.build(unique_id)
+
+    # Assert
+    # UUID should be a string of length 36
+    assert isinstance(result, str)
+    assert len(result) == 36
+
+
+def test_data_as_text_component():
+    # Arrange
+    data_as_text_component = helpers.DataToTextComponent()
+
+    # Act
+    # Replace with your actual test data
+    data = [Data(data={"key": "value", "bacon": "eggs"})]
+    template = "Data:{data} -- Bacon:{bacon}"
+    result = data_as_text_component.build(data, template=template)
+
+    # Assert
+    # Replace with your actual expected result
+    assert result == "Data:{'key': 'value', 'bacon': 'eggs'} -- Bacon:eggs"
+
+
+def test_text_to_data_component():
+    # Arrange
+    text_to_data_component = helpers.CreateDataComponent()
+
+    # Act
+    # Replace with your actual test data
+    dict_with_text = {"field_1": {"key": "value"}}
+    result = text_to_data_component.build(number_of_fields=1, **dict_with_text)
+
+    # Assert
+    # Replace with your actual expected result
+    assert result == Data(data={"key": "value"})