Application: gvSIG desktop

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

import java.io.File;

import java.util.Date;

import java.util.List;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataParameters;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.exception.ValidateDataParametersException;

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

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.store.dbf.utils.FieldFormatter;

import static org.gvsig.fmap.dal.store.shp.SHPStoreProvider.GEOMETRY_ATTIBUTE_NAME;

import org.gvsig.fmap.geom.Geometry;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestCreate extends TestCase {

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

    public TestCreate(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.SHAPE_PROVIDER_NAME;

    }

    protected String getTargetFilename() {

        return "testCreateTarget1.shp";

    }

    protected FeatureStore openTargetStore1() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

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

        FeatureStore store;

        try {

            store = (FeatureStore) dataManager.openStore(

                    getProviderName(),

                    "shpFile=",f,

                    DataParameters.CRS_PARAMTER_NAME, "EPSG:4326"

            );

        } catch(ValidateDataParametersException ex) {

            LOGGER.warn(ex.getLocalizedMessageStack());

            throw ex;

        }

        return store;

    }

    protected void createFrom(FeatureStore sourceStore) throws Exception {

        DataServerExplorer explorer = TestUtils.openServerExplorer();

        NewFeatureStoreParameters params = (NewFeatureStoreParameters) explorer.getAddParameters(

                getProviderName()

        );

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

        params.setDynValue(DataParameters.CRS_PARAMTER_NAME, "EPSG:4326");

        params.setDynValue("GeometryType", Geometry.TYPES.POINT);

        EditableFeatureType ft = params.getDefaultFeatureType();

        ft.copyFrom(sourceStore.getDefaultFeatureType());

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

    }

    protected void checkTypes(FeatureType sourceFeatureType) throws Exception {

//        DataType STRING_TYPE = ToolsLocator.getDataTypesManager().get(DataTypes.STRING);

//        DataType INT_TYPE = ToolsLocator.getDataTypesManager().get(DataTypes.INT);

        FeatureStore targetStore = openTargetStore1();

        FeatureType targetFeatureType = targetStore.getDefaultFeatureType();

        assertEquals("Feature type size",sourceFeatureType.size(), targetFeatureType.size());

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

            FeatureAttributeDescriptor sourceAttr = sourceFeatureType.get(i);

            FeatureAttributeDescriptor targetAttr = targetFeatureType.get(i);

            switch(sourceAttr.getType()) {

                case DataTypes.GEOMETRY:

                    if( targetAttr.getName().equalsIgnoreCase(GEOMETRY_ATTIBUTE_NAME) ) {

                        // Si no hay fichero DAL se le habra asignado el nombre por defecto.

                    } else {

                        assertEquals(

                                String.format("Field %s name mismatch", sourceAttr.getName()),

                                sourceAttr.getName(),

                                targetAttr.getName()

                        );

                    }

                    assertEquals(

                            String.format("Field %s type mismatch", sourceAttr.getName()),

                            sourceAttr.getDataTypeName(),

                            targetAttr.getDataTypeName()

                    );

                    assertEquals(

                            String.format("Field %s geometry type mismatch", sourceAttr.getName()),

                            sourceAttr.getGeomType().getName(),

                            targetAttr.getGeomType().getName()

                    );

                    assertEquals(

                            String.format("Field %s geometry SRS mismatch", sourceAttr.getName()),

                            sourceAttr.getSRS().toString(),

                            targetAttr.getSRS().toString()

                    );

                    assertEquals(

                            String.format("Field %s size mismatch", sourceAttr.getName()),

                            sourceAttr.getSize(),

                            targetAttr.getSize()

                    );

                    assertEquals(

                            String.format("Field %s precision mismatch", sourceAttr.getName()),

                            sourceAttr.getPrecision(),

                            targetAttr.getPrecision()

                    );

                    assertEquals(

                            String.format("Field %s scale mismatch", sourceAttr.getName()),

                            sourceAttr.getScale(),

                            targetAttr.getScale()

                    );

                    break;

                default:

                    // Solo comprobams que pasa con las geometrias, los otros

                    // tipos de datos confiamos que los comprueben los test

                    // del dbf.

                    break;

            }

        }

    }

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

        }

    }

    protected void checkData(FeatureStore sourceStore) throws Exception {

        FeatureStore targetStore = openTargetStore1();

        List<Feature> sourceFeatures = sourceStore.getFeatures();

        List<Feature> targetFeatures = targetStore.getFeatures();

        assertEquals("Count features", sourceFeatures.size(), targetFeatures.size());

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

            Feature sourceFeature = sourceFeatures.get(i);

            Feature targetFeature = targetFeatures.get(i);

            for (FeatureAttributeDescriptor sourceAttr : sourceStore.getDefaultFeatureType()) {

                switch(sourceAttr.getType()) {

                    case DataTypes.GEOMETRY:

                        assertEquals(

                                String.format("Feature %03d attribute %s", i, sourceAttr.getName()),

                                sourceFeature.get(sourceAttr.getName()),

                                targetFeature.get(sourceAttr.getName())

                        );

                        break;

                    default:

                      // Solo comprobams que pasa con las geometrias, los otros

                      // tipos de datos confiamos que los comprueben los test

                      // del dbf.

                      break;

                }

            }

        }

    }

    public void testCreateWithDALFile() throws Exception {

        FeatureStore sourceStore = TestUtils.openSourceStore1();

        createFrom(sourceStore);

        checkTypes(sourceStore.getDefaultFeatureType());

        copyFrom(sourceStore, FeatureStore.MODE_APPEND);

        checkData(sourceStore);

        createFrom(sourceStore);

        copyFrom(sourceStore, FeatureStore.MODE_FULLEDIT);

        checkData(sourceStore);

    }

    public void testCreateWithoutDALFile() throws Exception {

        FeatureStore sourceStore = TestUtils.openSourceStore1();

        createFrom(sourceStore);

        TestUtils.removeDALFile(getTargetFilename());

        checkTypes(sourceStore.getDefaultFeatureType());

        copyFrom(sourceStore, FeatureStore.MODE_APPEND);

        TestUtils.removeDALFile(getTargetFilename());

        checkData(sourceStore);

        createFrom(sourceStore);

        TestUtils.removeDALFile(getTargetFilename());

        copyFrom(sourceStore, FeatureStore.MODE_FULLEDIT);

        TestUtils.removeDALFile(getTargetFilename());

        checkData(sourceStore);

    }

}