gvSIG projects pool

package org.gvsig.vcsgis.lib.impl;

import java.io.File;

import java.util.List;

import java.util.Map;

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.exception.DataException;

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

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

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

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

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_NO_ERROR;

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

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

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.tables.RemoteChangesTable;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class Test13Merge extends TestCase {

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

    public Test13Merge(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() {}

    private void insert(FeatureStore store, int id, String text) throws DataException{

        EditableFeature ef = store.createNewFeature();

        ef.set("id",id);

        ef.set("text",text);

        store.insert(ef);

    }

    private void update(FeatureStore store, int id, String text) throws DataException{

        Feature f = store.findFirst("id="+id);

        EditableFeature ef = f.getEditable();

        ef.set("text", text);

        store.update(ef);

    }

    private void delete(FeatureStore store, int id, String text) throws DataException{

        Feature f = store.findFirst("id="+id);

        store.delete(f);

    }

    private void check(List<Feature> features, int index, int id, String text){

        assertEquals(id    , features.get(index).getInt("id"));

        assertEquals(text, features.get(index).getString("text"));

    }

    private void check(List<Feature> features, Map<Integer,String> testValues){

        features.forEach(feature -> {

            assertEquals(testValues.get(feature.getInt("id")), feature.getString("text"));

        });

    }

    private void selectRemoteChange(VCSGisWorkspace ws, boolean select, int id, String text, VCSGisWorkspace wsSource) throws Exception {

        Feature fTest;

        FeatureStore testStore = ws.getFeatureStore("test");

        FeatureStore testStoreSource = wsSource.getFeatureStore("test");

        fTest = testStore.findFirst("id = "+id);

        if(fTest == null){

            fTest = testStoreSource.findFirst("id = "+id);

        }

        FeatureStore rchangesStore = ws.getFeatureStore(RemoteChangesTable.TABLE_NAME);

        Feature fChanges = rchangesStore.findFirst("RCH_DATACODE = '"+fTest.getString(FEATURECODE_FIELD_NAME)+"'");

        EditableFeature efChanges = fChanges.getEditable();

        rchangesStore.edit(MODE_PASS_THROUGH);

        efChanges.set("RCH_SELECTED",select);

        rchangesStore.update(efChanges);

        rchangesStore.finishEditing();

        DisposeUtils.dispose(testStore);

        DisposeUtils.dispose(testStoreSource);

        DisposeUtils.dispose(rchangesStore);

    }

    private void checkLocalChange(FeatureStore data, FeatureStore changes, String code, int operation, int id, String text) throws DataException{

        Feature change = changes.findFirst("WSCH_FEATURECODE = '"+code+"'");

        assertNotNull("Change code not exists", change);

        assertEquals(operation, change.getInt(WorkspaceChangesTable.OPERATION));

        // Uf, ojo con operation delete, no tiene data. en la primera pasada lo saltaria, y a ver que pasa con

        // los inserts y updates.

        if(operation != OP_DELETE){

            Feature f = data.findFirst("VCSGISCODE = '"+code+"'");

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

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

        }

    }

    public void testMerge() throws Exception {

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

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

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

        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 = 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.

        //REV 0

        FeatureStore sourceTest = TestUtils.openSourceStore2();

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

        DisposeUtils.disposeQuietly(sourceTest);

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

        store1 = ws1.getFeatureStore("test");

        store1.edit();

        insert(store1, 4, "DDD");

        insert(store1, 5, "EEE");

        store1.finishEditing();

        DisposeUtils.disposeQuietly(store1);

        r = ws1.commit();

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

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

        // Inicializamos el workspace2 asociandolo al mismo repositorio que 

        // workspace1 y lo abrimos.

        r = manager.initWorkspace(ws2file, repo, "Test update2",null);

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

        VCSGisWorkspace ws2 = manager.openWorkspace(ws2file);

        TestUtils.h2sql_workspace(ws2);

        ws2.checkout("test");

        //REV 1

        store1 = ws1.getFeatureStore("test");

        store1.edit();

        update(store1, 1, "AA2");

        update(store1, 2, "BB2");

        insert(store1, 6, "FFF");

        insert(store1, 7, "GGG");

        delete(store1, 3, "CCC");

        delete(store1, 5, "EEE");

        store1.finishEditing();

        DisposeUtils.disposeQuietly(store1);

        r = ws1.commit();

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

        store2 = ws2.getFeatureStore("test");

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

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

        // y comprobamos que tiene lo que toca.

        r = ws2.updatePrepare("test");

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

        store2.edit();

        insert(store2, 8, "HHH");

        store2.finishEditing();

        DisposeUtils.disposeQuietly(store2);

//        selectRemoteChange(ws2, true, 1, "AAA", ws1);

//        selectRemoteChange(ws2, true, 6, "FFF", ws1);

        selectRemoteChange(ws2, true, 3, "CCC", ws1);

        selectRemoteChange(ws2, false, 2, "BB2", ws1);

        selectRemoteChange(ws2, false, 5, "EEE", ws1);

        selectRemoteChange(ws2, false, 7, "GGG", ws1);

        r = ws2.merge("test");

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

        assertEquals("ws2-update sizes", 6, store2.size64());

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

        check(features, 0, 1, "AA2");

        check(features, 1, 2, "BBB");

        check(features, 2, 4, "DDD");

        check(features, 3, 5, "EEE");

        check(features, 4, 6, "FFF");

        check(features, 5, 8, "HHH");

        FeatureStore changes = ws2.getFeatureStore(WorkspaceChangesTable.TABLE_NAME);

        checkLocalChange(store2, changes, "00000000000066abcdef0123456789", OP_INSERT, 8, "HHH");

        checkLocalChange(store2, changes, "00000000000008abcdef0123456789", OP_UPDATE, 2, "BBB");

        checkLocalChange(store2, changes, "00000000000069abcdef0123456789", OP_INSERT, 5, "EEE");

        checkLocalChange(store2, changes, "00000000000035abcdef0123456789", OP_DELETE, 7, "GGG");

        r = ws2.commit();

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

        r = ws1.updatePrepare("test");

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

        r = ws1.update("test");

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

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

        check(features, 0, 1, "AA2");

        check(features, 1, 2, "BBB");

        check(features, 2, 4, "DDD");

        check(features, 3, 5, "EEE");

        check(features, 4, 6, "FFF");

        check(features, 5, 8, "HHH");

        DisposeUtils.disposeQuietly(features);

        DisposeUtils.disposeQuietly(store2);

        DisposeUtils.disposeQuietly(store1);

    }

}