From 54a2cbf7244387ea2599e9965a5d6a0e2301f21d Mon Sep 17 00:00:00 2001 From: italojohnny Date: Fri, 14 Feb 2025 11:59:20 -0300 Subject: [PATCH 1/2] test: add test for MessageStoreComponent component --- .../helpers/test_store_message_component.py | 56 +++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 src/backend/tests/unit/components/helpers/test_store_message_component.py diff --git a/src/backend/tests/unit/components/helpers/test_store_message_component.py b/src/backend/tests/unit/components/helpers/test_store_message_component.py new file mode 100644 index 000000000000..2006e25fb91c --- /dev/null +++ b/src/backend/tests/unit/components/helpers/test_store_message_component.py @@ -0,0 +1,56 @@ +import pytest + +from langflow.components.helpers import MessageStoreComponent +from tests.base import ComponentTestBaseWithClient + + +@pytest.mark.usefixtures("client") +class TestMessageStoreComponent(ComponentTestBaseWithClient): + @pytest.fixture + def component_class(self): + return MessageStoreComponent + + @pytest.fixture + def default_kwargs(self): + return { + "message": "Hello, World!", + "sender": "User", + "sender_name": "Alice", + "session_id": "session_123", + "memory": None, + } + + @pytest.fixture + def file_names_mapping(self): + return [ + {"version": "1.0.0", "module": "messages", "file_name": "MessageStore"}, + {"version": "1.1.0", "module": "messages", "file_name": "MessageStore"}, + ] + + async def test_store_message_with_memory(self, component_class, default_kwargs): + component = component_class(**default_kwargs) + component.memory = Mock() # Mocking the memory input + result = await component.store_message() + assert result is not None + assert result.text == "Hello, World!" + assert result.sender == "User" + assert result.sender_name == "Alice" + assert result.session_id == "session_123" + + async def test_store_message_without_memory(self, component_class, default_kwargs): + component = component_class(**default_kwargs) + component.memory = None # No external memory + result = await component.store_message() + assert result is not None + assert result.text == "Hello, World!" + assert result.sender == "User" + assert result.sender_name == "Alice" + assert result.session_id == "session_123" + + async def test_store_message_no_messages_stored(self, component_class, default_kwargs): + component = component_class(**default_kwargs) + component.memory = None # No external memory + with pytest.raises( + ValueError, match="No messages were stored. Please ensure that the session ID and sender are properly set." + ): + await component.store_message() From dff77c66b068ec22108694789b27cd260e9a3fe4 Mon Sep 17 00:00:00 2001 From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com> Date: Fri, 14 Feb 2025 15:01:15 +0000 Subject: [PATCH 2/2] [autofix.ci] apply automated fixes --- .../unit/components/helpers/test_store_message_component.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/tests/unit/components/helpers/test_store_message_component.py b/src/backend/tests/unit/components/helpers/test_store_message_component.py index 2006e25fb91c..cff4be517675 100644 --- a/src/backend/tests/unit/components/helpers/test_store_message_component.py +++ b/src/backend/tests/unit/components/helpers/test_store_message_component.py @@ -1,6 +1,6 @@ import pytest - from langflow.components.helpers import MessageStoreComponent + from tests.base import ComponentTestBaseWithClient