Application: gvSIG desktop

package org.gvsig.fmap.dal.store.h2.operations.sql;

import java.util.List;

import junit.framework.TestCase;

import org.gvsig.fmap.dal.DataQuery;

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

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

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

import org.gvsig.fmap.dal.store.h2.TestUtils;

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

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

import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;

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

import org.gvsig.fmap.dal.store.jdbc2.spi.operations.ResultSetForSetProviderOperation;

import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestResultSetForSetProvider extends TestCase {

  private static final Logger LOGGER = LoggerFactory.getLogger(TestResultSetForSetProvider.class);

  public TestResultSetForSetProvider(String testName) {

    super(testName);

  }

  @Override

  protected void setUp() throws Exception {

    super.setUp();

    new DefaultLibrariesInitializer().fullInitialize();

  }

  @Override

  protected void tearDown() throws Exception {

    super.tearDown();

  }

  public void testSimple() throws Exception {

    JDBCHelper helper = TestUtils.getJDBCHelper();

    JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

    OperationsFactory operations = helper.getOperations();

    List<String> expectedSQLs = TestUtils.getSQLs("resultSetForSetProvider.sql");

    FeatureStore sourceStore = TestUtils.openSourceStore1();

    TableReference table = operations.createTableReference(

            "dbtest", 

            sqlbuilder.default_schema(), 

            "test", 

            null

    );

    FeatureType featureType = sourceStore.getDefaultFeatureType();

    ResultSetForSetProviderOperation resultSetForSetProvider = operations.createResultSetForSetProvider(

            table,

            null,

            null,

            null,

            featureType,

            featureType,

            0,

            0, 

            0

    );

    String sql = resultSetForSetProvider.getSQL();

    assertEquals("ResultSetForSetProvider SQL", expectedSQLs.get(0), sql);

  }

  public void testSimpleGroup() throws Exception {

    JDBCHelper helper = TestUtils.getJDBCHelper();

    JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

    OperationsFactory operations = helper.getOperations();

    List<String> expectedSQLs = TestUtils.getSQLs("resultSetForSetProvider.sql");

    FeatureStore sourceStore = TestUtils.openSourceStore1();

    TableReference table = operations.createTableReference(

            "dbtest", 

            sqlbuilder.default_schema(), 

            "test", 

            null

    );

    FeatureType featureType = sourceStore.getDefaultFeatureType();

    FeatureQuery query = sourceStore.createFeatureQuery();

    query.getGroupByColumns().add("Long");

    query.getAggregateFunctions().put("ID", "MIN");

    query.getAggregateFunctions().put("Byte", "MAX");

    query.getAggregateFunctions().put("Double", "SUM");

    ResultSetForSetProviderOperation resultSetForSetProvider = operations.createResultSetForSetProvider(

            table,

            null,

            null,

            query,

            featureType,

            featureType,

            0,

            0, 

            0

    );

    String sql = resultSetForSetProvider.getSQL();

    System.out.println("###### SQL:"+sql);

    System.out.println("###### EXP:"+expectedSQLs.get(1));

    assertEquals("ResultSetForSetProvider SQL", expectedSQLs.get(1), sql);

  }

  public void testSubselect() throws Exception {

    JDBCHelper helper = TestUtils.getJDBCHelper();

    JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

    OperationsFactory operations = helper.getOperations();

    List<String> expectedSQLs = TestUtils.getSQLs("resultSetForSetProvider.sql");

    FeatureStore sourceStore = TestUtils.openSourceStore1();

    TableReference table = operations.createTableReference(

            "dbtest", 

            sqlbuilder.default_schema(), 

            "test", 

            null

    );

    StringBuilder filter = new StringBuilder();

    filter.append("EXISTS(");

    filter.append(" SELECT \"ISO_A2\" FROM countries");

    filter.append("   WHERE ");

    filter.append("     TEST.STRING = countries.CONTINENT AND ");

    filter.append("     countries.LASTCENSUS < 0 ");

    filter.append("   LIMIT 1;, ");

    filter.append(" 'EXISTS62a964cd7bc24f409b97c03b9170408d' ");

    filter.append(")");

    FeatureType featureType = sourceStore.getDefaultFeatureType();

    FeatureQuery query = sourceStore.createFeatureQuery();

    query.addFilter(filter.toString());

    ResultSetForSetProviderOperation resultSetForSetProvider = operations.createResultSetForSetProvider(

            table,

            null,

            null,

            query,

            featureType,

            featureType,

            0,

            0, 

            0

    );

    String sql = resultSetForSetProvider.getSQL();

    System.out.println("###### SQL:"+sql);

    System.out.println("###### EXP:"+expectedSQLs.get(2));

    assertEquals("ResultSetForSetProvider SQL", expectedSQLs.get(2), sql);

  }

  // TODO: a?adir un test con where, group y order.

}