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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestS01Entities extends TestCase {

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

    public TestS01Entities(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 testCheckout() throws Exception {

        final String testid = "ServerEntities";

        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 the repository

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

        ws1.commit();

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

        DisposeUtils.disposeQuietly(store2);

        VCSGisRepository client = manager.openRepository(serverUrl);

        EntitiesRequestClient requestClient = (EntitiesRequestClient) client.createEntitiesRequest();

        EntitiesHandler handler = new EntitiesHandler(repo);

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

        ByteArrayOutputStream request_stream = new ByteArrayOutputStream(1024);

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

        requestClient.requestProducer(request_stream);

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

        assertEquals("request data", expected.get("request"), request_stream.toString());

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

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

        Writer response_writer = new ByteArrayWriter();

        handler.handle(null, null, is, response_writer);

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

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

    }

}