| gvSIG desktop 1

package org.gvsig.data.datastores.vectorial.driver.jdbc;

import java.io.File;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.SQLException;

import java.util.Iterator;

import junit.framework.TestCase;

import org.gvsig.data.DataSourceManager;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.datastores.vectorial.driver.DriverStore;

import org.gvsig.data.datastores.vectorial.driver.IDriverParameters;

import org.gvsig.data.datastores.vectorial.driver.IDriverStoreParameters;

import org.gvsig.data.datastores.vectorial.driver.jdbc.h2.H2DriverParameters;

import org.gvsig.data.datastores.vectorial.driver.jdbc.h2.Register;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.exception.WriteException;

import org.gvsig.data.vectorial.DefaultAttributeDescriptor;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureAttributeDescriptor;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureStore;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.visitor.PrintlnFeaturesVisitor;

import org.gvsig.exceptions.BaseException;

public class DataStoreTest extends TestCase {

//        private File dbffile = new File(DataStoreTest.class.getResource("data/prueba.dbf").getFile());

//        private File shpfile = new File(DataStoreTest.class.getResource("data/prueba.shp").getFile());

        public static void main(String[] args) {

                junit.textui.TestRunner.run(DataStoreTest.class);

        }

        protected void setUp() throws Exception {

                super.setUp();

        }

        public void testH2() {

                System.out.println("======= H2 ==============");

                org.gvsig.data.datastores.vectorial.driver.Register.selfRegister();

                Register.selfRegister();

                DataSourceManager dsm=DataSourceManager.getManager();

                H2DriverParameters dp=(H2DriverParameters)dsm.createDriverParameters("h2");

                dp.setHost("localhost");

                dp.setUser("SA");

                dp.setPassw("");

                dp.setFields("*");

                dp.setGeometryField("GEOMETRY");

                dp.setTableName("TablaPrueba28244");

                dp.setFieldId("id");

                driverTest((IDriverParameters)dp,null,null,true);

//                IFeatureStore fs = createFeatureStore(dp);

//                assertNotNull("Can't create Feature Store", fs);

//

//                fs.open();

//

//                Iterator it;

//                IFeatureCollection fc;

//                Comparable v1,v2;

//                IFeature feature,pfeature;

//                long count;

//

//

//                fc = (IFeatureCollection)fs.getDataCollection();

//

//                assertEquals(9, fc.size());

//

//                fc = (IFeatureCollection)fs.getDataCollection(null,"lower(NOMBRE) like 'b%'",null);

//

//                assertEquals(2, fc.size());

//

//                it = fc.iterator();

//                count=0;

//                while (it.hasNext()){

//                        feature = (IFeature)it.next();

//                        assertTrue("Filter error",feature.getString("NOMBRE").toLowerCase().startsWith("b"));

//                        count++;

//                }

//                assertEquals("Iteration error",2,count);

//

//

//                fc = (IFeatureCollection)fs.getDataCollection(null,null,"NOMBRE ASC");

//                assertEquals(9, fc.size());

//                it = fc.iterator();

//                count=0;

//                pfeature = (IFeature)it.next();

//                count++;

//                while (it.hasNext()){

//                        feature = (IFeature)it.next();

//                        v1 = (Comparable)pfeature.get("NOMBRE");

//                        v2 = (Comparable)feature.get("NOMBRE");

//                        pfeature=feature;

//                        assertTrue("Short error", (v1.compareTo(v1) <= 0));

//                        count++;

//                }

//                assertEquals("Iteration error",9,count);

//

//

//                fc = (IFeatureCollection)fs.getDataCollection(null,null,"NOMBRE DESC");

//                assertEquals(9, fc.size());

//                it = fc.iterator();

//

//                count=0;

//                pfeature = (IFeature)it.next();

//                count++;

//                while (it.hasNext()){

//                        feature = (IFeature)it.next();

//                        v1 = (Comparable)pfeature.get("NOMBRE");

//                        v2 = (Comparable)feature.get("NOMBRE");

//                        pfeature=feature;

//                        assertTrue("Short error", (v1.compareTo(v1) >= 0));

//                        count++;

//                }

//                assertEquals("Iteration error",9,count);

//

//

//                fc = (IFeatureCollection)fs.getDataCollection(null,"lower(NOMBRE) like 'b%'","NOMBRE");

//

//                assertEquals(2, fc.size());

//

//                it = fc.iterator();

//

//                count=0;

//                pfeature = (IFeature)it.next();

//                assertTrue(pfeature.getString("NOMBRE").toLowerCase().startsWith("b"));

//                count++;

//                while (it.hasNext()){

//                        feature = (IFeature)it.next();

//                        assertTrue("Filter error",feature.getString("NOMBRE").toLowerCase().startsWith("b"));

//                        v1 = (Comparable)pfeature.get("NOMBRE");

//                        v2 = (Comparable)feature.get("NOMBRE");

//                        pfeature=feature;

//                        assertTrue("Short error", (v1.compareTo(v1) <= 0));

//                        count++;

//                }

//                assertEquals("Iteration error",2,count);

//

//

//

//                fc = (IFeatureCollection)fs.getDataCollection(null,null,"Tipo,lower(NOMBRE) Desc");

//                assertEquals(9, fc.size());

//                it = fc.iterator();

//                count=0;

//                pfeature = (IFeature)it.next();

//                System.out.println(pfeature.getString("NOMBRE"));

//                count++;

//                while (it.hasNext()){

//                        feature = (IFeature)it.next();

//                        v1 = (Comparable)((String)pfeature.get("NOMBRE")).toLowerCase();

//                        v2 = (Comparable)((String)feature.get("NOMBRE")).toLowerCase();

//                        pfeature=feature;

//                        assertTrue("Short error", (v1.compareTo(v2) >= 0));

//                        System.out.println(pfeature.getString("NOMBRE"));

//                        count++;

//                }

//                assertEquals("Iteration error",9,count);

//

//

//                try {

//                        fs.close();

//                } catch (CloseException e) {

//                        e.printStackTrace();

//                        fail("Exception:" + e);

//                }

//

//

//                System.out.println("======= /H2 ==============");

        }

