1 files changed, 425 insertions, 0 deletions
diff --git a/verigraph/src/it/polito/verigraph/grpc/tosca/test/GrpcToscaTest.java b/verigraph/src/it/polito/verigraph/grpc/tosca/test/GrpcToscaTest.java
new file mode 100644
index 0000000..5771406
--- /dev/null
+++ b/verigraph/src/it/polito/verigraph/grpc/tosca/test/GrpcToscaTest.java
@@ -0,0 +1,425 @@
+/*******************************************************************************
+ * Copyright (c) 2018 Politecnico di Torino and others.
+ *
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Apache License, Version 2.0
+ * which accompanies this distribution, and is available at
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *******************************************************************************/
+package it.polito.verigraph.grpc.tosca.test;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.fail;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.FixMethodOrder;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+import org.junit.runners.MethodSorters;
+
+import it.polito.verigraph.grpc.NewTopologyTemplate;
+import it.polito.verigraph.grpc.NodeTemplateGrpc;
+import it.polito.verigraph.grpc.NodeTemplateGrpc.Type;
+import it.polito.verigraph.grpc.RelationshipTemplateGrpc;
+import it.polito.verigraph.grpc.Status;
+import it.polito.verigraph.grpc.TopologyTemplateGrpc;
+import it.polito.verigraph.grpc.ToscaConfigurationGrpc;
+import it.polito.verigraph.grpc.ToscaPolicy;
+import it.polito.verigraph.grpc.ToscaPolicy.PolicyType;
+import it.polito.verigraph.grpc.ToscaVerificationGrpc;
+import it.polito.verigraph.grpc.client.ToscaClient;
+import it.polito.verigraph.grpc.server.Service;
+
+@RunWith(JUnit4.class)
+@FixMethodOrder(MethodSorters.NAME_ASCENDING)
+public class GrpcToscaTest {
+    private Service server;
+    private ToscaClient client;
+    private TopologyTemplateGrpc testTemplate, simpleTestTemplate;
+
+    public GrpcToscaTest() {
+        this.generateTestTemplate();
+    }
+
+    @Before
+    public void setUpBeforeClass() throws Exception {
+        client = new ToscaClient("localhost" , 50051);
+        server = new Service(50051);
+        server.start();
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        server.stop();
+        client.shutdown();
+    }
+
+
+    @Test
+    public void test0Creation() {
+        System.out.println("\nTest A: Graph Creation.");
+
+        NewTopologyTemplate response = client.createTopologyTemplate(testTemplate);
+        assertNotNull("Returned a NULL graph", response);
+        assertEquals(response.getSuccess(), true);
+        assertEquals("Error report: " + response.getErrorMessage(), "", response.getErrorMessage());
+
+        Status resp = client.deleteTopologyTemplate(response.getTopologyTemplate().getId());
+        assertEquals("Error while deleting testTemplate.", true, resp.getSuccess());
+
+        System.out.println("Test A completed\n");
+
+        return;
+    }
+
+
+    @Test
+    public void test1Reading() {
+        System.out.println("\nTest B: Graph Reading.");
+
+        //Creating a test graph on remote repository
+        System.out.println("Phase B.1 -- Creating a test graph.");
+        NewTopologyTemplate response = client.createTopologyTemplate(simpleTestTemplate);
+        assertNotNull("Returned a NULL graph", response);
+        assertEquals(true, response.getSuccess());
+        assertEquals("Error report: " + response.getErrorMessage(), "", response.getErrorMessage());
+
+        //Reading remote graph.
+        System.out.println("Phase B.2 -- Reading remote graph.");
+        TopologyTemplateGrpc retrieved = client.getTopologyTemplate(response.getTopologyTemplate().getId());
+        assertNotNull("Retrieved a NULL graph", retrieved);
+        assertEquals(retrieved.getId(), response.getTopologyTemplate().getId());
+
+        //Nodes checking
+        System.out.println("Phase B.3 -- Checking graph's nodes.");
+        assertEquals(retrieved.getNodeTemplateCount(), 3);
+        assertEquals("Node1 name error", response.getTopologyTemplate().getNodeTemplateList().get(0).getName(),
+                retrieved.getNodeTemplateList().get(0).getName());
+        assertEquals("Node2 name error", response.getTopologyTemplate().getNodeTemplateList().get(1).getName(),
+                retrieved.getNodeTemplateList().get(1).getName());
+        assertEquals("Node3 name error", response.getTopologyTemplate().getNodeTemplateList().get(2).getName(),
+                retrieved.getNodeTemplateList().get(2).getName());
+
+        //Relationships checking
+        System.out.println("Phase B.4 -- Checking graph's relationships.");
+        assertEquals(retrieved.getRelationshipTemplateCount(), 4);
+        String source1=null, target1=null;
+        String source2=null, target2=null;
+        String source3=null, target3=null;
+        String source4=null, target4=null;
+        for (NodeTemplateGrpc node : retrieved.getNodeTemplateList()){
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(0).getIdSourceNodeTemplate()))
+                source1=node.getName();
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(0).getIdTargetNodeTemplate()))
+                target1=node.getName();
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(1).getIdSourceNodeTemplate()))
+                source2=node.getName();
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(1).getIdTargetNodeTemplate()))
+                target2=node.getName();
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(2).getIdSourceNodeTemplate()))
+                source3=node.getName();
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(2).getIdTargetNodeTemplate()))
+                target3=node.getName();
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(3).getIdSourceNodeTemplate()))
+                source4=node.getName();
+            if(node.getId().equals(retrieved.getRelationshipTemplateList().get(3).getIdTargetNodeTemplate()))
+                target4=node.getName();
+        }
+
+        assertEquals("Relat1 name error", retrieved.getRelationshipTemplateList().get(0).getName(), source1+"to"+target1);
+        assertEquals("Relat2 name error", retrieved.getRelationshipTemplateList().get(1).getName(), source2+"to"+target2);
+        assertEquals("Relat3 name error", retrieved.getRelationshipTemplateList().get(2).getName(), source3+"to"+target3);
+        assertEquals("Relat4 name error", retrieved.getRelationshipTemplateList().get(3).getName(), source4+"to"+target4);
+
+        System.out.println("Phase B.5 -- Deleting graph.");
+        Status resp = client.deleteTopologyTemplate(response.getTopologyTemplate().getId());
+        assertEquals("Error while deleting simpleTestTemplate", true, resp.getSuccess());
+
+
+        System.out.println("Test B completed.\n");
+        return;
+    }
+
+    @Test
+    public void test2Update() {
+        System.out.println("\nTest C: Update.");
+
+        //Creating a test graph on remote repository
+        System.out.println("Phase C.1 -- Creating a test graph.");
+        NewTopologyTemplate response = client.createTopologyTemplate(simpleTestTemplate);
+        assertNotNull("Returned a NULL graph", response);
+        assertEquals(true, response.getSuccess());
+        assertEquals("Error report: " + response.getErrorMessage(), "", response.getErrorMessage());
+
+        //Reading remote graph.
+        System.out.println("Phase C.2 -- Reading remote graph.");
+        TopologyTemplateGrpc retrieved = client.getTopologyTemplate(response.getTopologyTemplate().getId());
+        assertNotNull("Retrieved a NULL graph", retrieved);
+        assertEquals(retrieved.getId(), response.getTopologyTemplate().getId());
+
+        //Updating a TopologyTemplateGrpc
+        System.out.println("Phase C.3 -- Updating a test graph.");
+        TopologyTemplateGrpc.Builder templ = TopologyTemplateGrpc.newBuilder();
+        List<NodeTemplateGrpc> nodes = new ArrayList<NodeTemplateGrpc>();
+        List<RelationshipTemplateGrpc> relats = new ArrayList<RelationshipTemplateGrpc>();
+
+        ToscaConfigurationGrpc node1conf = ToscaConfigurationGrpc.newBuilder().setDescription("node1configuration")
+                .setId("15").setConfiguration("[]").build();
+        NodeTemplateGrpc node1 = NodeTemplateGrpc.newBuilder().setConfiguration(node1conf).setId("999")
+                .setName("webserver1").setType(Type.webserver).build();
+        nodes.add(node1);
+
+        ToscaConfigurationGrpc node2conf = ToscaConfigurationGrpc.newBuilder().setDescription("node2configuration")
+                .setId("16").setConfiguration("[{\r\n\"protocol\":\"HTTP_REQUEST\",\r\n \"url\":\"www.facebook.com\"\r\n }]").build();
+        NodeTemplateGrpc node2 = NodeTemplateGrpc.newBuilder().setConfiguration(node2conf).setId("888")
+                .setName("host2").setType(Type.endhost).build();
+        nodes.add(node2);
+
+        RelationshipTemplateGrpc rel0 = RelationshipTemplateGrpc.newBuilder().setId("1001")
+                .setIdSourceNodeTemplate("999").setIdTargetNodeTemplate("888").setName("webserver1tohost2").build();
+        relats.add(rel0);
+
+        RelationshipTemplateGrpc rel1 = RelationshipTemplateGrpc.newBuilder().setId("1002")
+                .setIdSourceNodeTemplate("888").setIdTargetNodeTemplate("999").setName("host2towebserver1").build();
+        relats.add(rel1);
+
+        TopologyTemplateGrpc newTestTemplate = templ.addAllNodeTemplate(nodes).addAllRelationshipTemplate(relats).setId("9").build();
+        NewTopologyTemplate updated = client.updateTopologyTemplate(newTestTemplate, response.getTopologyTemplate().getId());
+        assertNotNull("Returned a NULL graph", updated);
+        assertEquals(true, updated.getSuccess());
+        assertEquals("Error report: " + updated.getErrorMessage(), "", updated.getErrorMessage());
+
+        //Reading remote graph.
+        System.out.println("Phase C.4 -- Reading remote graph.");
+        TopologyTemplateGrpc retrieved2 = client.getTopologyTemplate(response.getTopologyTemplate().getId());
+        assertNotNull("Retrieved a NULL graph", retrieved2);
+        assertEquals(retrieved2.getId(), response.getTopologyTemplate().getId());
+
+        //Nodes checking
+        System.out.println("Phase C.5 -- Checking updated graph's nodes.");
+        assertEquals(retrieved2.getNodeTemplateCount(), 2);
+        assertEquals("Node1 name error", updated.getTopologyTemplate().getNodeTemplateList().get(0).getName(),
+                retrieved2.getNodeTemplateList().get(0).getName());
+        assertEquals("Node2 name error", updated.getTopologyTemplate().getNodeTemplateList().get(1).getName(),
+                retrieved2.getNodeTemplateList().get(1).getName());
+
+        //Relationships checking
+        System.out.println("Phase C.6 -- Checking updated graph's relationships.");
+        assertEquals(retrieved2.getRelationshipTemplateCount(), 2);
+        String source1=null, target1=null;
+        String source2=null, target2=null;
+        for (NodeTemplateGrpc node : retrieved2.getNodeTemplateList()){
+            if(node.getId().equals(retrieved2.getRelationshipTemplateList().get(0).getIdSourceNodeTemplate()))
+                source1=node.getName();
+            if(node.getId().equals(retrieved2.getRelationshipTemplateList().get(0).getIdTargetNodeTemplate()))
+                target1=node.getName();
+            if(node.getId().equals(retrieved2.getRelationshipTemplateList().get(1).getIdSourceNodeTemplate()))
+                source2=node.getName();
+            if(node.getId().equals(retrieved2.getRelationshipTemplateList().get(1).getIdTargetNodeTemplate()))
+                target2=node.getName();
+        }
+
+        assertEquals("Relat1 name error", retrieved2.getRelationshipTemplateList().get(0).getName(), source1+"to"+target1);
+        assertEquals("Relat2 name error", retrieved2.getRelationshipTemplateList().get(1).getName(), source2+"to"+target2);
+
+        System.out.println("Phase C.6 -- Deleting graph.");
+        Status resp = client.deleteTopologyTemplate(updated.getTopologyTemplate().getId());
+        assertEquals("Error while deleting simpleTestTemplate", true, resp.getSuccess());
+
+        System.out.println("Test C completed.\n");
+        return;
+    }
+
+
+    @Test
+    public void test3Verification() {
+        System.out.println("\nTest D: Verification.");
+        NewTopologyTemplate response = client.createTopologyTemplate(testTemplate);
+        if(response == null | response.getSuccess() != true) {
+            fail("Test failed, unable to load the graph.");
+            return;
+        }
+
+        //The Id of the graph on which we are going to perform tests
+        String testTemplateId = response.getTopologyTemplate().getId();
+
+        //REACHABILITY test
+        System.out.println("Phase 1.1 - Reachability SAT.");
+        ToscaPolicy policy = ToscaPolicy.newBuilder().setIdTopologyTemplate(testTemplateId)
+                .setType(PolicyType.reachability).setSource("host2").setDestination("host1").build();
+        ToscaVerificationGrpc result = client.verifyPolicy(policy);
+        assertNotNull("There was no response", result);
+        assertEquals("Unexpected result : " + result.getResult() + " - " + result.getComment(), "SAT", result.getResult());
+        assertEquals("Error report: " + result.getErrorMessage(), "", result.getErrorMessage());
+
+        result = null;
+        System.out.println("Phase 1.2 - Reachability UNSAT.");
+        policy = ToscaPolicy.newBuilder().setIdTopologyTemplate(testTemplateId)
+                .setType(PolicyType.reachability).setSource("host1").setDestination("antispamNode1").build();
+        result = client.verifyPolicy(policy);
+        assertNotNull("There was no response", result);
+        assertEquals("Unexpected result : " + result.getResult() + " - " + result.getComment(), "UNSAT", result.getResult());
+        assertEquals("Error report: " + result.getErrorMessage(), "", result.getErrorMessage());
+
+        //ISOLATION test
+        result = null;
+        System.out.println("Phase 2.1 - Isolation SAT.");
+        policy = ToscaPolicy.newBuilder().setIdTopologyTemplate(testTemplateId)
+                .setType(PolicyType.isolation).setSource("host2").setDestination("host1").setMiddlebox("webserver1").build();
+        result = client.verifyPolicy(policy);
+        assertNotNull("There was no response", result);
+        assertEquals("Unexpected result : " + result.getResult() + " - " + result.getComment(), "SAT", result.getResult());
+        assertEquals("Error report: " + result.getErrorMessage(), "", result.getErrorMessage());
+
+        System.out.println("Phase 2.2 - Isolation UNSAT.");
+        policy = ToscaPolicy.newBuilder().setIdTopologyTemplate(testTemplateId)
+                .setType(PolicyType.isolation).setSource("host2").setDestination("host1").setMiddlebox("fw").build();
+        result = client.verifyPolicy(policy);
+        assertNotNull("There was no response", result);
+        assertEquals("Unexpected result : " + result.getResult() + " - " + result.getComment(), "UNSAT", result.getResult());
+        assertEquals("Error report: " + result.getErrorMessage(), "", result.getErrorMessage());
+
+        //TRAVERSAL test
+        result = null;
+        System.out.println("Phase 3.1 - Traversal SAT.");
+        policy = ToscaPolicy.newBuilder().setIdTopologyTemplate(testTemplateId)
+                .setType(PolicyType.traversal).setSource("host2").setDestination("host1").setMiddlebox("fw").build();
+        result = client.verifyPolicy(policy);
+        assertNotNull("There was no response", result);
+        assertEquals("Unexpected result : " + result.getResult() + " - " + result.getComment(), "SAT", result.getResult());
+        assertEquals("Error report: " + result.getErrorMessage(), "", result.getErrorMessage());
+
+        System.out.println("Phase 3.2 - Traversal UNSAT.");
+        policy = ToscaPolicy.newBuilder().setIdTopologyTemplate(testTemplateId)
+                .setType(PolicyType.traversal).setSource("host2").setDestination("webserver1").setMiddlebox("fw").build();
+        result = client.verifyPolicy(policy);
+        assertNotNull("There was no response", result);
+        assertEquals("Unexpected result : " + result.getResult() + " - " + result.getComment(), "UNSAT", result.getResult());
+        assertEquals("Error report: " + result.getErrorMessage(), "", result.getErrorMessage());
+
+        Status resp = client.deleteTopologyTemplate(testTemplateId);
+        assertEquals("Error while deleting testTemplate", true, resp.getSuccess());
+
+        System.out.println("Test D completed.\n");
+        return;
+    }
+
+
+    @Test
+    public void test4Deletion() {
+        System.out.println("\nTest E: Deletion");
+        NewTopologyTemplate templ = client.createTopologyTemplate(testTemplate);
+
+        if(templ.getSuccess() != true) {
+            fail("Unable to create the graph.");
+            return;
+        }else {
+            Status resp = client.deleteTopologyTemplate(templ.getTopologyTemplate().getId());
+            assertEquals("Error while deleting testTemplate", true, resp.getSuccess());
+        }
+
+        System.out.println("Test E completed.\n");
+        return;
+    }
+
+
+    //Generates a correct instance of a TopologyTemplateGrpc to be used in tests
+    public void generateTestTemplate() {
+        TopologyTemplateGrpc.Builder templ = TopologyTemplateGrpc.newBuilder();
+        List<NodeTemplateGrpc> nodes = new ArrayList<NodeTemplateGrpc>();
+        List<RelationshipTemplateGrpc> relats = new ArrayList<RelationshipTemplateGrpc>();
+
+        //Definition of nodes
+        ToscaConfigurationGrpc node0conf = ToscaConfigurationGrpc.newBuilder().setDescription("node0configuration")
+                .setId("100").setConfiguration("[{\r\n\"webserver1\":\"host2\"\r\n}]").build();
+        NodeTemplateGrpc node0 = NodeTemplateGrpc.newBuilder().setConfiguration(node0conf).setId("100")
+                .setName("fw").setType(Type.firewall).build();
+        nodes.add(node0);
+
+        ToscaConfigurationGrpc node1conf = ToscaConfigurationGrpc.newBuilder().setDescription("node1configuration")
+                .setId("101").setConfiguration("[]").build();
+        NodeTemplateGrpc node1 = NodeTemplateGrpc.newBuilder().setConfiguration(node1conf).setId("101")
+                .setName("webserver1").setType(Type.webserver).build();
+        nodes.add(node1);
+
+        ToscaConfigurationGrpc node2conf = ToscaConfigurationGrpc.newBuilder().setDescription("node2configuration")
+                .setId("102").setConfiguration("[{\r\n\"protocol\":\"HTTP_REQUEST\",\r\n \"url\":\"www.facebook.com\"\r\n }]").build();
+        NodeTemplateGrpc node2 = NodeTemplateGrpc.newBuilder().setConfiguration(node2conf).setId("102")
+                .setName("host2").setType(Type.endhost).build();
+        nodes.add(node2);
+
+        ToscaConfigurationGrpc node3conf = ToscaConfigurationGrpc.newBuilder().setDescription("node3configuration")
+                .setId("103").setConfiguration("[ {\r\n\"protocol\":\"HTTP_REQUEST\",\r\n\"url\":\"www.google.com\",\r\n\"destination\":\"server1\"\r\n}]").build();
+        NodeTemplateGrpc node3 = NodeTemplateGrpc.newBuilder().setConfiguration(node3conf).setId("103")
+                .setName("host1").setType(Type.endhost).build();
+        nodes.add(node3);
+
+        ToscaConfigurationGrpc node4conf = ToscaConfigurationGrpc.newBuilder().setDescription("node4configuration")
+                .setId("104").setConfiguration("[\"host1\",\"host2\"]").build();
+        NodeTemplateGrpc node4 = NodeTemplateGrpc.newBuilder().setConfiguration(node4conf).setId("104")
+                .setName("antispamNode1").setType(Type.antispam).build();
+        nodes.add(node4);
+
+        //Building relationships
+        RelationshipTemplateGrpc rel0 = RelationshipTemplateGrpc.newBuilder().setId("1001")
+                .setIdSourceNodeTemplate("100").setIdTargetNodeTemplate("101").setName("fwToServ1").build();
+        relats.add(rel0);
+
+        RelationshipTemplateGrpc rel1 = RelationshipTemplateGrpc.newBuilder().setId("1002")
+                .setIdSourceNodeTemplate("101").setIdTargetNodeTemplate("100").setName("serv1ToFw").build();
+        relats.add(rel1);
+
+        RelationshipTemplateGrpc rel2 = RelationshipTemplateGrpc.newBuilder().setId("1003")
+                .setIdSourceNodeTemplate("100").setIdTargetNodeTemplate("103").setName("fwToHost1").build();
+        relats.add(rel2);
+
+        RelationshipTemplateGrpc rel3 = RelationshipTemplateGrpc.newBuilder().setId("1004")
+                .setIdSourceNodeTemplate("100").setIdTargetNodeTemplate("102").setName("fwToHost2").build();
+        relats.add(rel3);
+
+        RelationshipTemplateGrpc rel4 = RelationshipTemplateGrpc.newBuilder().setId("1005")
+                .setIdSourceNodeTemplate("102").setIdTargetNodeTemplate("100").setName("Host2Tofw").build();
+        relats.add(rel4);
+
+        RelationshipTemplateGrpc rel5 = RelationshipTemplateGrpc.newBuilder().setId("1006")
+                .setIdSourceNodeTemplate("103").setIdTargetNodeTemplate("100").setName("Host1Tofw").build();
+        relats.add(rel5);
+
+        this.testTemplate = templ.addAllNodeTemplate(nodes).addAllRelationshipTemplate(relats).setId("0").build();
+
+        TopologyTemplateGrpc.Builder templ2 = TopologyTemplateGrpc.newBuilder();
+        List<NodeTemplateGrpc> nodes2 = new ArrayList<NodeTemplateGrpc>();
+        List<RelationshipTemplateGrpc> relats2 = new ArrayList<RelationshipTemplateGrpc>();
+        nodes2.add(node0);
+        nodes2.add(node1);
+        nodes2.add(node2);
+        relats2.add(rel0);
+        relats2.add(rel1);
+        relats2.add(rel3);
+        relats2.add(rel4);
+
+        this.simpleTestTemplate = templ2.addAllNodeTemplate(nodes2).addAllRelationshipTemplate(relats2).setId("1").build();
+
+    }
+
+    /*class NodeTemplateGrpcComparator implements Comparator<NodeTemplateGrpc> {
+        public int compare(NodeTemplateGrpc n0, NodeTemplateGrpc n1) {
+            return n0.getName().compareTo(n1.getName());
+        }
+    }
+
+    class RelationshipTemplateGrpcComparator implements Comparator<RelationshipTemplateGrpc> {
+        public int compare(RelationshipTemplateGrpc n0, RelationshipTemplateGrpc n1) {
+            int source = n0.getIdSourceNodeTemplate().compareTo(n1.getIdSourceNodeTemplate());
+            int target = n0.getIdTargetNodeTemplate().compareTo(n1.getIdTargetNodeTemplate());
+            return source + target;
+        }
+    }*/
+
+}