org.geoserver.jdbcconfig.internal.JdbcConfigTestSupport.java Source code

Introduction

Here is the source code for org.geoserver.jdbcconfig.internal.JdbcConfigTestSupport.java

Source

/* Copyright (c) 2012 TOPP - www.openplans.org. All rights reserved.
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.jdbcconfig.internal;

import static org.mockito.Matchers.anyObject;
import static org.mockito.Mockito.when;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.nio.charset.Charset;
import java.sql.Connection;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.logging.Level;

import javax.sql.DataSource;

import org.apache.commons.dbcp.BasicDataSource;
import org.apache.commons.io.FileUtils;
import org.geoserver.catalog.impl.CatalogImpl;
import org.geoserver.config.util.XStreamPersisterFactory;
import org.geoserver.jdbcconfig.JDBCGeoServerLoader;
import org.geoserver.platform.GeoServerExtensions;
import org.mockito.Mockito;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.JdbcOperations;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.web.context.WebApplicationContext;

import com.google.common.collect.Lists;
import com.google.common.io.Resources;

public class JdbcConfigTestSupport {

    // String driver = "org.postgresql.Driver";
    //
    // String connectionUrl = "jdbc:postgresql://localhost:5432/test";
    //
    // String initScriptName = "initdb.postgres.sql";
    //
    // String dropScriptName = "dropdb.postgres.sql";

    String driver = "org.h2.Driver";

    String connectionUrl = "jdbc:h2:file:target/jdbcconfig/test";

    String initScriptName = "initdb.h2.sql";

    String dropScriptName = "dropdb.h2.sql";

    private CatalogImpl catalog;

    private BasicDataSource dataSource;

    private ConfigDatabase configDb;

    public void setUp() throws Exception {
        ConfigDatabase.LOGGER.setLevel(Level.FINER);
        // just to avoid hundreds of warnings in the logs about extension lookups with no app
        // context set
        WebApplicationContext applicationContext = Mockito.mock(WebApplicationContext.class);
        new GeoServerExtensions().setApplicationContext(applicationContext);
        when(applicationContext.getBeansOfType((Class) anyObject())).thenReturn(Collections.EMPTY_MAP);
        when(applicationContext.getBeanNamesForType((Class) anyObject())).thenReturn(new String[] {});
        //

        final File testDbDir = new File("target", "jdbcconfig");
        FileUtils.deleteDirectory(testDbDir);
        testDbDir.mkdirs();

        dataSource = new BasicDataSource();
        dataSource.setDriverClassName(driver);
        dataSource.setUrl(connectionUrl);
        dataSource.setUsername("postgres");
        dataSource.setPassword("geo123");

        dataSource.setMinIdle(3);
        dataSource.setMaxActive(10);
        try {
            Connection connection = dataSource.getConnection();
            connection.close();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }

        try {
            dropDb(dataSource);
        } catch (Exception ignored) {
        }
        initDb(dataSource);

        XStreamInfoSerialBinding binding = new XStreamInfoSerialBinding(new XStreamPersisterFactory());

        catalog = new CatalogImpl();
        configDb = new ConfigDatabase(dataSource, binding);
        configDb.setCatalog(catalog);
        configDb.initDb(null);
    }

    public void tearDown() throws Exception {
        if (dataSource != null) {
            dropDb(dataSource);
        }
        try {
            if (configDb != null) {
                configDb.dispose();
            }
        } finally {
            if (dataSource != null) {
                dataSource.close();
            }
        }
    }

    public ConfigDatabase getDatabase() {
        return configDb;
    }

    private void initDb(DataSource dataSource) throws Exception {
        runScript(initScriptName);
    }

    private void dropDb(DataSource dataSource) throws Exception {
        runScript(dropScriptName);
    }

    private void runScript(String dbScriptName) throws IOException {
        NamedParameterJdbcTemplate template = new NamedParameterJdbcTemplate(dataSource);

        String[] initScript = readDbScript(dbScriptName);
        JdbcOperations jdbcOperations = template.getJdbcOperations();
        for (String sentence : initScript) {
            try {
                jdbcOperations.update(sentence);
            } catch (DataAccessException e) {
                // e.printStackTrace();
                throw e;
            }
        }
    }

    private String[] readDbScript(final String scriptName) throws IOException {

        URL url = JDBCGeoServerLoader.class.getResource(scriptName);
        if (url == null) {
            throw new IllegalArgumentException("Script not found: " + getClass().getName() + "/" + scriptName);
        }
        List<String> lines = Lists.newArrayList(Resources.readLines(url, Charset.forName("UTF-8")));
        for (Iterator<String> it = lines.iterator(); it.hasNext();) {
            if (it.next().trim().length() == 0) {
                it.remove();
            }
        }
        return lines.toArray(new String[lines.size()]);
    }

    public DataSource getDataSource() {
        return this.dataSource;
    }

    public DbMappings getDbMappings() {
        return this.configDb.getDbMappings();
    }

    public CatalogImpl getCatalog() {
        return catalog;
    }
}