Application: gvSIG desktop

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

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

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

    public TestArena2OrderByExtraValue(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 testOrderByExtraValue() throws Exception {

        try {

            JDBCHelper helper = TestUtils.createJDBCHelper();

            JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

            OperationsFactory operations = helper.getOperations();

            List<String> expectedSQLs = getExpectedSQLs("usecases/arena2/testOrderByExtraValue.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("ano");

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

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

            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,

                    15,

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

    }

}