Another test timing issue

Author: kabir

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

@@ -355,20 +355,23 @@ void testReplicatedEventProcessedWhenTaskActive() 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");
+
         // Process a replicated event for an active task
         ReplicatedEventQueueItem replicatedEvent = new ReplicatedEventQueueItem(taskId, testEvent);
         queueManager.onReplicatedEvent(replicatedEvent);
 
-        // Queue should be created and event should be enqueued
-        EventQueue queue = queueManager.get(taskId);
-        assertNotNull(queue, "Queue should be created for active task");
-
-        // Verify the event was enqueued
-        // 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
         Event dequeuedEvent;
         try {
-            // Wait up to 5 seconds for event (async processing needs time)
+            // Wait up to 5 seconds for event (async MainEventBusProcessor needs time)
             EventQueueItem item = childQueue.dequeueEventItem(5000);
             assertNotNull(item, "Event should be available in queue");
             dequeuedEvent = item.getEvent();