Application: gvSIG desktop

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

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 static org.gvsig.fmap.dal.store.h2.TestCreate.DBNAME;

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 {

        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(6, 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);

    }

}