gvSIG projects pool

package org.gvsig.vcsgis.lib.impl;

import java.io.File;

import java.util.List;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

import org.apache.commons.io.FileUtils;

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

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

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

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

import org.gvsig.tools.dispose.DisposeUtils;

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

import org.gvsig.tools.util.GetItemWithSize64;

import org.gvsig.vcsgis.lib.VCSGisLocator;

import org.gvsig.vcsgis.lib.VCSGisManager;

import static org.gvsig.vcsgis.lib.VCSGisManager.ERR_NO_ERROR;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_DELETE;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_INSERT;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_UPDATE;

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class Test11GetLocalChanges extends TestCase {

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

    public Test11GetLocalChanges(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 testUpdate() throws Exception {

        VCSGisManager manager = VCSGisLocator.getVCSGisManager();

        JDBCServerExplorer server = TestUtils.openServerExplorer("srv-update");

        File ws1file = TestUtils.getFile(FileUtils.getFile("test-dbs","ws1-update")); 

        int r;

        Feature f;

        EditableFeature ef;

        List<Feature> features;

        FeatureStore store1;

        // ------------------------------------------------------------

        // Inicializamos el repositorio y lo abrimos

        r = manager.initRepository(server.getParameters(), null);

        assertEquals("init_server status", ERR_NO_ERROR, r);

        VCSGisRepository repo = manager.openRepository(server.getParameters());

        TestUtils.h2sql_repository(repo);

        // ------------------------------------------------------------

        // Creamos workspace1 y lo abrimos.

        r = manager.initWorkspace(ws1file, repo, "Test update1",null);

        assertEquals("init_ws1 status", ERR_NO_ERROR, r);

        VCSGisWorkspace ws1 = manager.openWorkspace(ws1file);

        TestUtils.h2sql_workspace(ws1);

        // ------------------------------------------------------------

        // Adicionamos al workspace1 la tabla "test" y la commitamos.

        FeatureStore sourceTest = TestUtils.openSourceStore2();

        r = ws1.add("test", sourceTest, "text");

        assertEquals("ws1.add status", ERR_NO_ERROR, r);

        r = ws1.commit();

        assertEquals("ws1.commit-1 status", ERR_NO_ERROR, r);

        DisposeUtils.disposeQuietly(sourceTest);

        // ------------------------------------------------------------

        // Modificamos la tabla "test" en el workspace1 y no la commitamos.

        store1 = ws1.getFeatureStore("test");

        store1.edit();

        f = store1.findFirst("id=2");

        ef = f.getEditable();

        ef.set("text", "BB2");

        store1.update(ef);

        String code1 = ef.getString(VCSGisManager.FEATURECODE_FIELD_NAME);

        assertNotNull("featureCode 1", code1);

        ef = store1.createNewFeature();

        ef.set("id",4);

        ef.set("text","DDD");

        store1.insert(ef);

        String code2 = ef.getString(VCSGisManager.FEATURECODE_FIELD_NAME);

        assertNotNull("featureCode 2", code2);

        f = store1.findFirst("id=3");

        store1.delete(f);        

        String code3 = f.getString(VCSGisManager.FEATURECODE_FIELD_NAME);

        assertNotNull("featureCode 3", code3);

        store1.finishEditing();

        DisposeUtils.disposeQuietly(store1);

        GetItemWithSize64<VCSGisWorkspaceChange> localChanges = ws1.getLocalChangesByEntityName("test");

        assertEquals("localChangesSize", 3L, localChanges.size64());

        VCSGisWorkspaceEntity entity = ws1.getWorkspaceEntityByName("test");

        for(long i = 0; i<localChanges.size64(); i++){

            VCSGisWorkspaceChange change = localChanges.get64(i);

            assertEquals(entity.getEntityCode(), change.getEntityCode());

            if(change.getRelatedFeatureCode().equalsIgnoreCase(code1)){

                assertEquals(OP_UPDATE, change.getOperation());

                assertEquals("BB2", change.getLabel());

                f = change.getRelatedFeature();

                assertEquals(2, f.getInt("id"));

                assertEquals("BB2", f.getString("text"));

            } else if(change.getRelatedFeatureCode().equalsIgnoreCase(code2)){

                assertEquals(OP_INSERT, change.getOperation());

                assertEquals("DDD", change.getLabel());

                f = change.getRelatedFeature();

                assertEquals(4, f.getInt("id"));

                assertEquals("DDD", f.getString("text"));

            } else if(change.getRelatedFeatureCode().equalsIgnoreCase(code3)){

                assertEquals(OP_DELETE, change.getOperation());

                assertEquals("CCC", change.getLabel());

                boolean ok = true;

                try {

                    change.getRelatedFeature();

                    ok = false;

                } catch (Exception e) {

                }

                assertTrue("code3 deleted", ok);

            } else {

                assertTrue("Incorrect code.", false);

            }

        }

    }

}