org.neo4j.mytests.SSShortestPathTest.java Source code

Introduction

Here is the source code for org.neo4j.mytests.SSShortestPathTest.java

Source

/**
 * The MIT License (MIT)
 *
 * Copyright (c) 2014 Francisco G.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
package org.neo4j.mytests;

import com.sun.jersey.api.client.Client;
import com.sun.jersey.api.client.ClientResponse;
import com.sun.jersey.api.client.config.DefaultClientConfig;
import org.codehaus.jackson.JsonNode;
import org.codehaus.jackson.jaxrs.JacksonJsonProvider;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mortbay.jetty.LocalConnector;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.graphdb.Node;
import org.neo4j.graphdb.Transaction;
import org.neo4j.kernel.GraphDatabaseAPI;
import org.neo4j.server.CommunityNeoServer;
import org.neo4j.server.helpers.CommunityServerBuilder;
import org.neo4j.graphdb.RelationshipType;
import org.neo4j.graphalgo.*;
import org.neo4j.tooling.GlobalGraphOperations;
import java.io.IOException;
import java.net.ServerSocket;
import static junit.framework.Assert.assertEquals;
import java.net.HttpURLConnection;
import java.net.URL;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import org.json.JSONObject;

public class SSShortestPathTest {
    private GraphDatabaseAPI db;
    private CommunityNeoServer server;

    enum MyRelationshipTypes implements RelationshipType {
        CONTAINED_IN, KNOWS, HAS_TERM
    }

    @Before
    public void before() throws IOException {
        ServerSocket serverSocket = new ServerSocket(0);
        server = CommunityServerBuilder.server().onPort(serverSocket.getLocalPort())
                .withThirdPartyJaxRsPackage("org.neo4j.hintplugin.utils", "/hintplugin/utils").build();
        server.start();
        db = server.getDatabase().getGraph();
    }

    @After
    public void after() {
        server.stop();
    }

    @Test
    public void shouldReturnMaxFlow() {
        Transaction tx = db.beginTx();
        try {
            Node a = db.createNode();
            Node b = db.createNode();
            Node c = db.createNode();
            Node d = db.createNode();
            Node e = db.createNode();
            a.createRelationshipTo(e, MyRelationshipTypes.KNOWS);
            a.createRelationshipTo(b, MyRelationshipTypes.KNOWS);
            e.createRelationshipTo(b, MyRelationshipTypes.KNOWS);
            e.createRelationshipTo(d, MyRelationshipTypes.KNOWS);
            b.createRelationshipTo(c, MyRelationshipTypes.KNOWS);
            c.createRelationshipTo(d, MyRelationshipTypes.KNOWS);

        } catch (Exception e) {
            System.err.println("Exception Error: MaxflowTest.shouldReturnMaxFlow: " + e);
            tx.failure();
        } finally {
            tx.success();
            tx.close();
        }

        String serverBaseUri = server.baseUri().toString();
        URL uriArray[] = new URL[6];
        String q1 = serverBaseUri + "hintplugin/utils/ssshortestpath/0/2"; // a-c 6
        String q2 = serverBaseUri + "hintplugin/utils/ssshortestpath/0/3"; // a-d 4
        String q3 = serverBaseUri + "hintplugin/utils/ssshortestpath/0/4"; // a-e 2
        String q4 = serverBaseUri + "hintplugin/utils/ssshortestpath/2/3"; // c-d 4
        String q5 = serverBaseUri + "hintplugin/utils/ssshortestpath/2/4"; // c-e 2
        String q6 = serverBaseUri + "hintplugin/utils/ssshortestpath/3/4"; // d-e 2

        try {
            uriArray[0] = new URL(q1);
            uriArray[1] = new URL(q2);
            uriArray[2] = new URL(q3);
            uriArray[3] = new URL(q4);
            uriArray[4] = new URL(q5);
            uriArray[5] = new URL(q6);
        } catch (Exception ex) {
            System.out.println("***** ERROR: " + ex);
        }
        //Establish a connection to the server and get Content.
        for (int i = 0; i < uriArray.length; i++) {
            try {
                HttpURLConnection http = (HttpURLConnection) uriArray[i].openConnection();
                http.setRequestMethod("GET");
                http.connect();
                StringBuffer text = new StringBuffer();
                InputStreamReader in = new InputStreamReader((InputStream) http.getContent());
                BufferedReader buff = new BufferedReader(in);
                String line = "";
                while (line != null) {
                    line = buff.readLine();
                    text.append(line + " ");
                }
                JSONObject obj = new JSONObject(text.toString());
                System.out.println("***SSSPTH-JSON: " + obj.optDouble("shortestPaths"));
            } catch (Exception ex) {
                System.out.println("MaxflowTest Exception: " + ex);
            }
        }
        assertEquals("200", "200");
    }

    private Client jerseyClient() {
        DefaultClientConfig defaultClientConfig = new DefaultClientConfig();
        defaultClientConfig.getClasses().add(JacksonJsonProvider.class);
        return Client.create(defaultClientConfig);
    }
}