gvSIG projects pool

package org.gvsig.vcsgis.lib.impl;

import java.io.File;

import java.util.Arrays;

import java.util.Collections;

import java.util.Comparator;

import java.util.List;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.lang3.tuple.ImmutablePair;

import org.apache.commons.lang3.tuple.Pair;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStoreParameters;

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

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

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

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

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.gvsig.vcsgis.lib.VCSGisLocator;

import org.gvsig.vcsgis.lib.VCSGisManager;

import org.gvsig.vcsgis.lib.repository.VCSGisRepository;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspace;

import org.gvsig.vcsgis.lib.workspace.tables.EntitiesTable;

import org.gvsig.vcsgis.lib.workspace.tables.WorkspaceChangesTable;

import org.gvsig.vcsgis.lib.workspace.tables.RemoteChangesTable;

import org.gvsig.vcsgis.lib.workspace.tables.VarsTable;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class Test02Initws extends TestCase {

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

    public Test02Initws(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

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

    // public void testHello() {}

    public void testCreateWS() throws Exception {

        String testid = "initws";

        VCSGisManager vcsgismanager = VCSGisLocator.getVCSGisManager();

        JDBCServerExplorerParameters server = TestUtils.getServerExplorerParameters("srv-"+testid);

        File wsfile = TestUtils.getFile(FileUtils.getFile("test-dbs","ws-"+testid)); 

        int r = vcsgismanager.initRepository(server, null);

        assertEquals("initRepository status", 0, r);

        TestUtils.h2sql_repository(server);

        VCSGisRepository repo = vcsgismanager.openRepository(server);

        r = vcsgismanager.initWorkspace(wsfile, repo, "Test initws", null);

        assertEquals("initWorkspace ws status", 0, r);

        TestUtils.h2sql_workspace(wsfile);

        VCSGisWorkspace workspace = vcsgismanager.openWorkspace(wsfile);

        assertNotNull("Can't open workspace", workspace);

        JDBCServerExplorer wsexplorer = TestUtils.openServerExplorer(wsfile);

        List<DataStoreParameters> tables = wsexplorer.list();

        Collections.sort(tables, new Comparator<DataStoreParameters>() {

            @Override

            public int compare(DataStoreParameters o1, DataStoreParameters o2) {

                return StringUtils.compare(

                        ((JDBCStoreParameters) o1).getTable(),

                        ((JDBCStoreParameters) o2).getTable()

                );

            }

        });

        Pair<String, FeatureType>[] expectedTables = new Pair[]{

            new ImmutablePair(WorkspaceChangesTable.TABLE_NAME,WorkspaceChangesTable.featureType()),

            new ImmutablePair(VarsTable.TABLE_NAME,VarsTable.featureType()),

            new ImmutablePair(EntitiesTable.TABLE_NAME,EntitiesTable.featureType()),

            new ImmutablePair(RemoteChangesTable.TABLE_NAME,RemoteChangesTable.featureType()),

            new ImmutablePair("gvsigd_config",null),

//            new ImmutablePair("gvsigd_repository",null),

            new ImmutablePair("gvsigd_resources",null),

        };

        Arrays.sort(expectedTables, new Comparator<Pair<String, FeatureType>>() {

            @Override

            public int compare(Pair<String, FeatureType> o1, Pair<String, FeatureType> o2) {

                return StringUtils.compare(o1.getKey(),o2.getKey());

            }

        });

        int n = 0;

        for (DataStoreParameters table : tables) {

            JDBCStoreParameters params = (JDBCStoreParameters) table;

            String tableName = params.getTable();

            if (!tableName.startsWith("VCSGIS_") && !tableName.startsWith("gvsigd_")) {

                continue;

            }

            assertEquals("table " + n, expectedTables[n++].getKey(), tableName);

        }

        DataManager dataManager = DALLocator.getDataManager();

        n = 0; 

        for (DataStoreParameters tableParams : tables) {

            JDBCStoreParameters params = (JDBCStoreParameters) tableParams;

            String tableName = params.getTable();

            if (!tableName.startsWith("VCSGIS_") && !tableName.startsWith("gvsigd_")) {

                continue;

            }

            FeatureStore store = (FeatureStore) dataManager.openStore(params.getProviderName(), params);

            assertEquals("table " + n, expectedTables[n].getKey(), store.getName());

            FeatureType expectedFeatureType = expectedTables[n].getValue();

            if( expectedFeatureType!=null ) {

                FeatureType ft = store.getDefaultFeatureTypeQuietly();

                for (FeatureAttributeDescriptor expectedAttr : expectedFeatureType) {

                    FeatureAttributeDescriptor attr = ft.getAttributeDescriptor(expectedAttr.getName());

                    assertNotNull("Attr "+expectedAttr.getName(), attr);

                    assertEquals("Attr "+expectedAttr.getName()+" type", expectedAttr.getType(), attr.getType());

                    assertEquals("Attr "+expectedAttr.getName()+" size", expectedAttr.getSize(), attr.getSize());

                    assertEquals("Attr "+expectedAttr.getName()+" isPrimaryKey", expectedAttr.isPrimaryKey(), attr.isPrimaryKey());

                    assertEquals("Attr "+expectedAttr.getName()+" isIndexed", expectedAttr.isIndexed(), attr.isIndexed());

                    assertEquals("Attr "+expectedAttr.getName()+" allowIndexDuplicateds", expectedAttr.allowIndexDuplicateds(), attr.allowIndexDuplicateds());

                    assertEquals("Attr "+expectedAttr.getName()+" isReadOnly", expectedAttr.isReadOnly(), attr.isReadOnly());

                    assertEquals("Attr "+expectedAttr.getName()+" DefaultValue", expectedAttr.getDefaultValue(), attr.getDefaultValue());

                }

            }

            store.dispose();

            n++;

        }

    }

}