Application: gvSIG desktop

package org.gvsig.fmap.data.feature.file;

import java.io.File;

import junit.framework.TestCase;

import org.gvsig.fmap.data.DataException;

import org.gvsig.fmap.data.DataExplorer;

import org.gvsig.fmap.data.DataExplorerParameters;

import org.gvsig.fmap.data.DataManager;

import org.gvsig.fmap.data.DataStore;

import org.gvsig.fmap.data.DataStoreParameters;

import org.gvsig.fmap.data.InitializeException;

import org.gvsig.fmap.data.OpenException;

import org.gvsig.fmap.data.ReadException;

import org.gvsig.fmap.data.WriteException;

import org.gvsig.fmap.data.feature.AttributeDescriptor;

import org.gvsig.fmap.data.feature.DefaultFeatureType;

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

import org.gvsig.fmap.data.feature.FeatureCollection;

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

import org.gvsig.fmap.data.feature.FeatureStoreExplorer;

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

import org.gvsig.fmap.data.feature.IsNotAttributeSettingException;

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

import org.gvsig.fmap.data.feature.file.dbf.DBFDataExplorer;

import org.gvsig.fmap.data.feature.file.dbf.DBFStoreParameters;

import org.gvsig.fmap.data.feature.file.shp.SHPDataExplorer;

import org.gvsig.fmap.data.feature.file.shp.SHPStoreParameters;

import com.iver.utiles.XMLEntity;

public class DataExplorerTest 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 void testDBFExplorer(){

                org.gvsig.fmap.data.feature.file.dbf.Register.selfRegister();

                DataStoreParameters[] params = null;

                params=list(DBFDataExplorer.DATASOURCE_NAME,dbffile.getParent());

                assertEquals(3, params.length);

                String file=dbffile.getParent()+File.separator+"pruebaCrear.dbf";

                DBFStoreParameters parameters=new DBFStoreParameters();

                parameters.setFile(new File(file));

                DefaultFeatureType dft=new DefaultFeatureType();

                AttributeDescriptor fad=(AttributeDescriptor) dft.createAttributeDescriptor();

                try {

                        fad.loading();

                        fad.setName("CAMPO1");

                        fad.setType(FeatureAttributeDescriptor.TYPE_STRING);

                        fad.setSize(10);

                        fad.stopLoading();

                } catch (IsNotAttributeSettingException e1) {

                        e1.printStackTrace();

                }

                dft.add(fad);

                add(DBFDataExplorer.DATASOURCE_NAME,file,dft);

                params=list(DBFDataExplorer.DATASOURCE_NAME,dbffile.getParent());

                assertEquals(4, params.length);

                FeatureStore fs=(FeatureStore)createDataStore(DBFDataExplorer.DATASOURCE_NAME,file,parameters);

                try {

//                        fs.open();

                        FeatureCollection fc = (FeatureCollection) fs.getDataCollection();

                        assertEquals(0, fc.size());

                        fc.dispose();

//                        fs.close();

                        fs.dispose();

                } catch (Exception e) {

                        e.printStackTrace();

                        fail();

                }

                try {

                        remove(DBFDataExplorer.DATASOURCE_NAME,file,parameters);

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                params=list(DBFDataExplorer.DATASOURCE_NAME,dbffile.getParent());

                assertEquals(3, params.length);

                persistenceTest(this.getParams(DBFDataExplorer.DATASOURCE_NAME, dbffile

                                .getParent()));

        }

