camel git commit: CAMEL-8522: Added unit test how this now can be done.

[da...@apache.org]

Repository: camel
Updated Branches:
  refs/heads/master ac06cc9dd -> 32fa4eabf


CAMEL-8522: Added unit test how this now can be done.


Project: http://git-wip-us.apache.org/repos/asf/camel/repo
Commit: http://git-wip-us.apache.org/repos/asf/camel/commit/32fa4eab
Tree: http://git-wip-us.apache.org/repos/asf/camel/tree/32fa4eab
Diff: http://git-wip-us.apache.org/repos/asf/camel/diff/32fa4eab

Branch: refs/heads/master
Commit: 32fa4eabf8154642b0dd76e74bc5fba36bc9bdfb
Parents: ac06cc9
Author: Claus Ibsen <da...@apache.org>
Authored: Wed May 6 15:13:02 2015 +0200
Committer: Claus Ibsen <da...@apache.org>
Committed: Wed May 6 15:13:02 2015 +0200

----------------------------------------------------------------------
 .../jms/ActiveMQOriginalDestinationTest.java    | 113 +++++++++++++++++++
 1 file changed, 113 insertions(+)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/camel/blob/32fa4eab/components/camel-jms/src/test/java/org/apache/camel/component/jms/ActiveMQOriginalDestinationTest.java
----------------------------------------------------------------------
diff --git a/components/camel-jms/src/test/java/org/apache/camel/component/jms/ActiveMQOriginalDestinationTest.java b/components/camel-jms/src/test/java/org/apache/camel/component/jms/ActiveMQOriginalDestinationTest.java
new file mode 100644
index 0000000..5510376
--- /dev/null
+++ b/components/camel-jms/src/test/java/org/apache/camel/component/jms/ActiveMQOriginalDestinationTest.java
@@ -0,0 +1,113 @@
+/**
+ * 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 javax.jms.ConnectionFactory;
+import javax.jms.Message;
+import javax.jms.Session;
+
+import org.apache.activemq.command.ActiveMQDestination;
+import org.apache.activemq.command.ActiveMQMessage;
+import org.apache.camel.CamelContext;
+import org.apache.camel.Exchange;
+import org.apache.camel.builder.RouteBuilder;
+import org.apache.camel.component.mock.MockEndpoint;
+import org.apache.camel.test.junit4.CamelTestSupport;
+import org.junit.Test;
+
+import static org.apache.camel.component.jms.JmsComponent.jmsComponentAutoAcknowledge;
+
+/**
+ * @version 
+ */
+public class ActiveMQOriginalDestinationTest extends CamelTestSupport {
+
+    protected String componentName = "activemq";
+
+    @Test
+    public void testActiveMQOriginalDestination() throws Exception {
+        MockEndpoint mock = getMockEndpoint("mock:result");
+        mock.expectedMessageCount(1);
+
+        template.sendBody("activemq:queue:foo", "Hello World");
+
+        assertMockEndpointsSatisfied();
+
+        // consume from bar
+        Exchange out = consumer.receive("activemq:queue:bar", 5000);
+        assertNotNull(out);
+
+        // and we should have foo as the original destination
+        JmsMessage msg = out.getIn(JmsMessage.class);
+        Message jms = msg.getJmsMessage();
+        ActiveMQMessage amq = assertIsInstanceOf(ActiveMQMessage.class, jms);
+        ActiveMQDestination original = amq.getOriginalDestination();
+        assertNotNull(original);
+        assertEquals("foo", original.getPhysicalName());
+        assertEquals("Queue", original.getDestinationTypeAsString());
+    }
+
+    protected CamelContext createCamelContext() throws Exception {
+        CamelContext camelContext = super.createCamelContext();
+
+        ConnectionFactory connectionFactory = CamelJmsTestHelper.createConnectionFactory();
+        camelContext.addComponent(componentName, jmsComponentAutoAcknowledge(connectionFactory));
+
+        JmsComponent jms = camelContext.getComponent(componentName, JmsComponent.class);
+        jms.setMessageCreatedStrategy(new OriginalDestinationPropagateStrategy());
+
+        return camelContext;
+    }
+
+    @Override
+    protected RouteBuilder createRouteBuilder() throws Exception {
+        return new RouteBuilder() {
+            @Override
+            public void configure() throws Exception {
+                from("activemq:queue:foo")
+                    .to("activemq:queue:bar")
+                    .to("mock:result");
+            }
+        };
+    }
+
+    /**
+     * A strategy to enrich JMS message with their original destination if the Camel
+     * route originates from a JMS destination.
+     * <p/>
+     * This implementation uses ActiveMQ specific code which can be moved to activemq-camel
+     * when it supports Camel 2.16
+     */
+    private class OriginalDestinationPropagateStrategy implements MessageCreatedStrategy {
+
+        @Override
+        public void onMessageCreated(Message message, Session session, Exchange exchange, Throwable cause) {
+            if (exchange.getIn() instanceof JmsMessage) {
+                JmsMessage msg = exchange.getIn(JmsMessage.class);
+                Message jms = msg.getJmsMessage();
+                if (message instanceof ActiveMQMessage) {
+                    ActiveMQMessage amq = (ActiveMQMessage) jms;
+                    ActiveMQDestination from = amq.getDestination();
+
+                    if (from != null && message instanceof ActiveMQMessage) {
+                        ((ActiveMQMessage) message).setOriginalDestination(from);
+                    }
+                }
+            }
+        }
+    }
+}