svn commit: r802661 - /camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java

[da...@apache.org]

Author: davsclaus
Date: Mon Aug 10 06:02:40 2009
New Revision: 802661

URL: http://svn.apache.org/viewvc?rev=802661&view=rev
Log:
CAMEL-1895: unit test how to poll enrich a 2nd file triggered by a file consumer

Added:
    camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java   (with props)

Added: camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java
URL: http://svn.apache.org/viewvc/camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java?rev=802661&view=auto
==============================================================================
--- camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java (added)
+++ camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java Mon Aug 10 06:02:40 2009
@@ -0,0 +1,94 @@
+/**
+ * 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.file;
+
+import org.apache.camel.CamelExchangeException;
+import org.apache.camel.ConsumerTemplate;
+import org.apache.camel.ContextTestSupport;
+import org.apache.camel.Exchange;
+import org.apache.camel.Processor;
+import org.apache.camel.builder.RouteBuilder;
+import org.apache.camel.component.mock.MockEndpoint;
+import org.apache.camel.util.FileUtil;
+
+/**
+ * @version $Revision$
+ */
+public class FileConsumePollEnrichFileUsingProcessorTest extends ContextTestSupport {
+
+    @Override
+    protected void setUp() throws Exception {
+        deleteDirectory("target/enrich");
+        deleteDirectory("target/enrichdata");
+        super.setUp();
+    }
+
+    public void testPollEnrich() throws Exception {
+        getMockEndpoint("mock:start").expectedBodiesReceived("Start");
+
+        MockEndpoint mock = getMockEndpoint("mock:result");
+        mock.expectedBodiesReceived("Big file");
+
+        mock.expectedFileExists("target/enrich/.done/AAA.fin");
+        mock.expectedFileExists("target/enrichdata/.done/AAA.dat");
+        mock.expectedFileExists("target/enrichdata/BBB.dat");
+
+        template.sendBodyAndHeader("file://target/enrichdata", "Big file", Exchange.FILE_NAME, "AAA.dat");
+        template.sendBodyAndHeader("file://target/enrichdata", "Other Big file", Exchange.FILE_NAME, "BBB.dat");
+        template.sendBodyAndHeader("file://target/enrich", "Start", Exchange.FILE_NAME, "AAA.fin");
+
+        assertMockEndpointsSatisfied();
+    }
+
+    @Override
+    protected RouteBuilder createRouteBuilder() throws Exception {
+        return new RouteBuilder() {
+            @Override
+            public void configure() throws Exception {
+                from("file://target/enrich?move=.done")
+                    .process(new Processor() {
+                        public void process(Exchange exchange) throws Exception {
+                            String name = exchange.getIn().getHeader(Exchange.FILE_NAME_ONLY, String.class);
+                            name = FileUtil.stripExt(name) + ".dat";
+
+                            // use a consumer template to get the data file
+                            Exchange data = null;
+                            ConsumerTemplate con = exchange.getContext().createConsumerTemplate();
+                            try {
+                                // try to get the data file
+                                data = con.receive("file://target/enrichdata?move=.done&fileName=" + name, 5000);
+                            } finally {
+                                // stop the consumer as it does not need to poll for files anymore
+                                con.stop();
+                            }
+
+                            // if we found the data file then process it by sending it to the direct:data endpoint
+                            if (data != null) {
+                                template.send("direct:data", data);
+                            } else {
+                                // otherwise do a rollback
+                                throw new CamelExchangeException("Cannot find the data file " + name, exchange);
+                            }
+                        }
+                    }).to("mock:start");
+
+                from("direct:data")
+                    .to("mock:result");
+            }
+        };
+    }
+}

Propchange: camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java
------------------------------------------------------------------------------
    svn:eol-style = native

Propchange: camel/trunk/camel-core/src/test/java/org/apache/camel/component/file/FileConsumePollEnrichFileUsingProcessorTest.java
------------------------------------------------------------------------------
    svn:keywords = Rev Date