More test timing fixes

Author: kabir

extras/queue-manager-replicated/core/src/test/java/io/a2a/extras/queuemanager/replicated/core/ReplicatedQueueManagerTest.java

@@ -149,19 +149,21 @@ void testReplicatedEventCreatesQueueIfNeeded() throws InterruptedException {
         // Verify no queue exists initially
         assertNull(queueManager.get(taskId));
 
+        // Create a ChildQueue BEFORE processing the replicated event
+        // This ensures the ChildQueue exists when MainEventBusProcessor distributes the event
+        EventQueue childQueue = queueManager.createOrTap(taskId);
+        assertNotNull(childQueue, "ChildQueue should be created");
+
+        // Verify MainQueue was created
+        EventQueue mainQueue = queueManager.get(taskId);
+        assertNotNull(mainQueue, "MainQueue should exist after createOrTap");
+
         ReplicatedEventQueueItem replicatedEvent = new ReplicatedEventQueueItem(taskId, testEvent);
 
         // Process the replicated event
         assertDoesNotThrow(() -> queueManager.onReplicatedEvent(replicatedEvent));
 
-        // Verify that a queue was created and the event was enqueued
-        EventQueue queue = queueManager.get(taskId);
-        assertNotNull(queue, "Queue should be created when processing replicated event for non-existent task");
-
-        // Verify the event was enqueued by dequeuing it
-        // Need to tap() the MainQueue to get a ChildQueue for consumption
-        EventQueue childQueue = queue.tap();
-
+        // Verify the event was enqueued and distributed to our ChildQueue
         // Use retry logic to handle async MainEventBusProcessor distribution
         EventQueueItem item = dequeueEventWithRetry(childQueue, 5000);
         assertNotNull(item, "Event should be available in queue");