JDBCOutboundFunctionalTestCase.java

package com.muleinaction;

import org.junit.BeforeClass;
import org.junit.Test;
import org.mockftpserver.fake.FakeFtpServer;
import org.mockftpserver.fake.UserAccount;
import org.mockftpserver.fake.filesystem.DirectoryEntry;
import org.mockftpserver.fake.filesystem.FileSystem;
import org.mockftpserver.fake.filesystem.UnixFakeFileSystem;
import org.mule.api.context.notification.EndpointMessageNotificationListener;
import org.mule.api.context.notification.ServerNotification;
import org.mule.context.notification.EndpointMessageNotification;
import org.mule.tck.junit4.FunctionalTestCase;
import org.mule.util.FileUtils;
import org.springframework.jdbc.BadSqlGrammarException;
import org.springframework.jdbc.core.JdbcTemplate;

import javax.sql.DataSource;
import java.io.File;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import static junit.framework.Assert.assertEquals;
import static org.junit.Assert.assertTrue;


public class JDBCOutboundFunctionalTestCase extends FunctionalTestCase {

    private FakeFtpServer fakeFtpServer;

    private CountDownLatch putFilesLatch;

    private  JdbcTemplate template;


    @Override
    protected String getConfigResources() {
        return "src/main/app/jdbc-outbound-config.xml";
    }

    @BeforeClass
    public static void setupDirectories() throws Exception {
        File dataDirectory = new File("./data");
        if (dataDirectory.exists()) {
            FileUtils.deleteDirectory(dataDirectory);
        }
        dataDirectory.mkdirs();
    }

    @Override
    protected void doSetUp() throws Exception {
        super.doSetUp();
        startServer();
        putFilesLatch = new CountDownLatch(1);


        muleContext.registerListener(new EndpointMessageNotificationListener() {
            public void onNotification(final ServerNotification notification) {

                EndpointMessageNotification messageNotification = (EndpointMessageNotification) notification;

                if (messageNotification.getEndpoint().contains("ftp")
                        && "end dispatch".equals(notification.getActionName())) {
                    putFilesLatch.countDown();
                }
            }
        });
    }

    @Override
    protected void doTearDown() throws Exception {
        stopServer();
    }

    @Test
    public void testCanInsertProducts() throws Exception {
        FileUtils.copyFileToDirectory(new File("src/test/resources/product.csv"),new File("./data"));
        assertTrue(putFilesLatch.await(15, TimeUnit.SECONDS));

        assertEquals(2, template.queryForList("SELECT * FROM products").size());
    }

    void startServer() {
        fakeFtpServer = new FakeFtpServer();
        fakeFtpServer.setServerControlPort(9879);
        fakeFtpServer.addUserAccount(new UserAccount("admin", "123456", "/"));

        FileSystem fileSystem = new UnixFakeFileSystem();
        fileSystem.add(new DirectoryEntry("/data/prancingdonkey/catalog"));
        fakeFtpServer.setFileSystem(fileSystem);
        fakeFtpServer.start();

        DataSource dataSource =  muleContext.getRegistry().lookupObject("dataSource");
        template = new JdbcTemplate(dataSource);

        createDatabase();
    }

    void stopServer() {
        fakeFtpServer.stop();
    }

    private void createDatabase() {

        try {
            template.update("DROP TABLE  products");
        } catch (BadSqlGrammarException ex) {
            logger.error(ex);
        }
        template.update("CREATE TABLE products " +
                "(id BIGINT NOT NULL, name VARCHAR(256), acv DOUBLE, cost DOUBLE, description VARCHAR(4096))");

    }
}