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.util.Map;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

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.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.remoteclient.requests.CommitRequestClient;

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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 = "ServerCommit";

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

        VCSGisManager manager = VCSGisLocator.getVCSGisManager();

        JDBCServerExplorer server = TestUtils.openServerExplorer("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");

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

        DisposeUtils.disposeQuietly(store2);

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

        ByteArrayOutputStream request_stream = new ByteArrayOutputStream(1024);

        LOGGER.debug("===: "+testid+"/produce request");

        request.requestDataProducer(request_stream);

        LOGGER.debug("===: "+testid+"/request built");

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

        assertTrue("request data", match(expected.get("request"), request_s));

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

        LOGGER.debug("===: "+testid+"/handle request");

        CommitHandler handler = new CommitHandler(repo);

        InputStream response_stream = handler.handle(is);

        LOGGER.debug("===: "+testid+"/Transform response to string");

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

        assertTrue("response data", match(expected.get("response"), response_s));

    }

}