added tests

Author: martimfasantos

tests/utils/test_artifact.py

@@ -3,8 +3,14 @@
 
 from unittest.mock import patch
 
-from a2a.types import DataPart, Part, TextPart
+from a2a.types import (
+    Artifact,
+    DataPart,
+    Part,
+    TextPart,
+)
 from a2a.utils.artifact import (
+    get_artifact_text,
     new_artifact,
     new_data_artifact,
     new_text_artifact,
@@ -83,5 +89,70 @@ def test_new_data_artifact_assigns_name_description(self):
         self.assertEqual(artifact.description, description)
 
 
+class TestGetArtifactText(unittest.TestCase):
+    def test_get_artifact_text_single_part(self):
+        # Setup
+        artifact = Artifact(
+            name="test-artifact",
+            parts=[Part(root=TextPart(text='Hello world'))],
+            artifact_id='test-artifact-id',
+        )
+
+        # Exercise
+        result = get_artifact_text(artifact)
+
+        # Verify
+        assert result == 'Hello world'
+
+    def test_get_artifact_text_multiple_parts(self):
+        # Setup
+        artifact = Artifact(
+            name="test-artifact",
+            parts=[
+                Part(root=TextPart(text='First line')),
+                Part(root=TextPart(text='Second line')),
+                Part(root=TextPart(text='Third line')),
+            ],
+            artifact_id='test-artifact-id',
+        )
+
+        # Exercise
+        result = get_artifact_text(artifact)
+
+        # Verify - default delimiter is newline
+        assert result == 'First line\nSecond line\nThird line'
+
+    def test_get_artifact_text_custom_delimiter(self):
+        # Setup
+        artifact = Artifact(
+            name="test-artifact",
+            parts=[
+                Part(root=TextPart(text='First part')),
+                Part(root=TextPart(text='Second part')),
+                Part(root=TextPart(text='Third part')),
+            ],
+            artifact_id='test-artifact-id',
+        )
+
+        # Exercise
+        result = get_artifact_text(artifact, delimiter=' | ')
+
+        # Verify
+        assert result == 'First part | Second part | Third part'
+
+    def test_get_artifact_text_empty_parts(self):
+        # Setup
+        artifact = Artifact(
+            name="test-artifact",
+            parts=[],
+            artifact_id='test-artifact-id',
+        )
+
+        # Exercise
+        result = get_artifact_text(artifact)
+
+        # Verify
+        assert result == ''
+
 if __name__ == '__main__':
     unittest.main()

tests/utils/test_message.py

@@ -4,19 +4,13 @@
 
 from a2a.types import (
     DataPart,
-    FilePart,
-    FileWithBytes,
-    FileWithUri,
     Message,
     Part,
     Role,
     TextPart,
 )
 from a2a.utils.message import (
-    get_data_parts,
-    get_file_parts,
     get_message_text,
-    get_text_parts,
     new_agent_parts_message,
     new_agent_text_message,
 )
@@ -147,177 +141,6 @@ def test_new_agent_parts_message(self):
         assert message.message_id == 'abcdefab-cdef-abcd-efab-cdefabcdefab'
 
 
-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 TestGetDataParts:
-    def test_get_data_parts_single_data_part(self):
-        # Setup
-        parts = [Part(root=DataPart(data={'key': 'value'}))]
-
-        # Exercise
-        result = get_data_parts(parts)
-
-        # Verify
-        assert result == [{'key': 'value'}]
-
-    def test_get_data_parts_multiple_data_parts(self):
-        # Setup
-        parts = [
-            Part(root=DataPart(data={'key1': 'value1'})),
-            Part(root=DataPart(data={'key2': 'value2'})),
-        ]
-
-        # Exercise
-        result = get_data_parts(parts)
-
-        # Verify
-        assert result == [{'key1': 'value1'}, {'key2': 'value2'}]
-
-    def test_get_data_parts_mixed_parts(self):
-        # Setup
-        parts = [
-            Part(root=TextPart(text='some text')),
-            Part(root=DataPart(data={'key1': 'value1'})),
-            Part(root=DataPart(data={'key2': 'value2'})),
-        ]
-
-        # Exercise
-        result = get_data_parts(parts)
-
-        # Verify
-        assert result == [{'key1': 'value1'}, {'key2': 'value2'}]
-
-    def test_get_data_parts_no_data_parts(self):
-        # Setup
-        parts = [
-            Part(root=TextPart(text='some text')),
-        ]
-
-        # Exercise
-        result = get_data_parts(parts)
-
-        # Verify
-        assert result == []
-
-    def test_get_data_parts_empty_list(self):
-        # Setup
-        parts = []
-
-        # Exercise
-        result = get_data_parts(parts)
-
-        # Verify
-        assert result == []
-
-
-class TestGetFileParts:
-    def test_get_file_parts_single_file_part(self):
-        # Setup
-        file_with_uri = FileWithUri(
-            uri='file://path/to/file', mimeType='text/plain'
-        )
-        parts = [Part(root=FilePart(file=file_with_uri))]
-
-        # Exercise
-        result = get_file_parts(parts)
-
-        # Verify
-        assert result == [file_with_uri]
-
-    def test_get_file_parts_multiple_file_parts(self):
-        # Setup
-        file_with_uri1 = FileWithUri(
-            uri='file://path/to/file1', mime_type='text/plain'
-        )
-        file_with_bytes = FileWithBytes(
-            bytes='ZmlsZSBjb250ZW50',
-            mime_type='application/octet-stream',  # 'file content'
-        )
-        parts = [
-            Part(root=FilePart(file=file_with_uri1)),
-            Part(root=FilePart(file=file_with_bytes)),
-        ]
-
-        # Exercise
-        result = get_file_parts(parts)
-
-        # Verify
-        assert result == [file_with_uri1, file_with_bytes]
-
-    def test_get_file_parts_mixed_parts(self):
-        # Setup
-        file_with_uri = FileWithUri(
-            uri='file://path/to/file', mime_type='text/plain'
-        )
-        parts = [
-            Part(root=TextPart(text='some text')),
-            Part(root=FilePart(file=file_with_uri)),
-        ]
-
-        # Exercise
-        result = get_file_parts(parts)
-
-        # Verify
-        assert result == [file_with_uri]
-
-    def test_get_file_parts_no_file_parts(self):
-        # Setup
-        parts = [
-            Part(root=TextPart(text='some text')),
-            Part(root=DataPart(data={'key': 'value'})),
-        ]
-
-        # Exercise
-        result = get_file_parts(parts)
-
-        # Verify
-        assert result == []
-
-    def test_get_file_parts_empty_list(self):
-        # Setup
-        parts = []
-
-        # Exercise
-        result = get_file_parts(parts)
-
-        # Verify
-        assert result == []
-
-
 class TestGetMessageText:
     def test_get_message_text_single_part(self):
         # Setup