More changes to for the TCK

Author: kabir

server-common/src/main/java/io/a2a/server/tasks/InMemoryTaskStore.java

@@ -10,6 +10,9 @@
 
 import jakarta.enterprise.context.ApplicationScoped;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 import io.a2a.spec.Artifact;
 import io.a2a.spec.ListTasksParams;
 import io.a2a.spec.ListTasksResult;
@@ -19,10 +22,12 @@
 @ApplicationScoped
 public class InMemoryTaskStore implements TaskStore, TaskStateProvider {
 
+    private static final Logger LOGGER = LoggerFactory.getLogger(InMemoryTaskStore.class);
     private final ConcurrentMap<String, Task> tasks = new ConcurrentHashMap<>();
 
     @Override
     public void save(Task task) {
+        LOGGER.debug("=== InMemoryTaskStore.save() === taskId=" + task.getId() + ", contextId=" + task.getContextId());
         tasks.put(task.getId(), task);
     }
 
@@ -38,6 +43,13 @@ public void delete(String taskId) {
 
     @Override
     public ListTasksResult list(ListTasksParams params) {
+        // DEBUG: Log store contents for totalSize investigation
+        LOGGER.debug("=== InMemoryTaskStore.list() DEBUG ===");
+        LOGGER.debug("Total tasks in store: {}", tasks.size());
+        LOGGER.debug("Filter contextId: {}", params.contextId());
+        LOGGER.debug("Filter status: {}", params.status());
+        tasks.values().forEach(t -> LOGGER.debug("  Task: id={}, contextId={}", t.getId(), t.getContextId()));
+
         // Filter and sort tasks in a single stream pipeline
         List<Task> allFilteredTasks = tasks.values().stream()
                 .filter(task -> params.contextId() == null || params.contextId().equals(task.getContextId()))
@@ -57,6 +69,8 @@ public ListTasksResult list(ListTasksParams params) {
                 .toList();
 
         int totalSize = allFilteredTasks.size();
+        LOGGER.debug("Filtered tasks count: {}", totalSize);
+        LOGGER.debug("=== END DEBUG ===");
 
         // Apply pagination
         int pageSize = params.getEffectivePageSize();

spec-grpc/src/main/java/io/a2a/grpc/utils/ProtoUtils.java

@@ -10,6 +10,9 @@
 import java.util.Map;
 import java.util.stream.Collectors;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 import com.google.protobuf.ByteString;
 import com.google.protobuf.Struct;
 import com.google.protobuf.Value;
@@ -70,6 +73,7 @@
 public class ProtoUtils {
 
     public static class ToProto {
+        private static final Logger LOGGER = LoggerFactory.getLogger(ToProto.class);
 
         public static io.a2a.grpc.AgentCard agentCard(AgentCard agentCard) {
             io.a2a.grpc.AgentCard.Builder builder = io.a2a.grpc.AgentCard.newBuilder();
@@ -155,6 +159,11 @@ public static io.a2a.grpc.Task task(Task task) {
         }
 
         public static io.a2a.grpc.ListTasksResponse listTasksResult(io.a2a.spec.ListTasksResult result) {
+            LOGGER.debug("=== ProtoUtils.ToProto.listTasksResult() DEBUG ===");
+            LOGGER.debug("Input result.totalSize(): {}", result.totalSize());
+            LOGGER.debug("Input result.pageSize(): {}", result.pageSize());
+            LOGGER.debug("Input result.tasks().size(): {}", result.tasks().size());
+
             io.a2a.grpc.ListTasksResponse.Builder builder = io.a2a.grpc.ListTasksResponse.newBuilder();
             if (result.tasks() != null) {
                 builder.addAllTasks(result.tasks().stream().map(ToProto::task).collect(Collectors.toList()));
@@ -164,7 +173,14 @@ public static io.a2a.grpc.ListTasksResponse listTasksResult(io.a2a.spec.ListTask
             }
             builder.setTotalSize(result.totalSize());
             builder.setPageSize(result.pageSize());
-            return builder.build();
+
+            io.a2a.grpc.ListTasksResponse response = builder.build();
+            LOGGER.debug("Output response.getTotalSize(): {}", response.getTotalSize());
+            LOGGER.debug("Output response.getPageSize(): {}", response.getPageSize());
+            LOGGER.debug("Output response.getTasksCount(): {}", response.getTasksCount());
+            LOGGER.debug("=== END ProtoUtils DEBUG ===");
+
+            return response;
         }
 
         public static io.a2a.grpc.Message message(Message message) {

tck/src/main/resources/application.properties

@@ -12,6 +12,7 @@ a2a.executor.keep-alive-seconds=60
 quarkus.log.category."io.a2a.server.requesthandlers".level=DEBUG
 quarkus.log.category."io.a2a.server.events".level=DEBUG
 quarkus.log.category."io.a2a.server.tasks".level=DEBUG
+quarkus.log.category."io.a2a.grpc.utils".level=DEBUG
 
 # Log to file for analysis
 quarkus.log.file.enable=true

transport/rest/src/main/java/io/a2a/transport/rest/handler/RestHandler.java

@@ -300,7 +300,7 @@ private void validate(String json) {
 
     private HTTPRestResponse createSuccessResponse(int statusCode, com.google.protobuf.Message.Builder builder) {
         try {
-            String jsonBody = JsonFormat.printer().print(builder);
+            String jsonBody = JsonFormat.printer().includingDefaultValueFields().print(builder);
             return new HTTPRestResponse(statusCode, "application/json", jsonBody);
         } catch (InvalidProtocolBufferException e) {
             return createErrorResponse(new InternalError("Failed to serialize response: " + e.getMessage()));