Application: gvSIG desktop

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

import java.io.File;

import junit.framework.TestCase;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataStore;

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

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

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

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

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

public class TestExtraColumn extends TestCase {

    public TestExtraColumn(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

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

    // public void testHello() {}

    protected String getProviderName() {

        return DataStore.DBASE_PROVIDER_NAME;

    }

    protected String getTargetFilename() {

        return "testCreateTarget1.dbf";

    }

    protected FeatureStore openTargetStore1() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        File f = TestUtils.getResource(getTargetFilename());

        FeatureStore store = (FeatureStore) dataManager.openStore(

                getProviderName(), 

                "DbfFile",f

        );

        return store;

    }

    protected void createFrom(FeatureStore sourceStore) throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        DataServerExplorer explorer = dataManager.openServerExplorer(

                DataServerExplorer.FILESYSTEM_SERVER_EXPLORER_NAME, 

                "root", TestUtils.getTargetFolder()

        );

        NewFeatureStoreParameters params = (NewFeatureStoreParameters) explorer.getAddParameters(

                getProviderName()

        );

        ((FilesystemStoreParameters)params).setFile(TestUtils.getResource(getTargetFilename()));

        EditableFeatureType ft = params.getDefaultFeatureType();

        ft.addAll(sourceStore.getDefaultFeatureType());

        explorer.add(getProviderName(), params, true);

    }

    protected void copyFrom(FeatureStore sourceStore, int mode) throws Exception {

        FeatureStore targetStore = openTargetStore1();

        targetStore.edit(mode);

        try {

            for (Feature sourceFeature : sourceStore.getFeatureSet()) {

                EditableFeature targetFeature = targetStore.createNewFeature(sourceFeature);

                targetStore.insert(targetFeature);

            }

        } finally {

            targetStore.finishEditing();

        }

    }

    public void testColumnExtra1() throws Exception {

        FeatureStore sourceStore = TestUtils.openSourceStore1();

        createFrom(sourceStore);        

        copyFrom(sourceStore, FeatureStore.MODE_APPEND);

        FeatureStore store = openTargetStore1();

        FeatureQuery query = store.createFeatureQuery();

        FeatureExtraColumn extra = query.getExtraColumn();

        EditableFeatureAttributeDescriptor column = extra.add("test1");

        column.setFeatureAttributeEmulator("22.2");

        Feature f = store.findFirst(query);

        EditableFeatureAttributeDescriptor column2 = f.getType().getExtraColumn().get("test1");

        assertTrue(column2!=null);

        Double value = (Double) f.getExtraValue("test1");

        assertEquals((Double)22.2, value);        

    }

}