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.expressionevaluator.Expression;

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

import org.gvsig.vcsgis.lib.VCSGisLocator;

import org.gvsig.vcsgis.lib.VCSGisManager;

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

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

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

import org.gvsig.vcsgis.lib.vcsgis;

import org.gvsig.vcsgis.lib.vcsgisadmin;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class Test10UpdateLocalModified extends TestCase {

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

    public Test10UpdateLocalModified(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    private void check(FeatureStore store2, String relatedFeatureCode, int id, String text) throws Exception {

        Feature f = store2.findFirst("VCSGISCODE = '"+relatedFeatureCode+"'");

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

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

    }

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

    // public void testHello() {}

    public void testUpdateLocalModified() throws Exception {

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

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

        File ws2file = TestUtils.getFile(FileUtils.getFile("test-dbs","ws2-updatelocalmodified")); 

        VCSGisManager manager = VCSGisLocator.getVCSGisManager();

        manager.setCodeGenerator(new SequentialCodeGenerator());

        int r;

        Feature f;

        EditableFeature ef;

        List<Feature> features;

        FeatureStore store1;

        FeatureStore store2;

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

        // Inicializamos el repositorio y lo abrimos

        r = vcsgisadmin.init_repository(server);

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

        VCSGisRepository repo = vcsgisadmin.open_repository(server);

        TestUtils.h2sql_repository(repo);

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

        // Creamos workspace1 y lo abrimos.

        r = vcsgis.init_workspace(ws1file, repo, "Ws1: Test update local modified");

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

        VCSGisWorkspace ws1 = vcsgis.open_workspace(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);

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

        // Inicializamos el workspace2 asociandolo al mismo repositorio que 

        // workspace1 y lo abrimos.

        r = vcsgis.init_workspace(ws2file, repo, "Ws2: Test update local modified");

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

        VCSGisWorkspace ws2 = vcsgis.open_workspace(ws2file);

        TestUtils.h2sql_workspace(ws2);

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

        // Hacemos un checkout en workspace2 de la tabla "test" 

        // y comprobamos que tiene lo que toca.

        r = ws2.checkout("test");

        assertEquals("ws2-checkout status", ERR_NO_ERROR, r);

        store2 = ws2.getFeatureStore("test");

        assertEquals("ws2-checkout sizes", 3, store2.size64());

        features = store2.getFeatures((Expression)null, "id", true);

        assertEquals(1    , features.get(0).getInt("id"));

        assertEquals("AAA", features.get(0).getString("text"));

        assertEquals(2    , features.get(1).getInt("id"));

        assertEquals("BBB", features.get(1).getString("text"));

        assertEquals(3    , features.get(2).getInt("id"));

        assertEquals("CCC", features.get(2).getString("text"));

        DisposeUtils.disposeQuietly(store2);

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

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

        store1 = ws1.getFeatureStore("test");

        store1.edit();

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

        ef = f.getEditable();

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

        store1.update(ef);

        ef = store1.createNewFeature();

        ef.set("id",4);

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

        store1.insert(ef);

        store1.finishEditing();

        DisposeUtils.disposeQuietly(store1);

        r = ws1.commit();

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

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

        // Modificamos la tabla "test" en el workspace2 

        // estando desactualizada. 

        // Una actualizacion, una insercion y un borrado.

        // No commitamos los cambios.

        store2 = ws2.getFeatureStore("test");

        store2.edit();

        f = store2.findFirst("id=1");

        ef = f.getEditable();

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

        store2.update(ef);

        ef = store2.createNewFeature();

        ef.set("id",5);

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

        store2.insert(ef);

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

        store2.delete(f);        

        store2.finishEditing();

        DisposeUtils.disposeQuietly(store2);

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

        // Commitamos la tabla "test" en el workspace2,

        // tiene que fallar por estar desactualizada.

        r = ws2.commit();

        assertEquals("ws2.commit-1 status", ERR_CANT_COMMIT_WORKSPACE_OUTDATED, r);

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

        // Actualizamos (update) la tabla "test" en el workspace2 

        // y comprobamos que tiene lo que toca.

        r = ws2.updatePrepare("test");

        assertEquals("ws2.update-2 status", ERR_NO_ERROR, r);

        r = ws2.update("test");

        assertEquals("ws2.update-1 status", ERR_NO_ERROR, r);

        store2 = ws2.getFeatureStore("test");

        assertEquals("ws2-upfate sizes", 4, store2.size64());

        features = store2.getFeatures((Expression)null, "id", true);

        assertEquals(1    , features.get(0).getInt("id"));

        assertEquals("AA2", features.get(0).getString("text"));

        assertEquals(2    , features.get(1).getInt("id"));

        assertEquals("BB2", features.get(1).getString("text"));

        assertEquals(4    , features.get(2).getInt("id"));

        assertEquals("DDD", features.get(2).getString("text"));

        assertEquals(5    , features.get(3).getInt("id"));

        assertEquals("EEE", features.get(3).getString("text"));

        DisposeUtils.disposeQuietly(store2);

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

        // Commitamos la tabla "test" en el workspace2.

        // Se supone que estamos actualizados y podemos commitar.

        r = ws2.commit();

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

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

        // Actualizamos (update) la tabla "test" en el workspace1

        // y comprobamos que tiene lo que toca.

        r = ws1.updatePrepare("test");

        assertEquals("ws2.update-2 status", ERR_NO_ERROR, r);

        r = ws1.update("test");

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

        store1 = ws1.getFeatureStore("test");

        assertEquals("ws1-upfate sizes", 4, store1.size64());

        features = store1.getFeatures((Expression)null, "id", true);

        assertEquals(1    , features.get(0).getInt("id"));

        assertEquals("AA2", features.get(0).getString("text"));

        assertEquals(2    , features.get(1).getInt("id"));

        assertEquals("BB2", features.get(1).getString("text"));

        assertEquals(4    , features.get(2).getInt("id"));

        assertEquals("DDD", features.get(2).getString("text"));

        assertEquals(5    , features.get(3).getInt("id"));

        assertEquals("EEE", features.get(3).getString("text"));

        DisposeUtils.disposeQuietly(store1);

    }

}