Contribute component tests
This guide outlines how to structure and implement tests for application components to ensure consistency and adequate coverage.
File naming
-
The test file should follow the same directory structure as the component being tested, but should be placed in the corresponding unit tests folder.
For example, if the file path for the component is
src/backend/base/langflow/components/prompts/
, then the test file should be located atsrc/backend/tests/unit/components/prompts
. -
The test file name should use snake case and follow the pattern
test_<file_name>.py
.For example, if the file to be tested is
PromptComponent.py
, then the test file should be namedtest_prompt_component.py
.
File structure
- Each test file should group tests into classes by component. There should be no standalone test functions in the file— only test methods within classes.
- Class names should follow the pattern
Test<ClassName>
. For example, if the component being tested isPromptComponent
, then the test class should be namedTestPromptComponent
.
Imports, inheritance, and mandatory methods
To standardize component tests, base test classes have been created and should be imported and inherited by all component test classes. These base classes are located in the file src/backend/tests/unit/base.py
.
To import the base test classes:
_10from tests.base import ComponentTestBaseWithClient_10from tests.base import ComponentTestBaseWithoutClient
These base classes enforce mandatory methods that the component test classes must implement. The base classes ensure that components built in previous versions continues to work in the current version. By inheriting from one of these base classes, the developer must define the following methods decorated with @pytest.fixture
:
-
component_class:
Returns the class of the component to be tested. For example:_10@pytest.fixture_10def component_class(self):_10return PromptComponent -
default_kwargs:
Returns a dictionary with the default arguments required to instantiate the component. For example:_10@pytest.fixture_10def default_kwargs(self):_10return {"template": "Hello {name}!", "name": "John", "_session_id": "123"} -
file_names_mapping:
Returns a list of dictionaries representing the relationship betweenversion
,module
, andfile_name
that the tested component has had over time. This can be left empty if it is an unreleased component. For example:_10@pytest.fixture_10def file_names_mapping(self):_10return [_10{"version": "1.0.15", "module": "prompts", "file_name": "Prompt"},_10{"version": "1.0.16", "module": "prompts", "file_name": "Prompt"},_10{"version": "1.0.17", "module": "prompts", "file_name": "Prompt"},_10{"version": "1.0.18", "module": "prompts", "file_name": "Prompt"},_10{"version": "1.0.19", "module": "prompts", "file_name": "Prompt"},_10]
Testing component functionalities
Once the basic structure of the test file is defined, implement test methods for the component's functionalities. The following guidelines must be followed:
- Test method names should be descriptive, use snake case, and follow the pattern
test_<case_name>
. - Each test should follow the Arrange, Act, Assert pattern:
- Arrange: Prepare the data.
- Act: Execute the component.
- Assert: Verify the result.
Example
-
Arrange: Prepare the data.
It is recommended, but not mandatory, that you use the fixtures defined in the basic structure.
_10def test_post_code_processing(self, component_class, default_kwargs):_10component = component_class(**default_kwargs) -
Act: Execute the component.
Call the
.to_frontend_node()
method of the component prepared during the Arrange step._10def test_post_code_processing(self, component_class, default_kwargs):_10component = component_class(**default_kwargs)_10_10frontend_node = component.to_frontend_node() -
Assert: Verify the result.
After executing the
.to_frontend_node()
method, the resulting data is available for verification in the dictionaryfrontend_node["data"]["node"]
. Assertions should be clear and cover the expected outcomes._10def test_post_code_processing(self, component_class, default_kwargs):_10component = component_class(**default_kwargs)_10_10frontend_node = component.to_frontend_node()_10_10node_data = frontend_node["data"]["node"]_10assert node_data["template"]["template"]["value"] == "Hello {name}!"_10assert "name" in node_data["custom_fields"]["template"]_10assert "name" in node_data["template"]_10assert node_data["template"]["name"]["value"] == "John"