Application: gvSIG desktop

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

import java.util.List;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

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.mdb.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.CountOperation;

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 TestCount extends TestCase {

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

  public TestCount(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 testCount() throws Exception {

    JDBCHelper helper = TestUtils.getJDBCHelper();

    JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

    OperationsFactory operations = helper.getOperations();

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

    FeatureStore sourceStore = TestUtils.openSourceStore1();

    TableReference table = operations.createTableReference(

            "dbtest", 

            sqlbuilder.default_schema(), 

            "test", 

            null

    );

    FeatureType featureType = sourceStore.getDefaultFeatureType();

    CountOperation count = operations.createCount(

            featureType,

            table, 

            null, 

            null

    );

    String sql = count.getSQL();

    assertEquals("Count 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("count.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");

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

    CountOperation count = operations.createCount(

            featureType,

            table, 

            null, 

            query

    );

    String sql = count.getSQL();

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

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

    assertEquals("Count with group SQL", expectedSQLs.get(1), sql);

  }

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

}