| /** |
| * 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 java.io.File; |
| |
| import org.apache.camel.ContextTestSupport; |
| import org.apache.camel.Exchange; |
| import org.apache.camel.Processor; |
| import org.apache.camel.RuntimeCamelException; |
| import org.apache.camel.ValidationException; |
| import org.apache.camel.builder.RouteBuilder; |
| import org.apache.camel.component.mock.MockEndpoint; |
| |
| /** |
| * Unit test for consuming files but the exchange fails and is handled |
| * by the failure handler (usually the DeadLetterChannel) |
| */ |
| public class FileConsumerFailureHandledTest extends ContextTestSupport { |
| |
| @Override |
| protected void setUp() throws Exception { |
| deleteDirectory("target/messages/input"); |
| super.setUp(); |
| } |
| |
| public void testParis() throws Exception { |
| MockEndpoint mock = getMockEndpoint("mock:valid"); |
| mock.expectedBodiesReceived("Hello Paris"); |
| |
| template.sendBodyAndHeader("file:target/messages/input/", "Paris", Exchange.FILE_NAME, "paris.txt"); |
| mock.assertIsSatisfied(); |
| |
| oneExchangeDone.matchesMockWaitTime(); |
| |
| assertFiles("paris.txt", true); |
| } |
| |
| public void testLondon() throws Exception { |
| MockEndpoint mock = getMockEndpoint("mock:invalid"); |
| // we get the original input so its not Hello London but only London |
| mock.expectedBodiesReceived("London"); |
| |
| template.sendBodyAndHeader("file:target/messages/input/", "London", Exchange.FILE_NAME, "london.txt"); |
| mock.assertIsSatisfied(); |
| |
| oneExchangeDone.matchesMockWaitTime(); |
| |
| // london should be deleted as we have failure handled it |
| assertFiles("london.txt", true); |
| } |
| |
| public void testDublin() throws Exception { |
| MockEndpoint mock = getMockEndpoint("mock:beer"); |
| // we get the original input so its not Hello London but only London |
| mock.expectedBodiesReceived("Dublin"); |
| |
| template.sendBodyAndHeader("file:target/messages/input/", "Dublin", Exchange.FILE_NAME, "dublin.txt"); |
| mock.assertIsSatisfied(); |
| |
| oneExchangeDone.matchesMockWaitTime(); |
| |
| // dublin should NOT be deleted, but should be retired on next consumer |
| assertFiles("dublin.txt", false); |
| } |
| |
| public void testMadrid() throws Exception { |
| MockEndpoint mock = getMockEndpoint("mock:error"); |
| // we get the original input so its not Hello London but only London |
| mock.expectedBodiesReceived("Madrid"); |
| |
| template.sendBodyAndHeader("file:target/messages/input/", "Madrid", Exchange.FILE_NAME, "madrid.txt"); |
| mock.assertIsSatisfied(); |
| |
| oneExchangeDone.matchesMockWaitTime(); |
| |
| // madrid should NOT be deleted, but should be retired on next consumer |
| assertFiles("madrid.txt", false); |
| } |
| |
| private static void assertFiles(String filename, boolean deleted) throws InterruptedException { |
| // file should be deleted as delete=true in parameter in the route below |
| File file = new File("target/messages/input/" + filename); |
| assertEquals("File " + filename + " should be deleted: " + deleted, deleted, !file.exists()); |
| |
| // and no lock files |
| String lock = filename + FileComponent.DEFAULT_LOCK_FILE_POSTFIX; |
| file = new File("target/messages/input/" + lock); |
| assertFalse("File " + lock + " should be deleted", file.exists()); |
| } |
| |
| protected RouteBuilder createRouteBuilder() throws Exception { |
| return new RouteBuilder() { |
| public void configure() throws Exception { |
| // make sure mock:error is the dead letter channel |
| // use no delay for fast unit testing |
| errorHandler(deadLetterChannel("mock:error").maximumRedeliveries(2).redeliveryDelay(0).logStackTrace(false).handled(false)); |
| |
| // special for not handled when we got beer |
| onException(ValidationException.class).onWhen(exceptionMessage().contains("beer")) |
| .handled(false).to("mock:beer"); |
| |
| // special failure handler for ValidationException |
| onException(ValidationException.class).handled(true).to("mock:invalid"); |
| |
| // our route logic to process files from the input folder |
| from("file:target/messages/input/?delete=true"). |
| process(new MyValidatorProcessor()). |
| to("mock:valid"); |
| } |
| }; |
| } |
| |
| private class MyValidatorProcessor implements Processor { |
| public void process(Exchange exchange) throws Exception { |
| String body = exchange.getIn().getBody(String.class); |
| if ("London".equals(body)) { |
| throw new ValidationException(exchange, "Forced exception by unit test"); |
| } else if ("Madrid".equals(body)) { |
| throw new RuntimeCamelException("Madrid is not a supported city"); |
| } else if ("Dublin".equals(body)) { |
| throw new ValidationException(exchange, "Dublin have good beer"); |
| } |
| exchange.getOut().setBody("Hello " + body); |
| } |
| } |
| |
| } |