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

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

    protected AbstractTestUtils utils;

    public AbstractTestWhereWithComputedField(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() {}

    protected void testWhereWithComputedField() throws Exception {

        try {

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

                return;

            }

            FeatureStore sourceStore = utils().openSourceStore2();

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

            utils().initWorkspace("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> features1 = dbstore.getFeatures("Compu1>8");

            System.out.println("## in output");

            assertEquals("Features 1 size:", 5, features1.size());

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

                Feature feature = features1.get(i);

                System.out.println("field Compu1: "+feature.format("Compu1") + " feature "+i+": "+feature.toString());

            }

            List<Feature> features2 = dbstore.getFeatures("Compu1<=8");

            System.out.println("## in output");

            assertEquals("Features 2 size:", 5, features2.size());

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

                Feature feature = features2.get(i);

                System.out.println("field Compu1: "+feature.format("Compu1") + " feature "+i+": "+feature.toString());

            }

            System.out.println("## end output");

            DisposeUtils.dispose(dbstore);

        } catch (Throwable th) {

            LOGGER.warn("", th);

            throw th;

        }

    }

}