Application: gvSIG desktop » gvSIG data provider for PostgreSQL

package org.gvsig.postgresql.dal;

import java.io.File;

import java.net.URL;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.impl.DefaultFeature;

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.spi.FakeConnectionParameters;

import org.gvsig.fmap.dal.store.jdbc2.spi.FakeConnectionProvider;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestUtils  {

    public static final Logger LOGGER = LoggerFactory.getLogger(TestUtils.class);

    public static final String PROVIDER_NAME = PostgreSQLLibrary.NAME;

    public static File getTargetFolder() throws Exception {

        URL url = TestUtils.class.getResource("/");

        File x = new File(url.toURI());

        File target = x.getParentFile();

        return target;

    }

    public static File getResource(String name) throws Exception {

        File x = new File(getTargetFolder(), name);

        return x;

    }

    public static File getResourceAsFile(String pathname) throws Exception {

        URL url = TestUtils.class.getResource(pathname);

        File x = new File(url.toURI());

        return x;

    }

    public static FeatureStore openSourceStore1() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        File f = getResourceAsFile("/org/gvsig/postgresql/dal/testCreateSource1.csv");

        FeatureStore store = (FeatureStore) dataManager.openStore(

                DataStore.CSV_PROVIDER_NAME, 

                "file=",f,

                "automaticTypesDetection=", false,

                "locale=","en"

        );

        return store;

    }

    public static List<String> getSQLs(String name) throws Exception {

      File f = getResourceAsFile("/org/gvsig/postgresql/dal/"+name);

      List<String> SQLs = new ArrayList<>();

      List<String> lines = FileUtils.readLines(f);

      StringBuilder sb = new StringBuilder();

      for (String line : lines) {

        line = StringUtils.stripStart(line, null);

        if( line.startsWith("--") ) {

          continue;

        }

        if( line.endsWith(";") ) {

          sb.append(line.substring(0, line.length()-1));

          SQLs.add(sb.toString());

          sb.setLength(0);

        } else {

          sb.append(line);

        }

      }

      return SQLs;

    }

    public static JDBCHelper getJDBCHelper() {

      JDBCHelper helper = new PostgreSQLHelper(new FakePostgreSQLConnectionParameters(), new FakeConnectionProvider());

      return helper;

    }

    public static FeatureProvider getFeatureProvider(Feature feature) {

      return ((DefaultFeature)feature).getData();

    }

}