        public void testSHPExplorer(){

                org.gvsig.fmap.data.feature.file.shp.Register.selfRegister();

                DataStoreParameters[] params = null;

                params=list(SHPDataExplorer.DATASOURCE_NAME,shpfile.getParent());

                assertEquals(3, params.length);

                String file=shpfile.getParent()+File.separator+"pruebaCrear.shp";

                SHPStoreParameters parameters=new SHPStoreParameters();

                parameters.setFile(new File(file));

//                parameters.setDBFFile(SHP.getDbfFile(new File(file)));

//                parameters.setSHXFile(SHP.getShxFile(new File(file)));

                DefaultFeatureType dft=new DefaultFeatureType();

                AttributeDescriptor fad1=(AttributeDescriptor) dft.createAttributeDescriptor();

                try{

                        fad1.loading();

                        fad1.setName("CAMPO1");

                        fad1.setType(FeatureAttributeDescriptor.TYPE_STRING);

                        fad1.setSize(10);

                        fad1.stopLoading();

                }catch (IsNotAttributeSettingException e) {

                        e.printStackTrace();

                }

                AttributeDescriptor fad2=(AttributeDescriptor) dft.createAttributeDescriptor();

                try{

                        fad2.loading();

                        fad2.setName("Geometry");

                        fad2.setType(FeatureAttributeDescriptor.TYPE_GEOMETRY);

                        fad2.stopLoading();

                }catch (IsNotAttributeSettingException e) {

                        e.printStackTrace();

                }

                dft.add(fad1);

                dft.add(fad2);

                add(SHPDataExplorer.DATASOURCE_NAME,file,dft);

                params=list(SHPDataExplorer.DATASOURCE_NAME,shpfile.getParent());

                assertEquals(4, params.length);

                FeatureStore fs=(FeatureStore)createDataStore(SHPDataExplorer.DATASOURCE_NAME,file,parameters);

                try {

//                        fs.open();

                        FeatureCollection fc = (FeatureCollection) fs.getDataCollection();

                        assertEquals(0, fc.size());

//                        fs.close();

                        fs.dispose();

                } catch (Exception e) {

                        e.printStackTrace();

                        fail();

                }

                try {

                        remove(SHPDataExplorer.DATASOURCE_NAME,file,parameters);

                } catch (ReadException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                params=list(SHPDataExplorer.DATASOURCE_NAME,shpfile.getParent());

                assertEquals(3, params.length);

                persistenceTest(this.getParams(SHPDataExplorer.DATASOURCE_NAME, shpfile

                                .getParent()));

        }

        private void remove(String name, String file,DataStoreParameters parameters) throws ReadException {

                DataManager dsm=DataManager.getManager();

                DataExplorerParameters dsp = dsm.createDataExplorerParameters(

                                name);

                ((FileExplorerParameters)dsp).setSource(new File(file).getParentFile());

                FeatureStoreExplorer src = (FeatureStoreExplorer)dsm.createDataExplorer(dsp);

                src.remove(parameters);

        }

        private void add(String name, String file,FeatureType featureType) {

                DataManager dsm=DataManager.getManager();

                DataExplorerParameters dsp=null;

                try {

                        dsp = dsm.createDataExplorerParameters(

                                        name);

                } catch (InitializeException e2) {

                        e2.printStackTrace();

                        fail();

                }

                ((FileExplorerParameters)dsp).setSource(new File(file).getParentFile());

                FeatureStoreExplorer src;

                try {

                        src = (FeatureStoreExplorer)dsm.createDataExplorer(dsp);

                } catch (InitializeException e1) {

                        e1.printStackTrace();

                        fail();return;

                }

                NewFeatureStoreParameters ndsp=(NewFeatureStoreParameters)src.createNewDataStoreParameter();

                ((FileStoreParameters)ndsp.getDataStoreParameters()).setFile(new File(file));

                ndsp.setFeatureType(featureType);

                try {

                        src.add(ndsp);

                } catch (InitializeException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                        fail();return;

                } catch (WriteException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                        fail();return;

                }

        }

        private DataStore createDataStore(String name, String file,DataStoreParameters parameters){

                DataManager dsm=DataManager.getManager();

                DataExplorerParameters dsp = getParams(name, new File(file)

                                .getParentFile().getAbsolutePath());

                FeatureStoreExplorer src;

                try {

                        src = (FeatureStoreExplorer)dsm.createDataExplorer(dsp);

                } catch (InitializeException e1) {

                        fail();return null;

                }

                try {

                        return src.createDataStore(parameters);

                } catch (OpenException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } catch (WriteException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                } catch (InitializeException e) {

                        // TODO Auto-generated catch block

                        e.printStackTrace();

                }

                return null;

        }

        private DataStoreParameters[] list(String name,String parent){

                DataManager dsm = DataManager.getManager();

                DataExplorerParameters dsp = getParams(name, parent);

                DataExplorer src;

                try {

                        src = dsm.createDataExplorer(dsp);

                } catch (InitializeException e1) {

                        e1.printStackTrace();

                        fail();return null;

                }

                DataStoreParameters[] params = null;

                try {

                        params = src.list();

                } catch (ReadException e) {

                        e.printStackTrace();

                        fail();

                }

                return params;

        }

        protected void persistenceTest(DataExplorerParameters params1) {

                DataManager manager = DataManager.getManager();

                XMLEntity params1XML;

                DataExplorerParameters params2 = null;

                XMLEntity params2XML;

                DataExplorer explorer1 = null;

                XMLEntity explorer1XML;

                DataExplorer explorer2 = null;

                XMLEntity explorer2XML;

                params1XML = params1.getXMLEntity();

                try {

                        params2 = manager.createDataExplorerParameters(params1XML);

                } catch (InitializeException e) {

                        e.printStackTrace();

                        fail();

                }

                params2XML = params2.getXMLEntity();

                assertEquals(params1XML.toString(), params2XML.toString());

                try {

                        explorer1 = manager.createDataExplorer(params1);

                } catch (InitializeException e1) {

                        e1.printStackTrace();

                        fail();

                }

                explorer1XML = explorer1.getXMLEntity();

                try {

                        explorer2 = manager.createDataExplorer(explorer1XML);

                } catch (InitializeException e1) {

                        e1.printStackTrace();

                        fail();

                }

                explorer2XML = explorer2.getXMLEntity();

                assertEquals(explorer1XML.toString(), explorer2XML.toString());

                DataStoreParameters list1[] = null;

                DataStoreParameters list2[] = null;

                try {

                        list1 = explorer1.list();

                        list2 = explorer2.list();

                } catch (ReadException e) {

                        e.printStackTrace();

                        fail();

                }

                assertEquals(list1.length, list2.length);

                for (int i=0;i< list1.length;i++){

                        assertEquals(list1[i].getXMLEntity().toString(), list2[i]

                                        .getXMLEntity().toString());

                }

                try {

                        explorer1.dispose();

                        explorer2.dispose();

                } catch (DataException e) {

                        e.printStackTrace();

                        fail();

                }

        }

        protected DataExplorerParameters getParams(String name, String parent) {

                DataManager dsm = DataManager.getManager();

                FileExplorerParameters dsp = null;

                try {

                        dsp = (FileExplorerParameters) dsm

                                        .createDataExplorerParameters(name);

                } catch (InitializeException e2) {

                        e2.printStackTrace();

                        fail();

                }

                dsp.setSource(new File(parent));

                return dsp;

        }

}