Application: gvSIG desktop

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

import java.util.ArrayList;

import java.util.List;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

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

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.tools.dispose.DisposeUtils;

import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public abstract class AbstractTestComputedAttributes extends TestCase {

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

    protected AbstractTestUtils utils;

    public AbstractTestComputedAttributes(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();

    // TODO add test methods here. The name must begin with 'test'. For example:

    // public void testHello() {}

    public void testComputed1() throws Exception {

        if( !utils().isTheDatabaseAvailable() ) {

            return;

        }

        FeatureStore sourceStore = utils().openSourceStore2();

        JDBCServerExplorer explorer = utils().openServerExplorer("testCreate");

        utils().info_jdbc(explorer);

        utils().drop_tables(explorer, "testCreateTarget2");

        utils().create_table_from(explorer, "testCreateTarget2", sourceStore);

        utils().insert_into_from(explorer, "testCreateTarget2", sourceStore, FeatureStore.MODE_APPEND);

        FeatureStore dbstore = utils().openStore(explorer,"testCreateTarget2");

        dbstore.edit();

        FeatureType featureType = dbstore.getDefaultFeatureType();

        EditableFeatureType eFeatureType = featureType.getEditable();

        eFeatureType.add("Compu1", 

                DataTypes.INTEGER, 

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("ID*2")

                ));

        eFeatureType.add("Compu2", 

                DataTypes.INTEGER, 

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("Poblacion+10000+Compu1")

                ));

        dbstore.update(eFeatureType);

        dbstore.finishEditing();

        List<Feature> features = dbstore.getFeatures();

        for (int i = 0; i < features.size(); i++) {

            Feature feature = features.get(i);

            assertEquals("Compu1 "+i, feature.getInt("ID") * 2, feature.getInt("Compu1"));

            if(feature.get("Poblacion")==null) {

                assertEquals("Compu2 "+i, null, feature.get("Compu2"));

            } else {

                assertEquals("Compu2 "+i, feature.getInt("Poblacion") + 10000 + feature.getInt("Compu1"), feature.getInt("Compu2"));

            }

        }

        DisposeUtils.dispose(dbstore);

    }

    @SuppressWarnings("UnusedAssignment")

    public void testComputed2() throws Exception {

        if( !utils().isTheDatabaseAvailable() ) {

            return;

        }

        String testTableName = "testComputedAttributes2";

        JDBCServerExplorer explorer = utils().openServerExplorer("computedAttr");

        utils().info_jdbc(explorer);

        utils().drop_tables(explorer, testTableName);

        FeatureStore sourceStore = utils().openSourceStore2();

        utils().create_table_from(explorer, testTableName, sourceStore);

        utils().insert_into_from(explorer, testTableName, sourceStore, FeatureStore.MODE_APPEND);

        FeatureStore dbstore = utils().openStore(explorer,testTableName);

        dbstore.edit();

        FeatureType featureType = dbstore.getDefaultFeatureType();

        EditableFeatureType eFeatureType = featureType.getEditable();

        FeatureQuery query = sourceStore.createFeatureQuery();

        eFeatureType.add("CompuID", 

                DataTypes.INTEGER, 

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("MOD(ID,10)")

                ));

        eFeatureType.add("CompuPob", 

                DataTypes.INTEGER, 

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("Poblacion+1")

                ));

        eFeatureType.add("CompuProv", 

                DataTypes.STRING, 

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("UPPER(Provincia)")

                ));

        EditableFeatureAttributeDescriptor extraColumn1 = 

                query.getExtraColumn().add("ExtraID900", DataTypes.INTEGER);

        EditableFeatureAttributeDescriptor extraColumn2 = 

                query.getExtraColumn().add("ExtraPobDen", DataTypes.INTEGER);

        EditableFeatureAttributeDescriptor extraColumn3 = 

                query.getExtraColumn().add("ExtraAno", DataTypes.INTEGER);

        extraColumn1.setFeatureAttributeEmulator(

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("CompuID+900")));

        extraColumn2.setFeatureAttributeEmulator(

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("CompuPob+Densidad")));

        extraColumn3.setFeatureAttributeEmulator(

                new DefaultFeatureAttributeEmulatorExpression(

                        eFeatureType, 

                        ExpressionUtils.createExpression("EXTRACT(YEAR FROM Fecha)")));

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

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

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

        query.getAggregateFunctions().put("ID", "MIN");

        query.getAggregateFunctions().put("Poblacion", "MIN");

        query.getAggregateFunctions().put("CompuID", "MIN");

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

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

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

        dbstore.update(eFeatureType);

        dbstore.finishEditing();

        List<Feature> features0 = dbstore.getFeatures(query);

        ArrayList<Feature> features = new ArrayList<>();

        System.out.println("ID,Comunidad,Provincia,Ciudad,Poblacion,Densidad,Fecha,CompuID,CompuPob,CompuProv,ExtraID900,ExtraPobDen,ExtraAno");

        for (int i = 0; i < features0.size(); i++) {

            Feature feature = features0.get(i);

            features.add(feature.getCopy());

            System.out.print(feature.toString());

            System.out.print(","+feature.get("ExtraID900"));

            System.out.print(","+feature.get("ExtraPobDen"));

            System.out.println(","+feature.get("ExtraAno"));

        }

        System.out.println("ID,Comunidad,Provincia,Ciudad,Poblacion,Densidad,Fecha,CompuID,CompuPob,CompuProv,ExtraID900,ExtraPobDen,ExtraAno");

        features0 = null;

        String[] header = new String[]{"ID","Comunidad","Provincia","Ciudad","Poblacion","Densidad","Fecha","CompuID","CompuPob","CompuProv","ExtraID900","ExtraPobDen","ExtraAno"};

        ArrayList<Object[]> values = new ArrayList<>();

        values.add(new Object[]{0, null, null, null, null, null, null, 0, null,"",900,null,null});

        values.add(new Object[]{1, "GVA", null, null, 500, null, null, 1, 1502, "VALENCIA",1803,1505,2019});

        values.add(new Object[]{3, "GVA", null, null, 50, null, null, 3, 352, "VALENCIA",1807,359,2020});

        values.add(new Object[]{5, "GVA", null, null, 200, null, null, 5, 201, "ALICANTE",905,206,2019});

        values.add(new Object[]{6, "GVA", null, null, 20, null, null, 6, 422, "ALICANTE",1813,1135,2020});

        values.add(new Object[]{8, "GVA", null, null, 100, null, null, 8, 702, "CASTELLON",1817,719,2019});

        for (int i = 0; i < features.size(); i++) {

                for (int j = 0; j < header.length; j++) {

                        assertEquals("feature["+i+"]["+header[j]+"]:", values.get(i)[j],features.get(i).get(header[j]));

                }

        }

        DisposeUtils.dispose(dbstore);

    }

}