Application: gvSIG desktop

package org.gvsig.sqlite.dal;

import java.io.File;

import java.net.URL;

import junit.framework.TestCase;

import org.apache.commons.io.IOUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

import org.gvsig.fmap.geom.GeometryUtils;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataType;

import org.gvsig.tools.dataTypes.DataTypeUtils;

import org.gvsig.tools.dataTypes.DataTypesManager;

import org.gvsig.tools.dispose.DisposeUtils;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestRead extends TestCase {

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

    public static final String DBNAME = "rte";

    public TestRead(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    private TestUtilsSQLite utils = null;

    public TestUtilsSQLite utils() {

        if( this.utils == null ) {

            this.utils = new TestUtilsSQLite();

        }

        return this.utils;

    }

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

    // public void testHello() {}

    protected FeatureStore openStore1(JDBCServerExplorer explorer) throws Exception {

        JDBCStoreParameters params = explorer.get("cats_feature");

        DataManager dataManager = DALLocator.getDataManager();

        FeatureStore store;

        try {

            store = (FeatureStore) dataManager.openStore(

                    utils().getProviderName(), 

                    params

            );

        } catch(ValidateDataParametersException ex) {

            LOGGER.warn(ex.getLocalizedMessageStack());

            throw ex;

        }

        return store;

    }

    public void testReadBasic1() throws Exception {

        try {

            String dbfilename = DBNAME+".gpkg";

            URL url = this.getClass().getResource(dbfilename);

            File tmp = new File( utils().getTargetFolder(), dbfilename);

            IOUtils.copy(url, tmp);

            JDBCServerExplorer explorer = utils().openServerExplorer(tmp);

            FeatureStore store1 = openStore1(explorer);

            FeatureType ft = store1.getDefaultFeatureType();

            Object[][] expecteds = new Object[][] {

              new Object[] { 0, "id", DataTypes.INTEGER},  

              new Object[] { 1, "geometry", DataTypes.GEOMETRY},  

              new Object[] { 2, "text_attribute", DataTypes.STRING},  

              new Object[] { 3, "real_attribute", DataTypes.DOUBLE},  

              new Object[] { 4, "relation", DataTypes.STRING}

            };

            assertEquals("Field count mismatch", expecteds.length, ft.size());

            for (Object[] expected : expecteds) {

                int index = (int)expected[0];

                String fieldname = (String) expected[1];

                FeatureAttributeDescriptor attr = ft.getAttributeDescriptor(index);

                assertEquals("field "+fieldname+" name mismatch", expected[1], attr.getName());

                assertEquals("field "+fieldname+" type mismatch", getTypeId(((Number)expected[2]).intValue()), getTypeId(attr.getType()));

            }

            expecteds = new Object[][] {

              new Object[] { 1, GeometryUtils.createFrom("0000000001c0110062413529f34044147906c0d66f"), "some cat", null, null },  

              new Object[] { 2, GeometryUtils.createFrom("0000000001401e9a4a755fb1e840478f5c6abe951a"), "other cat", null, null },  

              new Object[] { 3, GeometryUtils.createFrom("000000000140326b151bf99810404a77bd6b72d639"), "third cat", null, null },  

              new Object[] { 4, GeometryUtils.createFrom("000000000140338d3dcafd236840472bec6d188172"), "last cat", null, null },  

            };

            FeatureSet set = store1.getFeatureSet();

            assertEquals("Numero de lineas incorrecto", expecteds.length, set.size());

            int n = 0;

            for (Feature feature : set) {

                checkEquals("feature "+n+" id mismatch", feature, "id", expecteds[n][0]);

                checkEquals("feature "+n+" erroneo", feature, "geometry", expecteds[n][1]);

                checkEquals("feature "+n+" text_attribute mismatch", feature, "text_attribute", expecteds[n][2]);

                checkEquals("feature "+n+" real_attribute mismatch", feature, "real_attribute", expecteds[n][3]);

                checkEquals("feature "+n+" relation mismatch", feature, "relation", expecteds[n][4]);

                n++;

            }

            DisposeUtils.dispose(set);

            DisposeUtils.dispose(store1);

        } catch(Throwable t) {

            LOGGER.warn("",t);

            throw t;

        }

    }

    private void checkEquals(String s, Feature f, String attrname, Object expected) {

        s = s + " ("+attrname+")";

        Object value = f.get(attrname);

        if( expected == null ) {

            assertNull(s, value);

            return;

        }

        assertEquals(s, expected, value);

    }

    private String getTypeId(int type) {

        DataTypesManager manager = ToolsLocator.getDataTypesManager();

        DataType datatype = manager.get(type);

        return datatype.getName()+"("+type+")";

    }

}