Fix tests and add tests for append, last_chunk flags

Author: kthota-g

tests/server/tasks/test_task_updater.py

@@ -146,6 +146,37 @@ async def test_add_artifact_generates_id(
     assert isinstance(event, TaskArtifactUpdateEvent)
     assert event.artifact.artifactId == str(known_uuid)
     assert event.artifact.parts == sample_parts
+    assert event.append == None
+    assert event.lastChunk == None
+
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    "append_val, last_chunk_val",
+    [
+        (False, False),
+        (True, True),
+        (True, False),
+        (False, True),
+    ],
+)
+async def test_add_artifact_with_append_last_chunk(
+    task_updater, event_queue, sample_parts, append_val, last_chunk_val
+):
+    """Test add_artifact with append and last_chunk flags."""
+    await task_updater.add_artifact(
+        parts=sample_parts, artifact_id="id1", append=append_val, last_chunk=last_chunk_val
+    )
+
+    event_queue.enqueue_event.assert_called_once()
+    event = event_queue.enqueue_event.call_args[0][0]
+
+    assert isinstance(event, TaskArtifactUpdateEvent)
+    assert event.artifact.artifactId == "id1"
+    assert event.artifact.parts == sample_parts
+    assert event.append == append_val
+    assert event.lastChunk == last_chunk_val
 
 
 @pytest.mark.asyncio

tests/utils/test_proto_utils.py

@@ -237,7 +237,7 @@ def test_task_id_params_from_proto_invalid_name(self):
         assert isinstance(exc_info.value.error, types.InvalidParamsError)
 
     def test_task_push_config_from_proto_invalid_parent(self):
-        request = a2a_pb2.CreateTaskPushNotificationRequest(
+        request = a2a_pb2.CreateTaskPushNotificationConfigRequest(
             parent='invalid-parent'
         )
         with pytest.raises(ServerError) as exc_info: