gvSIG projects pool

package org.gvsig.vcsgis.lib.impl.server;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.InputStream;

import java.net.URL;

import java.sql.Timestamp;

import java.util.List;

import java.util.Map;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

import static junit.framework.TestCase.assertNotNull;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

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.jdbc.JDBCServerExplorerParameters;

import org.gvsig.tools.dispose.DisposeUtils;

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

import org.gvsig.vcsgis.lib.SequentialCodeGenerator;

import org.gvsig.vcsgis.lib.VCSGisEntity;

import org.gvsig.vcsgis.lib.VCSGisLocator;

import org.gvsig.vcsgis.lib.VCSGisManager;

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

import org.gvsig.vcsgis.lib.impl.TestUtils;

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

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

import org.gvsig.vcsgis.lib.repository.remoteclient.requests.CheckoutRequestClient;

import org.gvsig.vcsgis.lib.repository.remoteclient.requests.UpdateRequestClient;

import org.gvsig.vcsgis.lib.server.VCSGisServerHandler;

import org.gvsig.vcsgis.lib.server.handlers.CheckoutHandler;

import org.gvsig.vcsgis.lib.server.handlers.UpdateHandler;

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestS05Update extends TestCase {

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

    public TestS05Update(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 {

        final String testid = "ServerCheckout";

        final URL serverUrl = new URL("http://localhost:9810/");

        VCSGisManager manager = VCSGisLocator.getVCSGisManager();

        manager.setCodeGenerator(new SequentialCodeGenerator());

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

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

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

        int r;

        Feature f;

        EditableFeature ef;

        List<Feature> features;

        FeatureStore store1;

        FeatureStore store2;

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

        // Inicializamos el repositorio y lo abrimos

        r = manager.initRepository(serverParameters, null);

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

        VCSGisRepository repoLocaldb = manager.openRepository(serverParameters);

        TestUtils.h2sql_repository(repoLocaldb);

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

        // Creamos workspace1 y lo abrimos.

        r = manager.initWorkspace(ws1file, repoLocaldb, "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);

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

        // Inicializamos el workspace2 asociandolo al mismo repositorio que 

        // workspace1 y lo abrimos.

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

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

        VCSGisWorkspace ws2 = manager.openWorkspace(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);

        VCSGisEntity entity = ws2.getWorkspaceEntityByName("test");

        assertNotNull("ws2.update-1 entity", entity);

        assertEquals("ws2.update-1 label", "text", entity.getFieldForLabel());

        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(Timestamp.valueOf("2020-02-17 10:11:12.123"), "Insertado DDD y modificado BBB a BB2", null);

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

        String localRevisionCode = ws2.getWorkspaceEntity("test").getRepositoryRevisionCode();

        assertEquals("ws2 localRevisionCode", "00000000000015abcdef0123456789", localRevisionCode);

        ws1.dispose();

        ws2.dispose();

        ws1 = null;

        ws2 = null;

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

        // Check that the UpdateRequestClient produce the expected json output

        Map<String, String> expected = TestUtils.getExpecteds("S05Update.txt");

        VCSGisRepository repoRemote = manager.openRepository(serverUrl);

        UpdateRequestClient request = (UpdateRequestClient) repoRemote.createUpdateRequest("test");

        request.setLocalRevisionCode(localRevisionCode);

        ByteArrayOutputStream request_stream = new ByteArrayOutputStream(10024);

        request.requestProducer(request_stream);

        String request_s = request_stream.toString().trim();

        assertEquals("request data", expected.get("request"), request_s);

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

        // Check that CheckoutHandler produce the expected json as response to request

        ByteArrayInputStream is = new ByteArrayInputStream(request_stream.toByteArray());

        VCSGisServerHandler handler = new UpdateHandler(repoLocaldb);

        InputStream response_stream = handler.handle(is);

        String response_s = StringUtils.join(IOUtils.readLines(response_stream),'\n').trim();

        assertEquals("response data", expected.get("response"), response_s);

    }

}