Add helper component tests

tests/test_helper_components.py

@@ -0,0 +1,99 @@
+from langchain_core.documents import Document
+
+from langflow.components import helpers
+from langflow.interface.custom.utils import build_custom_component_template
+from langflow.schema import Record
+
+
+def test_update_record_component():
+    # Arrange
+    update_record_component = helpers.UpdateRecordComponent()
+
+    # Act
+    new_data = {"new_key": "new_value"}
+    existing_record = Record(data={"existing_key": "existing_value"})
+    result = update_record_component.build(existing_record, 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_record_component():
+    # Arrange
+    document_to_record_component = helpers.DocumentToRecordComponent()
+
+    # Act
+    # Replace with your actual test data
+    document = Document(
+        page_content="key: value", metadata={"url": "https://example.com"}
+    )
+    result = document_to_record_component.build(document)
+
+    # Assert
+    # Replace with your actual expected result
+    assert result == [Record(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_python_function_component():
+    # Arrange
+    python_function_component = helpers.PythonFunctionComponent()
+
+    # Act
+    # function must be a string representation
+    function = "def function():\n    return 'Hello, World!'"
+    # result is the callable function
+    result = python_function_component.build(function)
+
+    # Assert
+    assert result() == "Hello, World!"
+
+
+def test_records_as_text_component():
+    # Arrange
+    records_as_text_component = helpers.RecordsAsTextComponent()
+
+    # Act
+    # Replace with your actual test data
+    records = [Record(data={"key": "value", "bacon": "eggs"})]
+    template = "Data:{data} -- Bacon:{bacon}"
+    result = records_as_text_component.build(records, template=template)
+
+    # Assert
+    # Replace with your actual expected result
+    assert result == "Data:{'key': 'value', 'bacon': 'eggs'} -- Bacon:eggs"
+
+
+def test_text_to_record_component():
+    # Arrange
+    text_to_record_component = helpers.TextToRecordComponent()
+
+    # Act
+    # Replace with your actual test data
+    dict_with_text = {"key": "value"}
+    result = text_to_record_component.build(dict_with_text)
+
+    # Assert
+    # Replace with your actual expected result
+    assert result == Record(data={"key": "value"})