Application: gvSIG desktop

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

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

import org.gvsig.fmap.dal.DatabaseWorkspaceManager;

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

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

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

import org.gvsig.tools.dispose.DisposeUtils;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public abstract class AbstractTestExport2db extends TestCase {

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

    protected AbstractTestUtils utils;

    public AbstractTestExport2db(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    public AbstractTestUtils utils() {

        if (this.utils == null) {

            this.utils = this.createUtils();

        }

        return this.utils;

    }

    protected abstract AbstractTestUtils createUtils();

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

    // public void testHello() {}

    protected void testExport1() throws Exception {

        try {

            if (!utils().isTheDatabaseAvailable()) {

                return;

            }

            FeatureStore sourceStore = utils().openCSVStore("/org/gvsig/fmap/dal/store/jdbc2/continentes_nogeom.csv");

            JDBCServerExplorer explorer = utils().openServerExplorer("testExport1");

            utils().info_jdbc(explorer);

            // Importamos el CSV en la bbdd

            utils().drop_tables(explorer, "continentes", DatabaseWorkspaceManager.TABLE_RESOURCES_NAME);

            utils().create_table_from(explorer, "continentes", sourceStore);

            utils().insert_into_from(explorer, "continentes", sourceStore, FeatureStore.MODE_APPEND);

            // Le ponemos como Pk el campo ID

            FeatureStore dbstore = utils().openStore(explorer, "continentes");

            dbstore.edit();

            FeatureType featureType = dbstore.getDefaultFeatureType();

            EditableFeatureType eFeatureType = featureType.getEditable();

            eFeatureType.getEditableAttributeDescriptor("ID").setIsPrimaryKey(true);

            dbstore.update(eFeatureType);

            dbstore.finishEditing();

            DisposeUtils.dispose(dbstore);

            // Comprobamos los campos que tiene la tabla de la bbdd.

            utils().removeResource(explorer, "continentes", "dal");

            dbstore = utils().openStore(explorer, "continentes");

            featureType = dbstore.getDefaultFeatureType();

            assertEquals("Fields count", 2, featureType.size());

            assertEquals("Field(0).name", "ID", featureType.getAttributeDescriptor(0).getName());

            assertEquals("Field(1).name", "NAME", featureType.getAttributeDescriptor(1).getName());

            assertEquals("Field(ID).isPrimaryKey", true, featureType.getAttributeDescriptor(0).isPrimaryKey());

            assertEquals("Field(NAME).isPrimaryKey", false, featureType.getAttributeDescriptor(1).isPrimaryKey());

            assertEquals("Field(ID).isIndexed", true, featureType.getAttributeDescriptor(0).isIndexed());

            assertEquals("Field(NAME).isIndexed", false, featureType.getAttributeDescriptor(1).isIndexed());

            // Le ponemos como Pk el campo NAME y lo quitamos del ID

            dbstore = utils().openStore(explorer, "continentes");

            dbstore.edit();

            featureType = dbstore.getDefaultFeatureType();

            eFeatureType = featureType.getEditable();

            eFeatureType.getEditableAttributeDescriptor("ID").setIsPrimaryKey(false);

            eFeatureType.getEditableAttributeDescriptor("NAME").setIsPrimaryKey(true);

            dbstore.update(eFeatureType);

            dbstore.finishEditing();

            DisposeUtils.dispose(dbstore);

            // Comprobamos los campos que tiene la tabla de la bbdd.

            utils().removeResource(explorer, "continentes", "dal");

            dbstore = utils().openStore(explorer, "continentes");

            featureType = dbstore.getDefaultFeatureType();

            assertEquals("Fields count", 2, featureType.size());

            assertEquals("Field(0).name", "ID", featureType.getAttributeDescriptor(0).getName());

            assertEquals("Field(1).name", "NAME", featureType.getAttributeDescriptor(1).getName());

            assertEquals("Field(ID).isPrimaryKey", false, featureType.getAttributeDescriptor(0).isPrimaryKey());

            assertEquals("Field(NAME).isPrimaryKey", true, featureType.getAttributeDescriptor(1).isPrimaryKey());

            assertEquals("Field(ID).isIndexed", false, featureType.getAttributeDescriptor(0).isIndexed());

            assertEquals("Field(NAME).isIndexed", true, featureType.getAttributeDescriptor(1).isIndexed());

            DisposeUtils.dispose(dbstore);

            // Le ponemos como Pk el campo NAME y lo quitamos del ID

            dbstore = utils().openStore(explorer, "continentes");

            dbstore.edit();

            featureType = dbstore.getDefaultFeatureType();

            eFeatureType = featureType.getEditable();

            eFeatureType.getEditableAttributeDescriptor("ID").setIsPrimaryKey(false);

            eFeatureType.getEditableAttributeDescriptor("NAME").setIsPrimaryKey(true);

            eFeatureType.getEditableAttributeDescriptor("ID").setIsIndexed(true);

            eFeatureType.getEditableAttributeDescriptor("NAME").setIsIndexed(true);

            dbstore.update(eFeatureType);

            dbstore.finishEditing();

            DisposeUtils.dispose(dbstore);

            // Comprobamos los campos que tiene la tabla de la bbdd.

            utils().removeResource(explorer, "continentes", "dal");

            dbstore = utils().openStore(explorer, "continentes");

            featureType = dbstore.getDefaultFeatureType();

            assertEquals("Fields count", 2, featureType.size());

            assertEquals("Field(0).name", "ID", featureType.getAttributeDescriptor(0).getName());

            assertEquals("Field(1).name", "NAME", featureType.getAttributeDescriptor(1).getName());

            assertEquals("Field(ID).isPrimaryKey", false, featureType.getAttributeDescriptor(0).isPrimaryKey());

            assertEquals("Field(NAME).isPrimaryKey", true, featureType.getAttributeDescriptor(1).isPrimaryKey());

            assertEquals("Field(ID).isIndexed", true, featureType.getAttributeDescriptor(0).isIndexed());

            assertEquals("Field(NAME).isIndexed", true, featureType.getAttributeDescriptor(1).isIndexed());

            DisposeUtils.dispose(dbstore);

        } catch (Throwable th) {

            LOGGER.warn("", th);

            throw th;

        }

    }

}