org.structr.files.ftp.FtpFilesTest.java Source code

Introduction

Here is the source code for org.structr.files.ftp.FtpFilesTest.java

Source

/**
 * Copyright (C) 2010-2016 Structr GmbH
 *
 * This file is part of Structr <http://structr.org>.
 *
 * Structr is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * Structr is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with Structr.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.structr.files.ftp;

import java.io.IOException;
import java.io.InputStream;
import java.util.logging.Logger;
import static junit.framework.TestCase.assertEquals;
import static junit.framework.TestCase.assertNotNull;
import org.apache.commons.io.IOUtils;
import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.structr.common.error.FrameworkException;
import org.structr.core.app.StructrApp;
import org.structr.core.graph.Tx;
import org.structr.web.common.FtpTest;

/**
 * Tests for FTP files.
 *
 *
 */
public class FtpFilesTest extends FtpTest {

    private static final Logger logger = Logger.getLogger(FtpFilesTest.class.getName());

    public void test00StoreFile() {

        FTPClient ftp = setupFTPClient();
        final String name1 = "file1";
        final String name2 = "file2";

        try (final Tx tx = app.tx()) {

            FTPFile[] files = ftp.listFiles();

            assertNotNull(files);
            assertEquals(0, files.length);

            ftp.setFileType(FTP.ASCII_FILE_TYPE);
            ftp.setAutodetectUTF8(true);

            // Store a file
            InputStream in = IOUtils.toInputStream("Test Content");
            ftp.storeFile(name1, in);

            in.close();

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        String[] fileNames = null;

        try (final Tx tx = app.tx()) {

            fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(1, fileNames.length);
            assertEquals(name1, fileNames[0]);

            // Create second file in /
            createFTPFile(null, name2);

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(2, fileNames.length);
            assertEquals(name1, fileNames[0]);
            assertEquals(name2, fileNames[1]);

            ftp.disconnect();

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }
    }

    public void test01ListFiles() {

        FTPClient ftp = setupFTPClient();
        final String name1 = "file1";
        final String name2 = "file2";

        try (final Tx tx = app.tx()) {

            FTPFile[] files = ftp.listFiles();

            assertNotNull(files);
            assertEquals(0, files.length);

            // Create files by API methods
            createFTPFile(null, name1);

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        String[] fileNames = null;

        try (final Tx tx = app.tx()) {

            fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(1, fileNames.length);
            assertEquals(name1, fileNames[0]);

            // Create second file in /
            createFTPFile(null, name2);

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(2, fileNames.length);
            assertEquals(name1, fileNames[0]);
            assertEquals(name2, fileNames[1]);

            ftp.disconnect();

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }
    }

    public void test02RenameFile() {

        FTPClient ftp = setupFTPClient();
        final String name2 = "file2";
        final String name1 = "file1";

        try (final Tx tx = StructrApp.getInstance(securityContext).tx()) {

            FTPFile[] files = ftp.listFiles();

            assertNotNull(files);
            assertEquals(0, files.length);

            // Create files by API methods
            createFTPFile(null, name1);

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            ftp.rename(name1, name2);

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            String[] fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(1, fileNames.length);
            assertEquals(name2, fileNames[0]);

            ftp.disconnect();

            tx.success();

        } catch (IOException | FrameworkException ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }
    }

    public void test03MoveFile() {

        FTPClient ftp = setupFTPClient();
        final String name1 = "file1";
        final String name2 = "dir1";

        try (final Tx tx = app.tx()) {

            FTPFile[] files = ftp.listFiles();

            assertNotNull(files);
            assertEquals(0, files.length);

            // Create files by API methods
            createFTPFile(null, name1);

            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            // Create folder in /
            createFTPDirectory(null, name2);
            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            // Move file to dir
            ftp.rename("/" + name1, "/" + name2 + "/" + name1);
            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            ftp.changeWorkingDirectory("/" + name2);

            String[] fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(1, fileNames.length);
            assertEquals(name1, fileNames[0]);

            ftp.disconnect();

            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }
    }

    public void test04MoveFileToRoot() {

        FTPClient ftp = setupFTPClient();

        final String name1 = "file1";
        final String name2 = "dir1";

        try (final Tx tx = app.tx()) {

            FTPFile[] files = ftp.listFiles();

            assertNotNull(files);
            assertEquals(0, files.length);

            // Create files by API methods
            createFTPFile(null, name1);
            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            // Create folder in /
            createFTPDirectory(null, name2);
            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            // Move file to dir
            ftp.rename("/" + name1, "/" + name2 + "/" + name1);
            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        String[] fileNames = null;
        try (final Tx tx = app.tx()) {

            ftp.changeWorkingDirectory("/" + name2);

            fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(1, fileNames.length);
            assertEquals(name1, fileNames[0]);

            // Move file back to /
            ftp.rename("/" + name2 + "/" + name1, "/" + name1);
            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            ftp.changeWorkingDirectory("/");
            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }

        try (final Tx tx = app.tx()) {

            fileNames = ftp.listNames();

            assertNotNull(fileNames);
            assertEquals(2, fileNames.length);
            assertEquals(name2, fileNames[0]);
            assertEquals(name1, fileNames[1]);

            ftp.disconnect();

            tx.success();

        } catch (Exception ex) {
            ex.printStackTrace();
            fail("Unexpected exception: " + ex.getMessage());
        }
    }

}