Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

/*

 * AUTHORS (In addition to CIT):

 * 2009 {Iver T.I.}   {Task}

 */

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

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryUtils;

import org.gvsig.fmap.geom.exception.CreateGeometryException;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * @author <a href="mailto:jpiera@gvsig.org">Jorge Piera</a>

 */

public class MemoryStoreProviderTest extends AbstractLibraryAutoInitTestCase {

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

    protected static final String ID = "id";

    protected static final String GEOM = "geom";

    protected static final String IDSYM = "idsym";

    protected static final String LABEL = "label";

    protected static final String FEATUREID = "featureid";

    @Override

    protected void doSetUp() throws Exception {

    }

    public OpenFeatureStoreParameters createMemoryStoreParameters() throws DataException {

        DataManager dataManager = DALLocator.getDataManager();

        MemoryStoreParameters memoryParameters = (MemoryStoreParameters) dataManager

                .createStoreParameters(MemoryStoreProvider.NAME);

        return memoryParameters;

    }

    protected FeatureStore createStoreWithGeomField() throws ValidateDataParametersException, InitializeException, ProviderNotRegisteredException, DataException {

        DataManager dataManager = DALLocator.getDataManager();

        DataStoreParameters params = this.createMemoryStoreParameters();

        FeatureStore store = (FeatureStore) dataManager.openStore(

                params.getProviderName(),

                params

        );

        store.edit();

        EditableFeatureType editableFeatureType = store.getDefaultFeatureType().getEditable();

        editableFeatureType.add(ID, DataTypes.OBJECT)

                .setIsPrimaryKey(true)

                .setObjectClass(Object.class);

        editableFeatureType.add(IDSYM, DataTypes.OBJECT)

                .setObjectClass(Object.class);

        editableFeatureType.add(GEOM, DataTypes.GEOMETRY)

                .setGeometryType(Geometry.TYPES.GEOMETRY, Geometry.SUBTYPES.GEOM2D);

        editableFeatureType.add(LABEL, DataTypes.STRING);

        editableFeatureType.add(FEATUREID, DataTypes.OBJECT)

                .setObjectClass(Object.class);

        store.update(editableFeatureType);

        store.finishEditing();

        return store;

    }

    protected FeatureStore createStoreWithoutGeomField() throws ValidateDataParametersException, InitializeException, ProviderNotRegisteredException, DataException {

        DataManager dataManager = DALLocator.getDataManager();

        DataStoreParameters params = this.createMemoryStoreParameters();

        FeatureStore store = (FeatureStore) dataManager.openStore(

                params.getProviderName(),

                params

        );

        store.edit();

        EditableFeatureType editableFeatureType = store.getDefaultFeatureType().getEditable();

        editableFeatureType.add(ID, DataTypes.OBJECT)

                .setIsPrimaryKey(true)

                .setObjectClass(Object.class);

        editableFeatureType.add(IDSYM, DataTypes.OBJECT)

                .setObjectClass(Object.class);

        editableFeatureType.add(LABEL, DataTypes.STRING);

        editableFeatureType.add(FEATUREID, DataTypes.OBJECT)

                .setObjectClass(Object.class);

        store.update(editableFeatureType);

        store.finishEditing();

        return store;

    }

    protected FeatureStore createStoreWithGeomFieldFilled(List<Geometry> geoms) throws ValidateDataParametersException, InitializeException, ProviderNotRegisteredException, DataException {

        FeatureStore store = createStoreWithGeomField();

        store.edit();

        int id = -1;

        geoms.add(GeometryUtils.createPoint(1, 1));

        geoms.add(GeometryUtils.createLine(Geometry.SUBTYPES.GEOM2D)

            .addVertex(0, 0)

            .addVertex(1, 0)

            .addVertex(1, 1)

            .addVertex(0, 1)

            .addVertex(0, 0));

        geoms.add(GeometryUtils.createPolygon(Geometry.SUBTYPES.GEOM2D)

            .addVertex(0, 0)

            .addVertex(1, 0)

            .addVertex(1, 1)

            .addVertex(0, 1)

            .addVertex(0, 0));

        EditableFeature feature;

        feature = store.createNewFeature();

        feature.set(ID, ++id);

        feature.set(LABEL, "Hi");

        feature.set(GEOM, geoms.get(id));

        store.insert(feature);

        feature = store.createNewFeature();

        feature.set(ID, ++id);

        feature.set(LABEL, "Hi");

        feature.set(GEOM, geoms.get(id));

        store.insert(feature);

        feature = store.createNewFeature();

        feature.set(ID, ++id);

        feature.set(LABEL, "Hi");

        feature.set(GEOM, geoms.get(id));

        store.insert(feature);

        Envelope envelope = store.getEnvelope();

        assertNotNull(envelope);

        store.finishEditing();

        return store;

    }

    protected FeatureStore createStoreWithoutGeomFieldFilled() throws ValidateDataParametersException, InitializeException, ProviderNotRegisteredException, DataException {

        FeatureStore store = createStoreWithoutGeomField();

        store.edit();

        int id = 0;

        EditableFeature feature;

        feature = store.createNewFeature();

        feature.set(ID, ++id);

        feature.set(LABEL, "Hi");

        store.insert(feature);

        feature = store.createNewFeature();

        feature.set(ID, ++id);

        feature.set(LABEL, "Hi");

        store.insert(feature);

        feature = store.createNewFeature();

        feature.set(ID, ++id);

        feature.set(LABEL, "Hi");

        store.insert(feature);

        Envelope envelope = store.getEnvelope();

        assertNull(envelope);

        store.finishEditing();

        return store;

    }

