adding 12 tests for utils/message.py

Author: didier-durand

tests/utils/test_message.py

@@ -0,0 +1,192 @@
+from unittest.mock import patch
+import uuid
+
+from a2a.types import (
+    Message,
+    Part,
+    Role,
+    TextPart,
+)
+from a2a.utils import new_agent_text_message, get_text_parts, get_message_text
+
+
+class TestNewAgentTextMessage:
+    def test_new_agent_text_message_basic(self):
+        # Setup
+        text = "Hello, I'm an agent"
+
+        # Exercise - with a fixed uuid for testing
+        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+            message = new_agent_text_message(text)
+
+        # Verify
+        assert message.role == Role.agent
+        assert len(message.parts) == 1
+        assert message.parts[0].root.text == text
+        assert message.messageId == '12345678-1234-5678-1234-567812345678'
+        assert message.taskId is None
+        assert message.contextId is None
+
+    def test_new_agent_text_message_with_context_id(self):
+        # Setup
+        text = "Message with context"
+        context_id = "test-context-id"
+
+        # Exercise
+        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+            message = new_agent_text_message(text, context_id=context_id)
+
+        # Verify
+        assert message.role == Role.agent
+        assert message.parts[0].root.text == text
+        assert message.messageId == '12345678-1234-5678-1234-567812345678'
+        assert message.contextId == context_id
+        assert message.taskId is None
+
+    def test_new_agent_text_message_with_task_id(self):
+        # Setup
+        text = "Message with task id"
+        task_id = "test-task-id"
+
+        # Exercise
+        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+            message = new_agent_text_message(text, task_id=task_id)
+
+        # Verify
+        assert message.role == Role.agent
+        assert message.parts[0].root.text == text
+        assert message.messageId == '12345678-1234-5678-1234-567812345678'
+        assert message.taskId == task_id
+        assert message.contextId is None
+
+    def test_new_agent_text_message_with_both_ids(self):
+        # Setup
+        text = "Message with both ids"
+        context_id = "test-context-id"
+        task_id = "test-task-id"
+
+        # Exercise
+        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+            message = new_agent_text_message(text, context_id=context_id, task_id=task_id)
+
+        # Verify
+        assert message.role == Role.agent
+        assert message.parts[0].root.text == text
+        assert message.messageId == '12345678-1234-5678-1234-567812345678'
+        assert message.contextId == context_id
+        assert message.taskId == task_id
+
+    def test_new_agent_text_message_empty_text(self):
+        # Setup
+        text = ""
+
+        # Exercise
+        with patch('uuid.uuid4', return_value=uuid.UUID('12345678-1234-5678-1234-567812345678')):
+            message = new_agent_text_message(text)
+
+        # Verify
+        assert message.role == Role.agent
+        assert message.parts[0].root.text == ""
+        assert message.messageId == '12345678-1234-5678-1234-567812345678'
+
+
+class TestGetTextParts:
+    def test_get_text_parts_single_text_part(self):
+        # Setup
+        parts = [Part(root=TextPart(text="Hello world"))]
+
+        # Exercise
+        result = get_text_parts(parts)
+
+        # Verify
+        assert result == ["Hello world"]
+
+    def test_get_text_parts_multiple_text_parts(self):
+        # Setup
+        parts = [
+            Part(root=TextPart(text="First part")),
+            Part(root=TextPart(text="Second part")),
+            Part(root=TextPart(text="Third part")),
+        ]
+
+        # Exercise
+        result = get_text_parts(parts)
+
+        # Verify
+        assert result == ["First part", "Second part", "Third part"]
+
+    def test_get_text_parts_empty_list(self):
+        # Setup
+        parts = []
+
+        # Exercise
+        result = get_text_parts(parts)
+
+        # Verify
+        assert result == []
+
+
+class TestGetMessageText:
+    def test_get_message_text_single_part(self):
+        # Setup
+        message = Message(
+            role=Role.agent,
+            parts=[Part(root=TextPart(text="Hello world"))],
+            messageId="test-message-id",
+        )
+
+        # Exercise
+        result = get_message_text(message)
+
+        # Verify
+        assert result == "Hello world"
+
+    def test_get_message_text_multiple_parts(self):
+        # Setup
+        message = Message(
+            role=Role.agent,
+            parts=[
+                Part(root=TextPart(text="First line")),
+                Part(root=TextPart(text="Second line")),
+                Part(root=TextPart(text="Third line")),
+            ],
+            messageId="test-message-id",
+        )
+
+        # Exercise
+        result = get_message_text(message)
+
+        # Verify - default delimiter is newline
+        assert result == "First line\nSecond line\nThird line"
+
+    def test_get_message_text_custom_delimiter(self):
+        # Setup
+        message = Message(
+            role=Role.agent,
+            parts=[
+                Part(root=TextPart(text="First part")),
+                Part(root=TextPart(text="Second part")),
+                Part(root=TextPart(text="Third part")),
+            ],
+            messageId="test-message-id",
+        )
+
+        # Exercise
+        result = get_message_text(message, delimiter=" | ")
+
+        # Verify
+        assert result == "First part | Second part | Third part"
+
+    def test_get_message_text_empty_parts(self):
+        # Setup
+        message = Message(
+            role=Role.agent,
+            parts=[],
+            messageId="test-message-id",
+        )
+
+        # Exercise
+        result = get_message_text(message)
+
+        # Verify
+        assert result == ""