src/test/java/de/ids_mannheim/korap/server/TestResource.java

package de.ids_mannheim.korap.server;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.client.Entity;

import org.glassfish.grizzly.http.server.HttpServer;

import static org.junit.Assert.*;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.FileInputStream;

import de.ids_mannheim.korap.server.Node;
import de.ids_mannheim.korap.response.Result;
import de.ids_mannheim.korap.response.Response;
import static de.ids_mannheim.korap.util.KrillString.*;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

/**
 * @author diewald
 */
// http://harryjoy.com/2012/09/08/simple-rest-client-in-java/
public class TestResource {
    private HttpServer server;
    private WebTarget target;

    final ObjectMapper mapper = new ObjectMapper();
    long t1 = 0, t2 = 0, t3 = 0, t4 = 0;


    @Before
    public void setUp () throws Exception {
        // start the server
        t1 = System.nanoTime();
        server = Node.startServer(new String[] { "--name", "milena", "--dir",
                ":memory:", "--port", "9157" });
        // create the client
        Client c = ClientBuilder.newClient();

        t2 = System.nanoTime();

        // uncomment the following line if you want to enable
        // support for JSON in the client (you also have to uncomment
        // dependency on jersey-media-json module in pom.xml and Main.startServer())
        // --

        // c.configuration().enable(com.sun.jersey.api.json.POJOMappingFeature());
        // c.configuration().enable(new org.glassfish.jersey.media.json.JsonJaxbFeature());
        // c.register(JacksonFeatures.class);
        // c.register(com.fasterxml.jackson.jaxrs.annotation.JacksonFeatures.class);

        /*
        ClientConfig clientConfig = new DefaultClientConfig();
        clientConfig.getFeatures().put(JSONConfiguration.FEATURE_POJO_MAPPING, Boolean.TRUE);
        Client c = Client.create(clientConfig);
        */
        target = c.target(Node.getBaseURI());
    };


    @After
    public void tearDown () throws Exception {
        t3 = System.nanoTime();
        server.stop();
        Node.closeDBPool();
        t4 = System.nanoTime();

        double startup = (double) (t2 - t1) / 1000000000.0;
        double action = (double) (t3 - t2) / 1000000000.0;
        double shutdown = (double) (t4 - t3) / 1000000000.0;

        /*
        System.err.println("Startup:  " + startup + ", " +
                           "Action:   " + action  + ", " +
                           "Shutdown: " + shutdown);
        */
    };


    /**
     * Test to see that the message "Gimme 5 minutes, please!" is sent
     * in the response.
     */
    @Test
    public void testPing () {
        String responseMsg = target.path("ping").request().get(String.class);
        assertEquals("Gimme 5 minutes, please!", responseMsg);
    };


    // This tests the node info
    @Test
    public void testInfo () throws IOException {
        String responseMsg = target.path("/").request().get(String.class);
        JsonNode res = mapper.readTree(responseMsg);
        assertEquals("milena", res.at("/meta/node").asText());
        assertEquals(680, res.at("/messages/0/0").asInt());
    };


    @Test
    public void testIndexing () throws IOException {
        String resp;
        JsonNode res;

        for (String i : new String[] { "00001", "00002", "00003", "00004",
                "00005", "00006" }) {

            String json = StringfromFile(getClass().getResource(
                    "/wiki/" + i + ".json").getFile());

            Entity jsonE = Entity.json(json);

            try {
                // Put new documents to the index
                resp = target.path("/index/" + i).request("application/json")
                        .put(jsonE, String.class);

                res = mapper.readTree(resp);
                assertEquals("milena", res.at("/meta/node").asText());
            }
            catch (Exception e) {
                fail("Server response failed " + e.getMessage()
                        + " (Known issue)");
            }
        };

        String json = StringfromFile(getClass().getResource("/wiki/02439.json")
                .getFile());
        Entity jsonE = Entity.json(json);

        try {
            // Put new documents to the index
            resp = target.path("/index/02439").request("application/json")
                    .put(jsonE, String.class);

            res = mapper.readTree(resp);

            // Check mirroring
            assertEquals(2439, res.at("/text/UID").asInt());
            assertEquals("milena", res.at("/meta/node").asText());
        }
        catch (Exception e) {
            fail("Server response failed " + e.getMessage() + " (Known issue)");
        };

        // Commit!
        resp = target.path("/index").request("application/json")
                .post(Entity.text(""), String.class);
        res = mapper.readTree(resp);
        assertEquals("milena", res.at("/meta/node").asText());
        assertEquals(683, res.at("/messages/0/0").asInt());
    };


    @Test
    public void testCollection () throws IOException {

        // mate/l:sein
        String json = getString(getClass().getResource(
                "/queries/bsp-uid-example.jsonld").getFile());

        try {
            String resp = target.path("/").queryParam("uid", "1")
                    .queryParam("uid", "4").request("application/json")
                    .post(Entity.json(json), String.class);
            JsonNode res = mapper.readTree(resp);
            assertEquals(2, res.at("/meta/totalResults").asInt());
        }
        catch (Exception e) {
            fail("Server response failed: " + e.getMessage() + " (Known issue)");
        };

    };


    public static String getString (String path) {
        StringBuilder contentBuilder = new StringBuilder();
        try {
            BufferedReader in = new BufferedReader(new FileReader(path));
            String str;
            while ((str = in.readLine()) != null) {
                contentBuilder.append(str);
            };
            in.close();
        }
        catch (IOException e) {
            fail(e.getMessage());
        };
        return contentBuilder.toString();
    };
};