        private IFeatureStore createFeatureStore(IDriverParameters dp){

                DataSourceManager dsm=DataSourceManager.getManager();

                IDataStoreParameters dsp=dsm.createDataStoreParameters(DriverStore.DATASTORE_NAME);

                ((IDriverStoreParameters)dsp).setDriverParameters(dp);

                IFeatureStore fs=null;

                try {

                        fs = (IFeatureStore)dsm.createDataStore(dsp);

                } catch (InitializeException e) {

                        e.printStackTrace();

                        fail("Exception:" + e);

                }

                return fs;

        }

        private void driverTest(IDriverParameters dp,String filter, String order,boolean testEdit){

                DataSourceManager dsm=DataSourceManager.getManager();

                IDataStoreParameters dsp=dsm.createDataStoreParameters(DriverStore.DATASTORE_NAME);

                ((IDriverStoreParameters)dsp).setDriverParameters(dp);

                IFeatureStore fs=createFeatureStore(dp);

                try {

                        fs.open();

                } catch (OpenException e2) {

                        e2.printStackTrace();

                        fail();

                }

                if (fs.isEditable() && testEdit) {

                        fs.startEditing();

                        IFeature feature1 = fs.createDefaultFeature(false);

                        IFeature feature2 = fs.createDefaultFeature(false);

                        IFeature feature3 = fs.createDefaultFeature(false);

                        fs.insert(feature1);

                        fs.insert(feature2);

                        fs.update(feature3,feature1);

                        fs.delete(feature3);

                        fs.delete(feature2);

                }

                //Mostrar por consola todos los registros.

                IFeatureType ft= fs.getDefaultFeatureType();

                IFeatureCollection featureCollection;

//                featureCollection = (IFeatureCollection)fs.getDataCollection();

//                featureCollection = (IFeatureCollection)fs.getDataCollection(ft,"NOMBRE = 'CALPE'",null);

//                featureCollection = (IFeatureCollection)fs.getDataCollection(ft,"AREA > 3.2213163729E7 and AREA < 3.2213163749E7",null);

                featureCollection = (IFeatureCollection)fs.getDataCollection(ft,filter,order);

                PrintlnFeaturesVisitor visitor=new PrintlnFeaturesVisitor(ft);

                try {

                        featureCollection.accept(visitor);

                } catch (BaseException e1) {

                        e1.printStackTrace();

                        fail("Exception: "+e1);

                }

                if (fs.isEditable() && testEdit){

                        try {

                                fs.finishEditing();

                        } catch (WriteException e) {

                                e.printStackTrace();

                                fail("Exception: "+e);

                        } catch (ReadException e) {

                                e.printStackTrace();

                                fail("Exception: "+e);

                        }

                }

                try {

                        fs.close();

                } catch (CloseException e) {

                        e.printStackTrace();

                        fail("Exception: "+e);

                }

                fs.dispose();

        }

}