Gemini code review

Author: kabir

reference-impl/src/test/java/io/a2a/server/apps/quarkus/A2ATestRoutes.java

@@ -141,7 +141,7 @@ public void getStreamingSubscribedCount(RoutingContext rc) {
 
     private void errorResponse(Throwable t, RoutingContext rc) {
         rc.response()
-                .setStatusCode(200)
+                .setStatusCode(500)
                 .putHeader(CONTENT_TYPE, TEXT_PLAIN)
                 .end();
     }

tests/server-common/src/test/java/io/a2a/server/apps/common/AbstractA2AServerTest.java

@@ -122,7 +122,7 @@ public void testTaskStoreMethodsSanityTest() throws Exception {
         Task saved = getTaskFromTaskStore(task.getId());
         assertEquals(task.getId(), saved.getId());
         assertEquals(task.getContextId(), saved.getContextId());
-        assertEquals(task.getStatus(), saved.getStatus());
+        assertEquals(task.getStatus().state(), saved.getStatus().state());
 
         deleteTaskInTaskStore(task.getId());
         Task saved2 = getTaskFromTaskStore(task.getId());
@@ -160,7 +160,6 @@ private void testGetTask(String mediaType) throws Exception {
             assertEquals("session-xyz", response.getResult().getContextId());
             assertEquals(TaskState.SUBMITTED, response.getResult().getStatus().state());
             assertNull(response.getError());
-        } catch (Exception e) {
         } finally {
             deleteTaskInTaskStore(MINIMAL_TASK.getId());
         }
@@ -971,6 +970,7 @@ protected void enqueueEventOnServer(Event event) throws Exception {
                 .build();
         HttpRequest request = HttpRequest.newBuilder()
                 .uri(URI.create("http://localhost:" + serverPort + "/" + path))
+                .header("Content-Type", APPLICATION_JSON)
                 .POST(HttpRequest.BodyPublishers.ofString(Utils.OBJECT_MAPPER.writeValueAsString(event)))
                 .build();
 
@@ -986,15 +986,22 @@ private CompletableFuture<Void> awaitStreamingSubscription() {
 
         return CompletableFuture.runAsync(() -> {
             try {
-                while (true) {
+                boolean done = false;
+                long end = System.currentTimeMillis() + 15000;
+                while (System.currentTimeMillis() > end) {
                     int count = getStreamingSubscribedCount();
                     if (count > initialCount.get()) {
+                        done = true;
                         break;
                     }
                     Thread.sleep(500);
                 }
+                if (!done) {
+                    throw new RuntimeException("Timed out waiting for subscription");
+                }
             } catch (InterruptedException e) {
                 Thread.currentThread().interrupt();
+                throw new RuntimeException("Interrupted");
             }
         });
     }