org.eftp.ftpserver.st.FTPServerWrapperIT.java Source code

Introduction

Here is the source code for org.eftp.ftpserver.st.FTPServerWrapperIT.java

Source

package org.eftp.ftpserver.st;

/*
 * #%L
 * ftpserver-st
 * %%
 * Copyright (C) 2013 e2ftp
 * %%
 * Licensed 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.
 * #L%
 */

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import static org.hamcrest.CoreMatchers.is;
import org.junit.After;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;
import org.junit.Before;
import org.junit.Test;

public class FTPServerWrapperIT {

    private FTPClient client;
    private static final String HOST = "localhost";
    private static final int PORT = 8888;
    private static final String USER = "duke";
    private static final String PASSWORD = "duke";

    @Before
    public void init() {
        this.client = new FTPClient();
    }

    @Test
    public void connectAndDisconnect() throws IOException {
        this.client.connect(HOST, PORT);
        this.client.disconnect();
    }

    @Test
    public void login() throws IOException {
        connect();
    }

    @Test
    public void list() throws IOException {
        connect();
        FTPFile[] listDirectories = this.client.listDirectories();
        assertNotNull(listDirectories);
        assertTrue(listDirectories.length > 0);
    }

    @Test
    public void sendAndReceive() throws IOException {
        connect();
        String remoteFile = "e2ftp-client" + System.currentTimeMillis() + ".txt";
        final String expected = "just a test";
        sendFile(remoteFile, expected);
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        this.client.retrieveFile(remoteFile, baos);
        byte[] actual = baos.toByteArray();
        assertNotNull(actual);
        assertThat(new String(actual), is(expected));
        this.client.deleteFile(remoteFile);

        baos = new ByteArrayOutputStream();
        boolean fileExists = this.client.retrieveFile(remoteFile, baos);
        assertFalse(fileExists);
    }

    public void sendFile(String file, String content) throws IOException {
        ByteArrayInputStream bais = new ByteArrayInputStream(content.getBytes());
        this.client.storeFile(file, bais);
    }

    @After
    public void cleanup() throws IOException {
        this.client.disconnect();
    }

    public void connect() throws IOException {
        this.client.connect(HOST, PORT);
        this.client.login(USER, PASSWORD);
    }

}