    public void testInitializeStore() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        DataStoreParameters params = this.createMemoryStoreParameters();

        FeatureStore store = (FeatureStore) dataManager.openStore(

                params.getProviderName(),

                params

        );

        assertNotNull(store.getMetadataID());

        assertNotNull(store.getName());

        assertEquals(store.getEnvelope(), store.getDynValue("Envelope"));

        assertTrue(store.getFeatureCount() == 0);

        if (store.isLocksSupported()) {

            assertNotNull(store.getLocks());

        } else {

            assertNull(store.getLocks());

        }

        store.dispose();

    }

    public void testAddFeatureType() throws DataException, ValidateDataParametersException {

        FeatureStore store = createStoreWithGeomField();

        FeatureAttributeDescriptor attributeDescriptor = store.getDefaultFeatureType()

                .getAttributeDescriptor(ID);

        assertNotNull(attributeDescriptor);

        assertEquals(attributeDescriptor.getType(), DataTypes.OBJECT);

        store.dispose();

    }

    public void testAddFeaturesWithGeom() throws DataException, ValidateDataParametersException, CreateGeometryException {

        List<Geometry> geoms = new ArrayList<>();

        FeatureStore store = createStoreWithGeomFieldFilled(geoms);

        int n = -1;

        for (Feature feature : store.getFeatureSet()) {

            assertEquals(++n, feature.get(ID));

            assertEquals(geoms.get(n), feature.get(GEOM));

        }

        store.edit();

        EditableFeature feature = store.createNewFeature();

        feature.set(ID, 4);

        feature.set(GEOM, GeometryUtils.createPolygon(Geometry.SUBTYPES.GEOM2D)

            .addVertex(0, 0)

            .addVertex(1, 0)

            .addVertex(1, 1)

            .addVertex(0, 1)

            .addVertex(0, 0)

        );

        store.insert(feature);

        store.finishEditing();

        assertEquals(4, store.getFeatureCount());

        store.dispose();

    }

    public void testRemoveFeaturesWithGeom() throws DataException, ValidateDataParametersException, CreateGeometryException {

        List<Geometry> geoms = new ArrayList<>();

        FeatureStore store = createStoreWithGeomFieldFilled(geoms);

        Feature f;

        store.edit();

        FeatureSet featureSet = store.getFeatureSet();

        DisposableIterator it = featureSet.fastIterator();

        it.next();

        f = (Feature) it.next();

        store.delete(f);

        store.finishEditing();

        assertEquals(2, store.getFeatureCount());

        store.dispose();

    }

    public void testUpdateFeaturesWithGeom() throws DataException, ValidateDataParametersException, CreateGeometryException {

        List<Geometry> geoms = new ArrayList<>();

        FeatureStore store = createStoreWithGeomFieldFilled(geoms);

        store.edit();

        FeatureSet set = store.getFeatureSet();

        for (Feature feature : set) {

            EditableFeature ef = feature.getEditable();

            ef.set(LABEL, "Bye");

            set.update(ef);

        }

        store.finishEditing();

        for (Feature feature : store.getFeatureSet()) {

            assertEquals("Bye", feature.get(LABEL));

        }

        store.dispose();

    }

    public void testAddFeaturesWithoutGeom() throws DataException, ValidateDataParametersException, CreateGeometryException {

        FeatureStore store = createStoreWithoutGeomFieldFilled();

        int n = 0;

        for (Feature feature : store.getFeatureSet()) {

            assertEquals(++n, feature.get(ID));

        }

        store.edit();

        EditableFeature feature = store.createNewFeature();

        feature.set(ID, 4);

        store.insert(feature);

        store.finishEditing();

        assertEquals(4, store.getFeatureCount());

        store.dispose();

    }

    public void testRemoveFeaturesWithoutGeom() throws DataException, ValidateDataParametersException, CreateGeometryException {

        FeatureStore store = createStoreWithoutGeomFieldFilled();

        Feature f;

        store.edit();

        FeatureSet featureSet = store.getFeatureSet();

        DisposableIterator it = featureSet.fastIterator();

        it.next();

        f = (Feature) it.next();

        store.delete(f);

        store.finishEditing();

        assertEquals(2, store.getFeatureCount());

        store.dispose();

    }

    public void testUpdateFeaturesWithoutGeom() throws DataException, ValidateDataParametersException, CreateGeometryException {

        FeatureStore store = createStoreWithoutGeomFieldFilled();

        store.edit();

        FeatureSet set = store.getFeatureSet();

        for (Feature feature : set) {

            EditableFeature ef = feature.getEditable();

            ef.set(LABEL, "Bye");

            set.update(ef);

        }

        store.finishEditing();

        for (Feature feature : store.getFeatureSet()) {

            assertEquals("Bye", feature.get(LABEL));

        }

        store.dispose();

    }

}