11package io .a2a .examples .cloud ;
22
3+ import java .io .IOException ;
4+ import java .util .Collections ;
5+ import java .util .HashMap ;
6+ import java .util .List ;
7+ import java .util .Map ;
8+ import java .util .concurrent .CountDownLatch ;
9+ import java .util .concurrent .TimeUnit ;
10+ import java .util .concurrent .atomic .AtomicBoolean ;
11+ import java .util .concurrent .atomic .AtomicInteger ;
12+
313import io .a2a .A2A ;
414import io .a2a .client .Client ;
515import io .a2a .client .ClientEvent ;
919import io .a2a .client .config .ClientConfig ;
1020import io .a2a .client .transport .jsonrpc .JSONRPCTransport ;
1121import io .a2a .client .transport .jsonrpc .JSONRPCTransportConfigBuilder ;
12- import io .a2a .spec .A2AClientError ;
13- import io .a2a .spec .A2AClientException ;
1422import io .a2a .spec .AgentCard ;
1523import io .a2a .spec .Message ;
1624import io .a2a .spec .Part ;
17- import io .a2a .spec .Task ;
1825import io .a2a .spec .TaskArtifactUpdateEvent ;
1926import io .a2a .spec .TaskIdParams ;
2027import io .a2a .spec .TextPart ;
2128
22- import java .io .IOException ;
23- import java .util .Collections ;
24- import java .util .HashMap ;
25- import java .util .List ;
26- import java .util .Map ;
27- import java .util .concurrent .CountDownLatch ;
28- import java .util .concurrent .TimeUnit ;
29- import java .util .concurrent .atomic .AtomicBoolean ;
30- import java .util .concurrent .atomic .AtomicInteger ;
31-
3229/**
3330 * Test client demonstrating multi-pod A2A agent deployment with modernized message protocol.
3431 * <p>
@@ -52,6 +49,7 @@ public class A2ACloudExampleClient {
5249 private static final String AGENT_URL = System .getProperty ("agent.url" , "http://localhost:8080" );
5350 private static final int PROCESS_MESSAGE_COUNT = 8 ; // Number of "process" messages to send
5451 private static final int MESSAGE_INTERVAL_MS = 1500 ;
52+ private static final int MIN_PODS_TO_OBSERVE = 2 ;
5553
5654 // Test state
5755 private final Map <String , Integer > observedPods = Collections .synchronizedMap (new HashMap <>());
@@ -345,8 +343,8 @@ private void printResults() {
345343 if (testFailed .get ()) {
346344 System .out .println ("✗ TEST FAILED - Errors occurred during execution" );
347345 System .exit (1 );
348- } else if (observedPods .size () < 2 ) {
349- System .out .println ("✗ TEST FAILED - Expected at least 2 different pods, but only saw: " + observedPods .size ());
346+ } else if (observedPods .size () < MIN_PODS_TO_OBSERVE ) {
347+ System .out .printf ("✗ TEST FAILED - Expected at least %d different pods, but only saw: %d \n " , MIN_PODS_TO_OBSERVE , observedPods .size ());
350348 System .out .println (" This suggests load balancing is not working correctly." );
351349 System .exit (1 );
352350 } else {
0 commit comments