Gitiles
Code Review Sign In
gerrit.onosproject.org / spring-open / c86a2ea2886749956aed79c5b7ff0433b5e44915 / . / src / test / java / net / floodlightcontroller / linkdiscovery / internal / LinkStorageImplTest.java
blob: a4b24294c8b881b7adc168650f6abb2fafe921cc [file] [log] [blame]
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]1package net.floodlightcontroller.linkdiscovery.internal;
2
3import java.io.ByteArrayInputStream;
4import java.io.File;
5import java.io.IOException;
6import java.io.InputStream;
7import java.util.Iterator;
Jonathan Hartc86a2ea2013-01-15 22:39:42 -0800[diff] [blame^]8import java.util.List;
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]9import java.util.Set;
10
11import junit.framework.Assert;
12import junit.framework.TestCase;
Jonathan Hartc86a2ea2013-01-15 22:39:42 -0800[diff] [blame^]13import net.floodlightcontroller.core.INetMapStorage.DM_OPERATION;
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]14import net.floodlightcontroller.linkdiscovery.ILinkStorage;
15import net.floodlightcontroller.routing.Link;
16
17import org.apache.commons.io.FileUtils;
18import org.junit.After;
19import org.junit.Before;
20import org.junit.Test;
21
22import com.thinkaurelius.titan.core.TitanFactory;
23import com.thinkaurelius.titan.core.TitanGraph;
24import com.tinkerpop.blueprints.TransactionalGraph.Conclusion;
25import com.tinkerpop.blueprints.Vertex;
26import com.tinkerpop.blueprints.util.io.graphml.GraphMLReader;
27import com.tinkerpop.gremlin.java.GremlinPipeline;
28
29public class LinkStorageImplTest extends TestCase {
30 private static final String testDbLocation = "/tmp/onos-testdb";
31
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]32 private static final String testDbGraphML = "<?xml version=\"1.0\" ?><graphml xmlns=\"http://graphml.graphdrawing.org/xmlns\"> <key id=\"id\" for=\"node\" attr.name=\"id\" attr.type=\"string\"></key> <key id=\"type\" for=\"node\" attr.name=\"type\" attr.type=\"string\"></key> <key id=\"dpid\" for=\"node\" attr.name=\"dpid\" attr.type=\"string\"></key> <key id=\"desc\" for=\"node\" attr.name=\"desc\" attr.type=\"string\"></key> <key id=\"number\" for=\"node\" attr.name=\"number\" attr.type=\"int\"></key> <key id=\"dl_addr\" for=\"node\" attr.name=\"dl_addr\" attr.type=\"string\"></key> <key id=\"nw_addr\" for=\"node\" attr.name=\"nw_addr\" attr.type=\"string\"></key> <key id=\"id\" for=\"edge\" attr.name=\"id\" attr.type=\"string\"></key> <key id=\"source\" for=\"edge\" attr.name=\"source\" attr.type=\"string\"></key> <key id=\"target\" for=\"edge\" attr.name=\"target\" attr.type=\"string\"></key> <key id=\"label\" for=\"edge\" attr.name=\"label\" attr.type=\"string\"></key> <graph id=\"G\" edgedefault=\"directed\"> <node id=\"1\"> <data key=\"type\">switch</data> <data key=\"dpid\">00:00:00:00:00:00:0a:01</data> <data key=\"desc\">OpenFlow Switch at SEA</data> </node> <node id=\"2\"> <data key=\"type\">switch</data> <data key=\"dpid\">00:00:00:00:00:00:0a:02</data> <data key=\"desc\">OpenFlow Switch at LAX</data> </node> <node id=\"3\"> <data key=\"type\">switch</data> <data key=\"dpid\">00:00:00:00:00:00:0a:03</data> <data key=\"desc\">OpenFlow Switch at CHI</data> </node> <node id=\"4\"> <data key=\"type\">switch</data> <data key=\"dpid\">00:00:00:00:00:00:0a:04</data> <data key=\"desc\">OpenFlow Switch at IAH</data> </node> <node id=\"5\"> <data key=\"type\">switch</data> <data key=\"dpid\">00:00:00:00:00:00:0a:05</data> <data key=\"desc\">OpenFlow Switch at NYC</data> </node> <node id=\"6\"> <data key=\"type\">switch</data> <data key=\"dpid\">00:00:00:00:00:00:0a:06</data> <data key=\"desc\">OpenFlow Switch at ATL</data> </node> <node id=\"100\"> <data key=\"type\">port</data> <data key=\"number\">1</data> <data key=\"desc\">port 1 at SEA Switch</data> </node> <node id=\"101\"> <data key=\"type\">port</data> <data key=\"number\">2</data> <data key=\"desc\">port 2 at SEA Switch</data> </node> <node id=\"102\"> <data key=\"type\">port</data> <data key=\"number\">3</data> <data key=\"desc\">port 3 at SEA Switch</data> </node> <node id=\"103\"> <data key=\"type\">port</data> <data key=\"number\">4</data> <data key=\"desc\">port 4 at SEA Switch</data> </node> <node id=\"104\"> <data key=\"type\">port</data> <data key=\"number\">1</data> <data key=\"desc\">port 1 at LAX Switch</data> </node> <node id=\"105\"> <data key=\"type\">port</data> <data key=\"number\">2</data> <data key=\"desc\">port 2 at LAX Switch</data> </node> <node id=\"106\"> <data key=\"type\">port</data> <data key=\"number\">3</data> <data key=\"desc\">port 3 at LAX Switch</data> </node> <node id=\"107\"> <data key=\"type\">port</data> <data key=\"number\">1</data> <data key=\"desc\">port 1 at CHI Switch</data> </node> <node id=\"108\"> <data key=\"type\">port</data> <data key=\"number\">2</data> <data key=\"desc\">port 2 at CHI Switch</data> </node> <node id=\"109\"> <data key=\"type\">port</data> <data key=\"number\">3</data> <data key=\"desc\">port 3 at CHI Switch</data> </node> <node id=\"110\"> <data key=\"type\">port</data> <data key=\"number\">4</data> <data key=\"desc\">port 4 at CHI Switch</data> </node> <node id=\"111\"> <data key=\"type\">port</data> <data key=\"number\">1</data> <data key=\"desc\">port 1 at IAH Switch</data> </node> <node id=\"112\"> <data key=\"type\">port</data> <data key=\"number\">2</data> <data key=\"desc\">port 2 at IAH Switch</data> </node> <node id=\"113\"> <data key=\"type\">port</data> <data key=\"number\">3</data> <data key=\"desc\">port 3 at IAH Switch</data> </node> <node id=\"114\"> <data key=\"type\">port</data> <data key=\"number\">1</data> <data key=\"desc\">port 1 at NYC Switch</data> </node> <node id=\"115\"> <data key=\"type\">port</data> <data key=\"number\">2</data> <data key=\"desc\">port 2 at NYC Switch</data> </node> <node id=\"116\"> <data key=\"type\">port</data> <data key=\"number\">3</data> <data key=\"desc\">port 3 at NYC Switch</data> </node> <node id=\"117\"> <data key=\"type\">port</data> <data key=\"number\">1</data> <data key=\"desc\">port 1 at ATL Switch</data> </node> <node id=\"118\"> <data key=\"type\">port</data> <data key=\"number\">2</data> <data key=\"desc\">port 2 at ATL Switch</data> </node> <node id=\"119\"> <data key=\"type\">port</data> <data key=\"number\">3</data> <data key=\"desc\">port 3 at ATL Switch</data> </node> <node id=\"1000\"> <data key=\"type\">device</data> <data key=\"dl_addr\">20:c9:d0:4a:e1:73</data> <data key=\"nw_addr\">192.168.10.101</data> </node> <node id=\"1001\"> <data key=\"type\">device</data> <data key=\"dl_addr\">20:c9:d0:4a:e1:62</data> <data key=\"nw_addr\">192.168.20.101</data> </node> <node id=\"1002\"> <data key=\"type\">device</data> <data key=\"dl_addr\">10:40:f3:e6:8d:55</data> <data key=\"nw_addr\">192.168.10.1</data> </node> <node id=\"1003\"> <data key=\"type\">device</data> <data key=\"dl_addr\">a0:b3:cc:9c:c6:88</data> <data key=\"nw_addr\">192.168.20.1</data> </node> <node id=\"1004\"> <data key=\"type\">device</data> <data key=\"dl_addr\">00:04:20:e2:50:a2</data> <data key=\"nw_addr\">192.168.30.1</data> </node> <node id=\"1005\"> <data key=\"type\">device</data> <data key=\"dl_addr\">58:55:ca:c4:1b:a0</data> <data key=\"nw_addr\">192.168.40.1</data> </node> <edge id=\"10000\" source=\"1\" target=\"101\" label=\"on\"></edge> <edge id=\"10001\" source=\"1\" target=\"102\" label=\"on\"></edge> <edge id=\"10002\" source=\"1\" target=\"103\" label=\"on\"></edge> <edge id=\"10003\" source=\"2\" target=\"104\" label=\"on\"></edge> <edge id=\"10004\" source=\"2\" target=\"105\" label=\"on\"></edge> <edge id=\"10005\" source=\"2\" target=\"106\" label=\"on\"></edge> <edge id=\"10006\" source=\"3\" target=\"107\" label=\"on\"></edge> <edge id=\"10007\" source=\"3\" target=\"108\" label=\"on\"></edge> <edge id=\"10008\" source=\"3\" target=\"109\" label=\"on\"></edge> <edge id=\"10009\" source=\"3\" target=\"110\" label=\"on\"></edge> <edge id=\"10010\" source=\"4\" target=\"111\" label=\"on\"></edge> <edge id=\"10011\" source=\"4\" target=\"112\" label=\"on\"></edge> <edge id=\"10012\" source=\"4\" target=\"113\" label=\"on\"></edge> <edge id=\"10013\" source=\"5\" target=\"114\" label=\"on\"></edge> <edge id=\"10014\" source=\"5\" target=\"115\" label=\"on\"></edge> <edge id=\"10015\" source=\"5\" target=\"116\" label=\"on\"></edge> <edge id=\"10016\" source=\"6\" target=\"117\" label=\"on\"></edge> <edge id=\"10017\" source=\"6\" target=\"118\" label=\"on\"></edge> <edge id=\"10018\" source=\"6\" target=\"119\" label=\"on\"></edge> <edge id=\"11000\" source=\"101\" target=\"107\" label=\"link\"></edge> <edge id=\"11003\" source=\"105\" target=\"111\" label=\"link\"></edge> <edge id=\"11004\" source=\"107\" target=\"101\" label=\"link\"></edge> <edge id=\"11005\" source=\"108\" target=\"112\" label=\"link\"></edge> <edge id=\"11006\" source=\"109\" target=\"114\" label=\"link\"></edge> <edge id=\"11007\" source=\"111\" target=\"105\" label=\"link\"></edge> <edge id=\"11008\" source=\"112\" target=\"108\" label=\"link\"></edge> <edge id=\"11009\" source=\"113\" target=\"117\" label=\"link\"></edge> <edge id=\"11010\" source=\"114\" target=\"109\" label=\"link\"></edge> <edge id=\"11011\" source=\"115\" target=\"118\" label=\"link\"></edge> <edge id=\"11012\" source=\"117\" target=\"113\" label=\"link\"></edge> <edge id=\"11013\" source=\"118\" target=\"115\" label=\"link\"></edge> <edge id=\"12000\" source=\"103\" target=\"1000\" label=\"host\"></edge> <edge id=\"12001\" source=\"103\" target=\"1001\" label=\"host\"></edge> <edge id=\"12002\" source=\"110\" target=\"1002\" label=\"host\"></edge> <edge id=\"12003\" source=\"116\" target=\"1003\" label=\"host\"></edge> <edge id=\"12004\" source=\"106\" target=\"1004\" label=\"host\"></edge> <edge id=\"12005\" source=\"119\" target=\"1005\" label=\"host\"></edge> </graph> </graphml>";
33
34 private static ILinkStorage linkStorage;
35 private static TitanGraph titanGraph;
36
Jonathan Hartc86a2ea2013-01-15 22:39:42 -0800[diff] [blame^]37 //TODO Future ideas:
38 //Test add links with CREATE and UPDATE
39 //Test adding existing link again
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]40
41 @Before
42 public void setUp() {
43 deleteTestDatabase();
44
45 titanGraph = TitanFactory.open(testDbLocation);
46
47 populateTestData();
48
Jonathan Hartc86a2ea2013-01-15 22:39:42 -0800[diff] [blame^]49 linkStorage = new TestableLinkStorageImpl(titanGraph);
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]50 }
51
52 private void populateTestData(){
53 Set<String> s = titanGraph.getIndexedKeys(Vertex.class);
54 if (!s.contains("dpid")) {
55 titanGraph.createKeyIndex("dpid", Vertex.class);
56 titanGraph.stopTransaction(Conclusion.SUCCESS);
57 }
58 if (!s.contains("type")) {
59 titanGraph.createKeyIndex("type", Vertex.class);
60 titanGraph.stopTransaction(Conclusion.SUCCESS);
61 }
62
63 InputStream graphMLStream = new ByteArrayInputStream(testDbGraphML.getBytes());
64 try {
65 GraphMLReader.inputGraph(titanGraph, graphMLStream);
66 } catch (IOException e) {
67 e.printStackTrace();
68 Assert.fail("IOException thrown");
69 }
70
71 }
72
73 /*
74 * Add a link between port 1.102 and 2.104
75 * i.e SEA switch port 3 to LAX switch port 1
76 */
77 @Test
78 public void testAddSingleLink(){
79 Link linkToAdd = new Link(Long.decode("0x0000000000000a01"), 3, Long.decode("0x0000000000000a02"), 1);
80
81 //Use the link storage API to add the link
82 linkStorage.update(linkToAdd, ILinkStorage.DM_OPERATION.INSERT);
83
84 //Test if it was added correctly with the Gremlin API
85 GremlinPipeline<Vertex, Vertex> pipe = new GremlinPipeline<Vertex, Vertex>();
86 Iterator<Vertex> it = titanGraph.getVertices("dpid", "00:00:00:00:00:00:0a:01").iterator();
87
88 assertTrue(it.hasNext());
89 Vertex sw1 = it.next();
90 assertFalse(it.hasNext());
91
92 pipe.start(sw1).out("on").has("number", 3).out("link").in("on");
93
94 assertTrue(pipe.hasNext());
95 Vertex sw2 = pipe.next();
96 assertFalse(pipe.hasNext());
97
98 //Check we ended up at the right vertex
99 assertEquals((String)sw2.getProperty("dpid"), "00:00:00:00:00:00:0a:02");
100 }
101
102 @Test
103 public void testGetLinks(){
Jonathan Hartc86a2ea2013-01-15 22:39:42 -0800[diff] [blame^]104 //TODO Make sure this works when the implementation is written
105 List<Link> list = linkStorage.getLinks(Long.decode("0x0000000000000a01"), 2);
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]106
Jonathan Hartc86a2ea2013-01-15 22:39:42 -0800[diff] [blame^]107 assertEquals(list.size(), 1);
108
109 Link l = list.get(0);
110 assertEquals(l.getSrc(), 2561L);
111 assertEquals(l.getSrcPort(), (short)2);
112 assertEquals(l.getDst(), 2563L);
113 assertEquals(l.getDstPort(), (short)1);
114 }
115
116 @Test
117 public void testUpdateDelete(){
118 //TODO Make sure this works when the implementation is written
119 Link linkToDelete = new Link(Long.decode("0x0000000000000a01"), 2, Long.decode("0x0000000000000a03"), 1);
120
121 linkStorage.update(linkToDelete, DM_OPERATION.DELETE);
122
123 //Test if it was deleted correctly with the Gremlin API
124 GremlinPipeline<Vertex, Vertex> pipe = new GremlinPipeline<Vertex, Vertex>();
125 Iterator<Vertex> it = titanGraph.getVertices("dpid", "00:00:00:00:00:00:0a:01").iterator();
126
127 assertTrue(it.hasNext());
128 Vertex sw1 = it.next();
129 assertFalse(it.hasNext());
130
131 pipe.start(sw1).out("on").has("number", 2).out("link");
132
133 assertFalse(pipe.hasNext());
134 }
135
136 @Test
137 public void testDeleteLinks(){
138 //TODO Make sure this works when the implementation is written
139
140 linkStorage.deleteLinks(Long.decode("0x0000000000000a01"), 2);
141
142 //Test if it was deleted correctly with the Gremlin API
143 GremlinPipeline<Vertex, Vertex> pipe = new GremlinPipeline<Vertex, Vertex>();
144 Iterator<Vertex> it = titanGraph.getVertices("dpid", "00:00:00:00:00:00:0a:01").iterator();
145
146 assertTrue(it.hasNext());
147 Vertex sw1 = it.next();
148 assertFalse(it.hasNext());
149
150 pipe.start(sw1).out("on").has("number", 2).out("link");
151
152 assertFalse(pipe.hasNext());
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]153 }
154
155 @After
156 public void tearDown() {
Jonathan Hartc86a2ea2013-01-15 22:39:42 -0800[diff] [blame^]157 //TODO Reenable this once the tests are written. It's good to clean up afterwards.
Jonathan Hartb7e3d2c2013-01-15 18:45:19 -0800[diff] [blame]158 //deleteTestDatabase();
159 }
160
161 private void deleteTestDatabase(){
162 try {
163 FileUtils.deleteDirectory(new File(testDbLocation));
164 } catch (IOException e) {
165 System.out.println("delete failed");
166 e.printStackTrace();
167 }
168 }
169}