Application: gvSIG desktop

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

import java.util.List;

import junit.framework.TestCase;

import org.gvsig.expressionevaluator.ExpressionUtils;

import org.gvsig.fmap.dal.DataTypes;

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

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

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.impl.expressionevaluator.DefaultFeatureAttributeEmulatorExpression;

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

@SuppressWarnings("UseSpecificCatch")

public class TestArena2GroupByForeignValue extends TestCase {

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

  public TestArena2GroupByForeignValue(String testName) {

    super(testName);

  }

  @Override

  protected void setUp() throws Exception {

    super.setUp();

    new DefaultLibrariesInitializer().fullInitialize();

  }

  @Override

  protected void tearDown() throws Exception {

    super.tearDown();

  }

  protected List<String> getExpectedSQLs(String name) throws Exception {

          return TestUtils.getSQLs(name);

  }

    public void testGroupByForeignValue() throws Exception {

        try {

            JDBCHelper helper = TestUtils.createJDBCHelper();

            JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

            OperationsFactory operations = helper.getOperations();

            List<String> expectedSQLs = getExpectedSQLs("usecases/arena2/testGroupByForeignValue.sql");

            FeatureStore arena2_conductores = TestUtils.openStoreTest("usecases/arena2/ARENA2_CONDUCTORES.csv");

            FeatureStore arena2_accidentes = TestUtils.openStoreTest("usecases/arena2/ARENA2_ACCIDENTES.csv");

            FeatureStore arena2_vehiculos = TestUtils.openStoreTest("usecases/arena2/ARENA2_VEHICULOS.csv");

            TableReference table = operations.createTableReference(

                    "dbtest",

                    "public", //sqlbuilder.default_schema(),

                    "ARENA2_CONDUCTORES",

                    null

            );

            FeatureType featureType = arena2_conductores.getDefaultFeatureType();

            EditableFeatureType eFeatureType = featureType.getEditable();

            FeatureQuery query = arena2_conductores.createFeatureQuery();

                        query.setFilter("( ( ( (FOREING_VALUE('ID_ACCIDENTE.TITULARIDAD_VIA') = 2) )) AND \n" +

"( (( (FOREING_VALUE('LID_VEHICULO.TIPO_VEHICULO') = 5) ) OR ( (FOREING_VALUE('LID_VEHICULO.TIPO_VEHICULO') = 6) ) OR ( (FOREING_VALUE('LID_VEHICULO.TIPO_VEHICULO') = 7) )))) AND \n" +

"( (( (FOREING_VALUE('ID_ACCIDENTE.FECHA_ACCIDENTE') >= DATE '2019-01-01') ) AND ( (FOREING_VALUE('ID_ACCIDENTE.FECHA_ACCIDENTE') <= DATE '2019-12-31') )))");

            addExtraColumn(eFeatureType, query, "ano", DataTypes.INTEGER, "EXTRACT(YEAR FROM FOREING_VALUE('ID_ACCIDENTE.FECHA_ACCIDENTE'))");

            addExtraColumn(eFeatureType, query, "provincia", DataTypes.STRING, "FOREING_VALUE('ID_ACCIDENTE.COD_PROVINCIA')");

            addExtraColumn(eFeatureType, query, "lesividad", DataTypes.STRING, "CASE\n" +

"WHEN ASISTENCIA_SANITARIA=1 THEN 'Muerto'\n" +

"WHEN ASISTENCIA_SANITARIA=2 THEN 'Grave'\n" +

"WHEN ASISTENCIA_SANITARIA=3 or ASISTENCIA_SANITARIA=4 or ASISTENCIA_SANITARIA=5 or ASISTENCIA_SANITARIA=6 or ASISTENCIA_SANITARIA=7 THEN 'Leve'\n" +

"WHEN ASISTENCIA_SANITARIA=8 THEN 'Ileso'\n" +

"ELSE 'Se Desconoce'\n" +

"END CASE");

            addExtraColumn(eFeatureType, query, "Acc", DataTypes.INTEGER, "1");

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

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

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

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

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

                        CountOperation count = operations.createCount(eFeatureType, table, null, query);

                        String sqlcount = count.getSQL();

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

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

            assertEquals("Count SQL", expectedSQLs.get(1), sqlcount);

            ResultSetForSetProviderOperation resultSetForSetProvider = operations.createResultSetForSetProvider(

                    table,

                    null,

                    null,

                    query,

                    eFeatureType,

                    eFeatureType,

                    0,

                    0,

                    0

            );

            String sql = resultSetForSetProvider.getSQL();

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

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

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

        } catch (Throwable th) {

                        th.printStackTrace();

            throw th;

        }

    }

        private void addExtraColumn(EditableFeatureType eFeatureType, FeatureQuery query, String name, int type, String exp) {

                EditableFeatureAttributeDescriptor extraColumn = query.getExtraColumn().add(name, type);

                extraColumn.setFeatureAttributeEmulator(new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression(exp)));

        }

}