Author: cmueller Date: Tue Feb 28 23:31:58 2012 New Revision: 1294893 URL: http://svn.apache.org/viewvc?rev=1294893&view=rev Log: CAMEL-5052: DefaultUnitOfWork should copy headers from the original message if it's a JmsMessage
Added: camel/trunk/components/camel-jms/src/test/java/org/apache/camel/component/jms/JmsMessageAsOriginalMessageInDefaultUnitOfWorkTest.java Modified: camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultUnitOfWork.java Modified: camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultUnitOfWork.java URL: http://svn.apache.org/viewvc/camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultUnitOfWork.java?rev=1294893&r1=1294892&r2=1294893&view=diff ============================================================================== --- camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultUnitOfWork.java (original) +++ camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultUnitOfWork.java Tue Feb 28 23:31:58 2012 @@ -86,7 +86,7 @@ public class DefaultUnitOfWork implement if (exchange.getIn().getClass().getSimpleName().equals("JmsMessage")) { this.originalInMessage = new DefaultMessage(); this.originalInMessage.setBody(exchange.getIn().getBody()); - // cannot copy headers with a JmsMessage as the underlying javax.jms.Message object goes nuts + this.originalInMessage.setHeaders(exchange.getIn().getHeaders()); } else { this.originalInMessage = exchange.getIn().copy(); } Added: camel/trunk/components/camel-jms/src/test/java/org/apache/camel/component/jms/JmsMessageAsOriginalMessageInDefaultUnitOfWorkTest.java URL: http://svn.apache.org/viewvc/camel/trunk/components/camel-jms/src/test/java/org/apache/camel/component/jms/JmsMessageAsOriginalMessageInDefaultUnitOfWorkTest.java?rev=1294893&view=auto ============================================================================== --- camel/trunk/components/camel-jms/src/test/java/org/apache/camel/component/jms/JmsMessageAsOriginalMessageInDefaultUnitOfWorkTest.java (added) +++ camel/trunk/components/camel-jms/src/test/java/org/apache/camel/component/jms/JmsMessageAsOriginalMessageInDefaultUnitOfWorkTest.java Tue Feb 28 23:31:58 2012 @@ -0,0 +1,72 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.camel.component.jms; + +import static org.apache.camel.component.jms.JmsComponent.jmsComponentAutoAcknowledge; + +import javax.jms.ConnectionFactory; + +import org.apache.camel.CamelContext; +import org.apache.camel.EndpointInject; +import org.apache.camel.builder.RouteBuilder; +import org.apache.camel.component.mock.MockEndpoint; +import org.apache.camel.test.junit4.CamelTestSupport; +import org.junit.Test; + +public class JmsMessageAsOriginalMessageInDefaultUnitOfWorkTest extends CamelTestSupport { + + @EndpointInject(uri = "mock:result") + private MockEndpoint mockResult; + + @Test + public void testUseOriginalMessage() throws Exception { + context.addRoutes(new RouteBuilder() { + @Override + public void configure() throws Exception { + onException(Exception.class) + .useOriginalMessage() + .to(mockResult); + + from("jms:queue:foo") + .throwException(new Exception("forced exception for test")); + } + }); + context.start(); + + mockResult.expectedBodiesReceived("Hello World"); + mockResult.expectedHeaderReceived("header-key", "header-value"); + + template.sendBodyAndHeader("jms:queue:foo", "Hello World", "header-key", "header-value"); + + assertMockEndpointsSatisfied(); + } + + protected CamelContext createCamelContext() throws Exception { + CamelContext camelContext = super.createCamelContext(); + + ConnectionFactory connectionFactory = CamelJmsTestHelper.createConnectionFactory(); + camelContext.addComponent("jms", jmsComponentAutoAcknowledge(connectionFactory)); + + return camelContext; + } + + @Override + public boolean isUseRouteBuilder() { + return false; + } +} +