org.structr.web.files.FtpTest.java Source code

Introduction

Here is the source code for org.structr.web.files.FtpTest.java

Source

/**
 * Copyright (C) 2010-2018 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.web.files;

import java.io.IOException;
import static junit.framework.TestCase.assertEquals;
import static junit.framework.TestCase.assertNotNull;
import static junit.framework.TestCase.assertTrue;
import static junit.framework.TestCase.fail;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.structr.common.error.FrameworkException;
import org.structr.core.app.StructrApp;
import org.structr.core.entity.Principal;
import org.structr.core.graph.Tx;
import org.structr.core.property.PropertyMap;
import org.structr.web.common.FileHelper;
import org.structr.web.entity.AbstractFile;
import org.structr.web.entity.File;
import org.structr.web.entity.Folder;
import org.structr.web.entity.User;

/**
 * Common class for FTP tests
 *
 *
 */
public abstract class FtpTest extends StructrUiTest {

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

    protected User ftpUser;

    protected User createFTPUser(final String username, final String password) throws FrameworkException {
        PropertyMap props = new PropertyMap();
        props.put(StructrApp.key(Principal.class, "name"), username);
        props.put(StructrApp.key(Principal.class, "password"), password);
        return (User) createTestNodes(User.class, 1, props).get(0);
    }

    protected Folder createFTPDirectory(final String path, final String name) throws FrameworkException {
        PropertyMap props = new PropertyMap();
        props.put(Folder.name, name);
        props.put(Folder.owner, ftpUser);
        Folder dir = (Folder) createTestNodes(Folder.class, 1, props).get(0);

        if (StringUtils.isNotBlank(path)) {
            AbstractFile parent = FileHelper.getFileByAbsolutePath(securityContext, path);
            if (parent != null && parent instanceof Folder) {
                Folder parentFolder = (Folder) parent;
                dir.setParent(parentFolder);
            }
        }

        logger.info("FTP directory {} created successfully.", dir);

        return dir;
    }

    protected File createFTPFile(final String path, final String name) throws FrameworkException {
        PropertyMap props = new PropertyMap();
        props.put(StructrApp.key(File.class, "name"), name);
        props.put(StructrApp.key(File.class, "size"), 0L);
        props.put(StructrApp.key(File.class, "owner"), ftpUser);
        File file = (File) createTestNodes(File.class, 1, props).get(0);

        if (StringUtils.isNotBlank(path)) {
            AbstractFile parent = FileHelper.getFileByAbsolutePath(securityContext, path);
            if (parent != null && parent instanceof Folder) {
                Folder parentFolder = (Folder) parent;
                file.setParent(parentFolder);
            }
        }

        logger.info("FTP file {} created successfully.", file);

        return file;
    }

    protected void assertEmptyDirectory(final FTPClient ftp) {

        FTPFile[] dirs = null;

        try {

            dirs = ftp.listDirectories();

        } catch (IOException ex) {

            logger.error("Error in FTP test", ex);
            fail("Unexpected exception: " + ex.getMessage());

        }

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

    }

    /**
     * Creates an FTP client, a backend user and logs this user in.
     *
     * @param username
     * @return
     */
    protected FTPClient setupFTPClient(final String username) {

        FTPClient ftp = new FTPClient();

        try {

            ftp.connect("localhost", ftpPort);
            logger.info("Reply from FTP server:", ftp.getReplyString());

            int reply = ftp.getReplyCode();
            assertTrue(FTPReply.isPositiveCompletion(reply));

            String password = "ftpuserpw1";

            try (final Tx tx = StructrApp.getInstance(securityContext).tx()) {
                ftpUser = createFTPUser(username, password);
                tx.success();
            } catch (FrameworkException fex) {
                logger.error("Unable to create FTP user", fex);
            }

            boolean loginSuccess = ftp.login(username, password);
            logger.info("Tried to login as {}/{}: {}", new Object[] { username, password, loginSuccess });
            assertTrue(loginSuccess);

            reply = ftp.getReplyCode();
            assertEquals(FTPReply.USER_LOGGED_IN, reply);

        } catch (IOException ex) {
            logger.error("Error in FTP test", ex);
            fail("Unexpected exception: " + ex.getMessage());
        }

        return ftp;

    }

    protected void disconnect(final FTPClient ftp) {
        try {
            ftp.disconnect();
        } catch (IOException ex) {
            logger.error("Error while disconnecting from FTP server", ex);
            fail("Unexpected exception: " + ex.getMessage());
        }
    }
}