View Javadoc

1   /*
2    * $Id: AbstractPipelineTestCase.java 22422 2011-07-15 08:22:16Z dirk.olmes $
3    * --------------------------------------------------------------------------------------
4    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
5    *
6    * The software in this package is published under the terms of the CPAL v1.0
7    * license, a copy of which has been included with this distribution in the
8    * LICENSE.txt file.
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  }