Application: gvSIG desktop

package org.gvsig.sqlite.dal;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import junit.framework.TestCase;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.DatabaseWorkspaceManager;

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

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

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

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

public class WorkspaceTest extends TestCase {

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

    public static final String DBNAME = "test_workspace";

    public WorkspaceTest(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    protected String getProviderName() {

        return DataStore.H2SPATIAL_PROVIDER_NAME;

    }

    protected String getTargetName() {

        return "testCreateTarget1";

    }

    protected FeatureStore openTargetStore1(JDBCServerExplorer explorer) throws Exception {

        JDBCStoreParameters params = explorer.get(getTargetName());

        DataManager dataManager = DALLocator.getDataManager();

        FeatureStore store;

        try {

            store = (FeatureStore) dataManager.openStore(

                    getProviderName(), 

                    params

            );

        } catch(ValidateDataParametersException ex) {

            LOGGER.warn(ex.getLocalizedMessageStack());

            throw ex;

        }

        return store;

    }

    protected void createFrom(JDBCServerExplorer explorer, FeatureStore sourceStore) throws Exception {

        NewFeatureStoreParameters params = (NewFeatureStoreParameters) explorer.getAddParameters(

                getTargetName()

        );

        EditableFeatureType ft = params.getDefaultFeatureType();

        ft.addAll(sourceStore.getDefaultFeatureType());

        explorer.add(getProviderName(), params, true);

    }

    public void testWorkspace() throws Exception {

        try {

            FeatureStore sourceStore = TestUtils.openSourceStore1();

            JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);

            createFrom(explorer, sourceStore);        

            DataManager dataManager = DALLocator.getDataManager();

            DatabaseWorkspaceManager workspace = dataManager.createDatabaseWorkspaceManager(explorer.getParameters());

            workspace.create("Testid", "Test label");

            workspace.disconnect();

            workspace = dataManager.createDatabaseWorkspaceManager(explorer.getParameters());

            assertEquals("Testid",workspace.getId());

            assertEquals("Test label",workspace.getLabel());

            List<DataStoreParameters> tables_l = explorer.list();

            assertEquals(12, tables_l.size()); 

            Map<String, JDBCStoreParameters> tables = new HashMap<>();        

            for (DataStoreParameters table1 : tables_l) {

                JDBCStoreParameters table = (JDBCStoreParameters) table1;

                tables.put(table.getTable(), table);

            }

            JDBCStoreParameters params = tables.get("testCreateTarget1");

            assertNotNull("Table testCreateTarget1 not found",params);

            workspace.writeStoresRepositoryEntry(params.getTable(), params);

            DataStoreParameters store = workspace.getStoresRepository().get(params.getTable());

            assertNotNull("Can't open table testCreateTarget1 ",store);

        } catch (Throwable th) {

            LOGGER.warn("testWorkspace", th);

            throw th;

        }

    }

}