| /* |
| * 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.remote; |
| |
| import java.io.File; |
| |
| import org.apache.camel.Endpoint; |
| import org.apache.camel.Exchange; |
| import org.apache.camel.Producer; |
| import org.apache.camel.component.file.GenericFileOperationFailedException; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| public class FromFtpConsumerTemplateRollbackTest extends FtpServerTestSupport { |
| |
| protected String getFtpUrl() { |
| return "ftp://admin@localhost:" + getPort() + "/deletefile?password=admin&binary=false&delete=true"; |
| } |
| |
| protected String getFtpUrlInvalid() { |
| // use invalid starting directory and do not allow creating it so we force the poll to fail |
| return "ftp://admin@localhost:" + getPort() + "/unknown?password=admin&binary=false&delete=true&autoCreate=false"; |
| } |
| |
| @Override |
| @Before |
| public void setUp() throws Exception { |
| super.setUp(); |
| prepareFtpServer(); |
| } |
| |
| @Test |
| public void testConsumerTemplateRollback() throws Exception { |
| try { |
| consumer.receiveBody(getFtpUrlInvalid(), 2000, String.class); |
| fail("Should fail and rollback"); |
| } catch (Exception e) { |
| GenericFileOperationFailedException ge = assertIsInstanceOf(GenericFileOperationFailedException.class, e); |
| assertEquals(550, ge.getCode()); |
| } |
| } |
| |
| private void prepareFtpServer() throws Exception { |
| // prepares the FTP Server by creating a file on the server that we want to unit |
| // test that we can pool and store as a local file |
| Endpoint endpoint = context.getEndpoint(getFtpUrl()); |
| Exchange exchange = endpoint.createExchange(); |
| exchange.getIn().setBody("Hello World this file will be deleted"); |
| exchange.getIn().setHeader(Exchange.FILE_NAME, "hello.txt"); |
| Producer producer = endpoint.createProducer(); |
| producer.start(); |
| producer.process(exchange); |
| producer.stop(); |
| |
| // assert file is created |
| File file = new File(FTP_ROOT_DIR + "/deletefile/hello.txt"); |
| assertTrue("The file should exists", file.exists()); |
| } |
| |
| } |