refactor: Add test for adding messages

tests/test_messages_endpoints.py

@@ -1,9 +1,12 @@
+import uuid
 from uuid import UUID
 
 import pytest
 from fastapi.testclient import TestClient
+from sqlmodel import select
 
-from langflow.memory import add_messagetables
+from langflow.memory import add_messages, add_messagetables
+from langflow.schema.message import Message
 
 # Assuming you have these imports available
 from langflow.services.database.models.message import MessageCreate, MessageRead, MessageUpdate
@@ -37,6 +40,21 @@ def created_messages(session):
         return messages_read
 
 
+def test_add_message(session, flow):
+    session_id = str(uuid.uuid4())
+    message = Message(text="Test message", sender="User", sender_name="User", session_id=session_id)
+    messages = add_messages([message], flow.id)
+
+    with session_scope() as session:
+        message = session.exec(select(MessageTable).where(MessageTable.session_id == session_id)).first()
+        assert message is not None
+        assert len(messages) == 1
+        assert message.text == "Test message"
+        assert message.sender == "User"
+        assert message.sender_name == "User"
+        assert message.session_id == session_id
+
+
 def test_delete_messages(client: TestClient, created_messages, logged_in_headers):
     response = client.request(
         "DELETE", "api/v1/monitor/messages", json=[str(msg.id) for msg in created_messages], headers=logged_in_headers