View Javadoc
1   /*
2    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
3    * The software in this package is published under the terms of the CPAL v1.0
4    * license, a copy of which has been included with this distribution in the
5    * LICENSE.txt file.
6    */
7   package org.mule.endpoint.outbound;
8   
9   import org.mule.DefaultMuleEvent;
10  import org.mule.DefaultMuleMessage;
11  import org.mule.api.MuleEvent;
12  import org.mule.api.MuleException;
13  import org.mule.api.config.MuleProperties;
14  import org.mule.api.endpoint.EndpointBuilder;
15  import org.mule.api.endpoint.OutboundEndpoint;
16  import org.mule.api.processor.InterceptingMessageProcessor;
17  import org.mule.api.processor.MessageProcessor;
18  import org.mule.endpoint.AbstractEndpointBuilder;
19  import org.mule.endpoint.AbstractMessageProcessorTestCase;
20  
21  import org.junit.Test;
22  
23  import static org.junit.Assert.assertEquals;
24  import static org.junit.Assert.assertNotNull;
25  
26  public class OutboundResponsePropertiesMessageProcessorTestCase extends AbstractMessageProcessorTestCase
27  {
28  
29      private static String MY_PROPERTY_KEY = "myProperty";
30      private static String MY_PROPERTY_VAL = "myPropertyValue";
31      private static String MULE_CORRELATION_ID_VAL = "152";
32  
33      @Test
34      public void testProcess() throws Exception
35      {
36          OutboundEndpoint endpoint = createTestOutboundEndpoint(null, null);
37          InterceptingMessageProcessor mp = new OutboundResponsePropertiesMessageProcessor(endpoint);
38          mp.setListener(new MessageProcessor()
39          {
40              public MuleEvent process(MuleEvent event) throws MuleException
41              {
42                  // return event with same payload but no properties
43                  return new DefaultMuleEvent(new DefaultMuleMessage(event.getMessage().getPayload(),
44                      muleContext), null, null);
45              }
46          });
47  
48          MuleEvent event = createTestOutboundEvent(endpoint);
49          event.getMessage().setOutboundProperty(MY_PROPERTY_KEY, MY_PROPERTY_VAL);
50          event.getMessage().setOutboundProperty(MuleProperties.MULE_CORRELATION_ID_PROPERTY, MULE_CORRELATION_ID_VAL);
51          MuleEvent result = mp.process(event);
52  
53          assertNotNull(result);
54          assertEquals(TEST_MESSAGE, result.getMessageAsString());
55          assertEquals(MY_PROPERTY_VAL, result.getMessage().getOutboundProperty(MY_PROPERTY_KEY));
56          assertEquals(MULE_CORRELATION_ID_VAL,
57                       result.getMessage().getOutboundProperty(MuleProperties.MULE_CORRELATION_ID_PROPERTY));
58      }
59  
60      @Override
61      protected void customizeEndpointBuilder(EndpointBuilder endpointBuilder)
62      {
63          endpointBuilder.setProperty(AbstractEndpointBuilder.PROPERTY_RESPONSE_PROPERTIES, "myProperty");
64      }
65  }