1
2
3
4
5
6
7
8
9
10
11 package org.mule.test.integration.models;
12
13 import org.mule.api.MuleMessage;
14 import org.mule.module.client.MuleClient;
15 import org.mule.tck.junit4.FunctionalTestCase;
16
17 import java.util.ArrayList;
18 import java.util.Iterator;
19 import java.util.List;
20
21 import org.junit.Test;
22
23 import static org.junit.Assert.assertEquals;
24 import static org.junit.Assert.assertNotNull;
25
26 public abstract class AbstractPipelineTestCase extends FunctionalTestCase
27 {
28
29 protected int getNumberOfMessages()
30 {
31 return 100;
32 }
33
34 @Test
35 public void testPipelineSynchronous() throws Exception
36 {
37 MuleClient client = new MuleClient(muleContext);
38 List results = new ArrayList();
39 for (int i = 0; i < getNumberOfMessages(); i++)
40 {
41 MuleMessage result = client.send("component1.endpoint", "test", null);
42 assertNotNull(result);
43 results.add(result);
44 }
45
46 assertEquals(results.size(), getNumberOfMessages());
47 for (Iterator iterator = results.iterator(); iterator.hasNext();)
48 {
49 MuleMessage message = (MuleMessage)iterator.next();
50 assertEquals("request received by service 3", message.getPayloadAsString());
51 }
52 }
53
54 @Test
55 public void testPipelineAsynchronous() throws Exception
56 {
57 MuleClient client = new MuleClient(muleContext);
58
59 List results = new ArrayList();
60 for (int i = 0; i < getNumberOfMessages(); i++)
61 {
62 client.dispatch("component1.endpoint", "test", null);
63 }
64
65 for (int i = 0; i < getNumberOfMessages(); i++)
66 {
67 MuleMessage result = client.request("results.endpoint", 1000);
68 assertNotNull(result);
69 results.add(result);
70 }
71 assertEquals(results.size(), getNumberOfMessages());
72 for (Iterator iterator = results.iterator(); iterator.hasNext();)
73 {
74 MuleMessage message = (MuleMessage)iterator.next();
75 assertEquals("request received by service 3", message.getPayloadAsString());
76 }
77 }
78 }