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.Writer;

import java.net.URL;

import java.util.ArrayList;

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.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.VCSGisRepositoryData;

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

import org.gvsig.vcsgis.lib.repository.requests.VCSGisCheckoutRequest;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisEntitiesRequest;

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

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

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestS02Commit extends TestCase {

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

    public TestS02Commit(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 testCommit() throws Exception {

        final String testid = "CommitHandler";

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

        VCSGisManager manager = VCSGisLocator.getVCSGisManager();

        manager.setCodeGenerator(new SequentialCodeGenerator());

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

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

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

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

        VCSGisRepository repo = manager.openRepository(server);

        TestUtils.h2sql_repository(repo);

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

        // Prepare the data in workspace

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

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

        VCSGisWorkspace ws1 = manager.openWorkspace(ws1file);

        assertNotNull("openWorkspace status", ws1);

        TestUtils.h2sql_workspace(ws1);

        FeatureStore store2 = TestUtils.openSourceStore2();

        ws1.add("test", store2, "text", null, "test");

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

        DisposeUtils.disposeQuietly(store2);

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

        // Check that in ws are the expected changes

        VCSGisWorkspaceChanges<VCSGisWorkspaceChange> wschanges = ws1.getLocalChanges();

        assertEquals("Changes before commit", 4, wschanges.size64());

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

        // Check that in the repo are not tables

        VCSGisEntitiesRequest localdbEntityRequest = repo.createEntitiesRequest();

        localdbEntityRequest.execute();

        List<VCSGisEntity> entities = localdbEntityRequest.getRepositoryEntities();

        assertEquals("Entities in repo before commit", 0, entities.size());

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

        // Check that the CommitRequestClient produce the expected json output

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

        VCSGisRepository client = manager.openRepository(serverUrl);

        CommitRequestClient request = (CommitRequestClient) client.createCommitRequest();

        ((VCSGisWorkspaceImpl)ws1).prepareCommitRequest(request, null, null, null, null, null);

        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 CommitHandler produce the expected json as response to request

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

        VCSGisServerHandler handler = new CommitHandler(repo);

        Writer response_writer = new ByteArrayWriter();

        handler.handle(null, is, response_writer);

        String response_s = response_writer.toString().trim();

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

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

        // Check that in the repository are created the new table

        localdbEntityRequest = repo.createEntitiesRequest();

        localdbEntityRequest.execute();

        entities = localdbEntityRequest.getRepositoryEntities();

        assertEquals("Entities in repo after commit", 1, entities.size());

        VCSGisEntity entity = entities.get(0);

        assertEquals("Entity[0].name in repo after commit", "test", entity.getEntityName());

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

        // Check that the table in the repository have the expecteds values

        VCSGisCheckoutRequest localdbCheckoutRequest = repo.createCheckoutRequest("test");

        localdbCheckoutRequest.execute();

        List<VCSGisRepositoryData> data = new ArrayList<>();

        for (VCSGisRepositoryData repoData : localdbCheckoutRequest.getData()) {

            data.add(repoData);

        }            

        assertEquals("Num rows in repo after commit", 3, data.size());

    }

}