Application: gvSIG desktop

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

import junit.framework.TestCase;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataTypes;

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

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

import org.gvsig.fmap.dal.store.jdbc.JDBCNewStoreParameters;

import org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters;

import org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryUtils;

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

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

/**

 *

 * @author jjdelcerro

 */

public class STAsBinaryTest extends TestCase {

    private static final String TABLE_EMPTY_WITH_GEOM = "TABLE_EMPTY_WITH_GEOM";

    private static final String TABLE_WITH_NULL_GEOMS = "TABLE_WITH_NULL_GEOMS";

    public static final String DBNAME = "STAsBinaryTest";

    public STAsBinaryTest(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    private void createTableWithNullGeoms(JDBCServerExplorer explorer, String tableName) throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        JDBCNewStoreParameters create_params = explorer.getAddParameters();

        create_params.setTable(tableName);

        EditableFeatureType ft = create_params.getDefaultFeatureType();

        ft.add("ID", DataTypes.INT).setIsPrimaryKey(true);

        ft.add("ISNULL", DataTypes.BOOLEAN);

        ft.add("GEOM", DataTypes.GEOMETRY)

                .setGeometryType(Geometry.TYPES.POINT, Geometry.SUBTYPES.GEOM2D).

                setSRS("EPSG:4326");

        explorer.add(TestUtils.PROVIDER_NAME, create_params, false);

        JDBCStoreParameters open_params = explorer.getOpenParameters();

        open_params.setTable(tableName);

        FeatureStore store = (FeatureStore) dataManager.openStore(TestUtils.PROVIDER_NAME, open_params);

        store.edit(FeatureStore.MODE_FULLEDIT);

        EditableFeature f;

        int id=1;

        f = store.createNewFeature();

        f.set("ID", id++);

        f.set("ISNULL", false);

        f.set("GEOM", GeometryUtils.createPoint(10, 10));

        store.insert(f);

        f = store.createNewFeature();

        f.set("ID", id++);

        f.set("ISNULL", true);

        f.set("GEOM", null);

        store.insert(f);

        f = store.createNewFeature();

        f.set("ID", id++);

        f.set("ISNULL", true);

        f.set("GEOM", null);

        store.insert(f);

        f = store.createNewFeature();

        f.set("ID", id++);

        f.set("ISNULL", false);

        f.set("GEOM", GeometryUtils.createPoint(20, 20));

        store.insert(f);

        store.finishEditing();

    }

    private void createEmptyTable(JDBCServerExplorer explorer, String tableName) throws Exception {

        JDBCNewStoreParameters params = explorer.getAddParameters();

        params.setTable(tableName);

        EditableFeatureType ft = params.getDefaultFeatureType();

        ft.add("ID", DataTypes.INT).setIsPrimaryKey(true);

        ft.add("ISNULL", DataTypes.BOOLEAN);

        ft.add("GEOM", DataTypes.GEOMETRY)

                .setGeometryType(Geometry.TYPES.POINT, Geometry.SUBTYPES.GEOM2D).

                setSRS("EPSG:4326");

        explorer.add(TestUtils.PROVIDER_NAME, params, false);

    }

    public void testGetEnvelopeOfEmptyTable() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);

        createEmptyTable(explorer, TABLE_EMPTY_WITH_GEOM);

        JDBCStoreParameters params = explorer.getOpenParameters();

        params.setTable(TABLE_EMPTY_WITH_GEOM);

        FeatureStore store = (FeatureStore) dataManager.openStore(TestUtils.PROVIDER_NAME, params);

        Envelope env = store.getEnvelope();

        assertNull(env);

    }

    public void testGetEnvelopeOfTableWithNullGeoms() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);

        createTableWithNullGeoms(explorer, TABLE_WITH_NULL_GEOMS);

        JDBCStoreParameters params = explorer.getOpenParameters();

        params.setTable(TABLE_WITH_NULL_GEOMS);

        FeatureStore store = (FeatureStore) dataManager.openStore(TestUtils.PROVIDER_NAME, params);

        Envelope env = store.getEnvelope();

        assertNotNull(env);

        assertEquals("POLYGON ((10.0 10.0, 10.0 20.0, 20.0 20.0, 20.0 10.0, 10.0 10.0))", env.toString());

    }

    public void testIterateOverTableWithNullGeoms() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);

        createTableWithNullGeoms(explorer, TABLE_WITH_NULL_GEOMS);

        JDBCStoreParameters params = explorer.getOpenParameters();

        params.setTable(TABLE_WITH_NULL_GEOMS);

        FeatureStore store = (FeatureStore) dataManager.openStore(TestUtils.PROVIDER_NAME, params);

        FeatureSet set = store.getFeatureSet();

        int sz = (int) set.getSize();

//        System.out.println("#### count: "+sz);

        for (Feature f : set ) {

//            System.out.println("#### feature: "+f.toString());

            if( !f.getBoolean("ISNULL") ) {

                assertNotNull(f.getGeometry("GEOM"));

            }

        }

        assertEquals(4, sz);

    }

}