Application: gvSIG desktop

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

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.jdbc2.AbstractTestUtils;

import org.gvsig.fmap.dal.store.jdbc2.AbstractTestUtils.Expecteds;

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.tools.library.impl.DefaultLibrariesInitializer;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("UseSpecificCatch")

public abstract class AbstractTestCount extends TestCase {

    protected static final Logger LOGGER = LoggerFactory.getLogger(AbstractTestCount.class);

    protected AbstractTestUtils utils;

    public AbstractTestCount(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    public AbstractTestUtils utils() {

        if (this.utils == null) {

            this.utils = this.createUtils();

        }

        return this.utils;

    }

    protected abstract AbstractTestUtils createUtils();

    public void testCount() throws Exception {

        try {

            JDBCHelper helper = utils().createJDBCHelper();

            JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

            OperationsFactory operations = helper.getOperations();

            Expecteds expectedSQLs = utils().getExpecteds("count.txt");

            FeatureStore sourceStore = utils().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();

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

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

            assertEquals("Count SQL", expectedSQLs.get("count"), sql);

        } catch (Throwable th) {

            LOGGER.warn("", th);

            throw th;

        }

    }

    public void testSimpleGroup() throws Exception {

        try {

            JDBCHelper helper = utils().createJDBCHelper();

            JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

            OperationsFactory operations = helper.getOperations();

            Expecteds expectedSQLs = utils().getExpecteds("count.txt");

            FeatureStore sourceStore = utils().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("SimpleGroup")+"###");

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

        } catch (Throwable th) {

            LOGGER.warn("", th);

            throw th;

        }

    }

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

}