Application: gvSIG desktop

package org.gvsig.sqlite.dal.operations.usecases.arena2;

import java.util.List;

import junit.framework.TestCase;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.DatabaseWorkspaceManager;

import org.gvsig.fmap.dal.SQLBuilder;

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

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

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

import org.gvsig.sqlite.dal.TestUtils;

import org.gvsig.sqlite.dal.TestUtilsSQLite;

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 AbstractTestUtils utils;

    public AbstractTestUtils utils() {

        if (this.utils == null) {

            this.utils = this.createUtils();

        }

        return this.utils;

    }

    protected AbstractTestUtils createUtils() {

        return new TestUtilsSQLite();

    }

    public void testGroupByForeignValue() throws Exception {

        try {

            JDBCHelper helper = TestUtils.createJDBCHelper();

            OperationsFactory operations = helper.getOperations();

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

            DatabaseWorkspaceManager ws = TestArena2Utils.initDatabase(utils(), "testGroupByForeignValue");

            SQLBuilder sqlbuilder = TestArena2Utils.createSQLBuilder(ws);

            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",

                    sqlbuilder.default_schema(),

                    "ARENA2_CONDUCTORES",

                    null

            );

            FeatureType featureType = arena2_conductores.getDefaultFeatureType();

            EditableFeatureType eFeatureType = featureType.getEditable();

            FeatureQuery query = arena2_conductores.createFeatureQuery();

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

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

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

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

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

            utils().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");

            utils().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("###### testGroupByForeignValue(1): Count");

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

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

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

            TestArena2Utils.runSQLToCheckSyntax(utils(), "testGroupByForeignValue", sqlcount);

            ResultSetForSetProviderOperation resultSetForSetProvider = operations.createResultSetForSetProvider(

                    table,

                    null,

                    null,

                    query,

                    eFeatureType,

                    eFeatureType,

                    0,

                    0,

                    0

            );

            String sql = resultSetForSetProvider.getSQL();

            System.out.println("###### testGroupByForeignValue(0)");

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

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

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

            TestArena2Utils.runSQLToCheckSyntax(ws, "testGroupByForeignValue", sql);

        } catch (Throwable th) {

            th.printStackTrace();

            throw th;

        }

    }

}