Author: davsclaus Date: Mon Dec 6 12:16:17 2010 New Revision: 1042602 URL: http://svn.apache.org/viewvc?rev=1042602&view=rev Log: Added unit test based on user issue.
Added: camel/trunk/camel-core/src/test/java/org/apache/camel/issues/SetHeaderInDoCatchIssueTest.java Added: camel/trunk/camel-core/src/test/java/org/apache/camel/issues/SetHeaderInDoCatchIssueTest.java URL: http://svn.apache.org/viewvc/camel/trunk/camel-core/src/test/java/org/apache/camel/issues/SetHeaderInDoCatchIssueTest.java?rev=1042602&view=auto ============================================================================== --- camel/trunk/camel-core/src/test/java/org/apache/camel/issues/SetHeaderInDoCatchIssueTest.java (added) +++ camel/trunk/camel-core/src/test/java/org/apache/camel/issues/SetHeaderInDoCatchIssueTest.java Mon Dec 6 12:16:17 2010 @@ -0,0 +1,115 @@ +/** + * 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.issues; + +import org.apache.camel.ContextTestSupport; +import org.apache.camel.Exchange; +import org.apache.camel.ExchangeTimedOutException; +import org.apache.camel.Processor; +import org.apache.camel.builder.RouteBuilder; +import org.apache.camel.impl.JndiRegistry; + +/** + * @version $Revision$ + */ +public class SetHeaderInDoCatchIssueTest extends ContextTestSupport { + + public void testSuccess() { + Exchange exchange = template.request("direct:start", new Processor() { + public void process(Exchange exchange) throws Exception { + // empty message + } + }); + + assertEquals("CamsResponse", exchange.getOut().getHeader("Status")); + } + + public void testExchangeTimedOutException() { + Exchange exchange = template.request("direct:start", new Processor() { + public void process(Exchange exchange) throws Exception { + exchange.getIn().setBody("ExchangeTimedOutException"); + } + }); + + assertEquals("TimeOut", exchange.getOut().getHeader("Status")); + } + + public void testException() { + Exchange exchange = template.request("direct:start", new Processor() { + public void process(Exchange exchange) throws Exception { + exchange.getIn().setBody("Exception"); + } + }); + + assertEquals("ExceptionGeneral", exchange.getOut().getHeader("Status")); + } + + protected JndiRegistry createRegistry() throws Exception { + JndiRegistry registry = new JndiRegistry(createJndiContext()); + + registry.bind("A", new Processor() { + public void process(Exchange exchange) throws Exception { + log.info("A headers " + exchange.getIn().getHeaders()); + } + }); + + registry.bind("B", new Processor() { + public void process(Exchange exchange) throws Exception { + log.info("B headers " + exchange.getIn().getHeaders()); + + if ("ExchangeTimedOutException".equals(exchange.getIn().getBody(String.class))) { + throw new ExchangeTimedOutException(exchange, 1); + } else if ("Exception".equals(exchange.getIn().getBody(String.class))) { + throw new Exception(); + } + } + }); + + registry.bind("C", new Processor() { + public void process(Exchange exchange) throws Exception { + log.info("C headers " + exchange.getIn().getHeaders()); + } + }); + + return registry; + } + + @Override + protected RouteBuilder createRouteBuilder() throws Exception { + return new RouteBuilder() { + @Override + public void configure() throws Exception { + context.setTracing(true); + + from("direct:start") + .doTry() + .to("A") + .setHeader("CamelJmsDestinationName", constant("queue:outQueue")) + .inOut("B") + .setHeader("Status", constant("CamsResponse")) + .doCatch(ExchangeTimedOutException.class) + .setHeader("Status", constant("TimeOut")) + .doCatch(Exception.class) + .setHeader("Status", constant("ExceptionGeneral")) + .end() + .to("C") + .transform(body()); + } + }; + } + +}