1
2
3
4
5
6
7
8
9
10
11 package org.mule.test.integration.routing;
12
13 import org.mule.api.MuleMessage;
14 import org.mule.api.MuleMessageCollection;
15 import org.mule.api.context.notification.RoutingNotificationListener;
16 import org.mule.api.context.notification.ServerNotification;
17 import org.mule.context.notification.RoutingNotification;
18 import org.mule.module.client.MuleClient;
19 import org.mule.tck.FunctionalTestCase;
20
21 import edu.emory.mathcs.backport.java.util.concurrent.CountDownLatch;
22 import edu.emory.mathcs.backport.java.util.concurrent.TimeUnit;
23
24 public class AsyncReplyTimeoutTestCase extends FunctionalTestCase
25 {
26 private CountDownLatch latch;
27
28 protected String getConfigResources()
29 {
30 return "org/mule/test/integration/routing/multi-async-repy-timeout.xml";
31 }
32
33 public void testAggregatorTimeoutWithoutFailure() throws Exception
34 {
35 latch = new CountDownLatch(1);
36
37 muleContext.registerListener(new RoutingNotificationListener() {
38 public void onNotification(ServerNotification notification)
39 {
40 if(notification.getAction() == RoutingNotification.MISSED_ASYNC_REPLY)
41 {
42 latch.countDown();
43 assertEquals("test Received Late!", ((MuleMessage)((RoutingNotification)notification).getSource()).getPayload());
44 }
45 }
46 });
47
48 String message = "test";
49 MuleClient client = new MuleClient();
50 MuleMessage result = client.send("vm://distributor.queue", message, null);
51 assertNotNull(result);
52 assertTrue(result instanceof MuleMessageCollection);
53 MuleMessageCollection mc = (MuleMessageCollection)result;
54 assertEquals(2, mc.size());
55 for (int i = 0; i < mc.getMessagesAsArray().length; i++)
56 {
57 MuleMessage msg = mc.getMessagesAsArray()[i];
58 assertEquals("test Received", msg.getPayload());
59 }
60
61 assertTrue(latch.await(3000, TimeUnit.MILLISECONDS));
62 }
63 }