Blame - src/test/java/net/floodlightcontroller/linkdiscovery/internal/LinkStorageImplTest.java - spring-open

2013-01-15 18:45:19 -0800

[diff] [blame]

24

25

private static ILinkStorage linkStorage;

26

private static TitanGraph titanGraph;

27

Jonathan Hart

2013-01-15 22:39:42 -0800

[diff] [blame]

28

//TODO Future ideas:

29

//Test add links with CREATE and UPDATE

30

//Test adding existing link again

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

31

32

@Before

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

33

public void setUp() throws Exception{

34

super.setUp();

35

TestDatabaseManager.deleteTestDatabase();

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

36

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

37

//titanGraph = TitanFactory.open(testDbLocation);

38

titanGraph = TestDatabaseManager.getTestDatabase();

39

TestDatabaseManager.populateTestData(titanGraph);

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

40

Jonathan Hart

2013-01-15 22:39:42 -0800

[diff] [blame]

41

linkStorage = new TestableLinkStorageImpl(titanGraph);

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

42

}

43

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

44

/*

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

45

private void populateTestData(){

46

Set<String> s = titanGraph.getIndexedKeys(Vertex.class);

47

if (!s.contains("dpid")) {

48

titanGraph.createKeyIndex("dpid", Vertex.class);

49

titanGraph.stopTransaction(Conclusion.SUCCESS);

50

}

51

if (!s.contains("type")) {

52

titanGraph.createKeyIndex("type", Vertex.class);

53

titanGraph.stopTransaction(Conclusion.SUCCESS);

54

}

55

56

InputStream graphMLStream = new ByteArrayInputStream(testDbGraphML.getBytes());

57

try {

58

GraphMLReader.inputGraph(titanGraph, graphMLStream);

59

} catch (IOException e) {

60

e.printStackTrace();

61

Assert.fail("IOException thrown");

62

}

63

64

}

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

65

*/

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

66

67

/*

68

* Add a link between port 1.102 and 2.104

69

* i.e SEA switch port 3 to LAX switch port 1

70

*/

71

@Test

72

public void testAddSingleLink(){

73

Link linkToAdd = new Link(Long.decode("0x0000000000000a01"), 3, Long.decode("0x0000000000000a02"), 1);

74

75

//Use the link storage API to add the link

76

linkStorage.update(linkToAdd, ILinkStorage.DM_OPERATION.INSERT);

77

78

//Test if it was added correctly with the Gremlin API

79

GremlinPipeline<Vertex, Vertex> pipe = new GremlinPipeline<Vertex, Vertex>();

80

Iterator<Vertex> it = titanGraph.getVertices("dpid", "00:00:00:00:00:00:0a:01").iterator();

81

82

assertTrue(it.hasNext());

83

Vertex sw1 = it.next();

84

assertFalse(it.hasNext());

85

86

pipe.start(sw1).out("on").has("number", 3).out("link").in("on");

87

88

assertTrue(pipe.hasNext());

89

Vertex sw2 = pipe.next();

90

assertFalse(pipe.hasNext());

91

92

//Check we ended up at the right vertex

93

assertEquals((String)sw2.getProperty("dpid"), "00:00:00:00:00:00:0a:02");

}

@Test

public void testGetLinks(){

Jonathan Hart

2013-01-15 22:39:42 -0800

[diff] [blame]

98

//TODO Make sure this works when the implementation is written

Umesh Krishnaswamy

f962d64

2013-01-23 19:04:23 -0800

[diff] [blame^]

99

List<Link> list = linkStorage.getLinks(Long.decode("0x0000000000000a01"), (short)2);

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

100

Jonathan Hart

2013-01-15 22:39:42 -0800

[diff] [blame]

101

assertEquals(list.size(), 1);

102

103

Link l = list.get(0);

104

assertEquals(l.getSrc(), 2561L);

105

assertEquals(l.getSrcPort(), (short)2);

106

assertEquals(l.getDst(), 2563L);

107

assertEquals(l.getDstPort(), (short)1);

}

@Test

public void testUpdateDelete(){

112

//TODO Make sure this works when the implementation is written

113

Link linkToDelete = new Link(Long.decode("0x0000000000000a01"), 2, Long.decode("0x0000000000000a03"), 1);

114

115

linkStorage.update(linkToDelete, DM_OPERATION.DELETE);

116

117

//Test if it was deleted correctly with the Gremlin API

118

GremlinPipeline<Vertex, Vertex> pipe = new GremlinPipeline<Vertex, Vertex>();

119

Iterator<Vertex> it = titanGraph.getVertices("dpid", "00:00:00:00:00:00:0a:01").iterator();

120

121

assertTrue(it.hasNext());

122

Vertex sw1 = it.next();

123

assertFalse(it.hasNext());

124

125

pipe.start(sw1).out("on").has("number", 2).out("link");

126

127

assertFalse(pipe.hasNext());

}

@Test

public void testDeleteLinks(){

132

//TODO Make sure this works when the implementation is written

133

Umesh Krishnaswamy

f962d64

2013-01-23 19:04:23 -0800

[diff] [blame^]

134

linkStorage.deleteLinksOnPort(Long.decode("0x0000000000000a01"), (short)2);

Jonathan Hart

2013-01-15 22:39:42 -0800

[diff] [blame]

135

136

//Test if it was deleted correctly with the Gremlin API

137

GremlinPipeline<Vertex, Vertex> pipe = new GremlinPipeline<Vertex, Vertex>();

138

Iterator<Vertex> it = titanGraph.getVertices("dpid", "00:00:00:00:00:00:0a:01").iterator();

139

140

assertTrue(it.hasNext());

141

Vertex sw1 = it.next();

142

assertFalse(it.hasNext());

143

144

pipe.start(sw1).out("on").has("number", 2).out("link");

145

146

assertFalse(pipe.hasNext());

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

147

}

148

149

@After

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

150

public void tearDown() throws Exception {

151

super.tearDown();

Jonathan Hart

2013-01-15 22:39:42 -0800

[diff] [blame]

152

//TODO Reenable this once the tests are written. It's good to clean up afterwards.

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

153

//TestDatabaseManager.deleteTestDatabase();

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

154

}

155

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

156

/*

Jonathan Hart

2013-01-15 18:45:19 -0800

[diff] [blame]

157

private void deleteTestDatabase(){

158

try {

159

FileUtils.deleteDirectory(new File(testDbLocation));

160

} catch (IOException e) {

161

System.out.println("delete failed");

162

e.printStackTrace();

163

}

164

}

Jonathan Hart

2013-01-16 14:20:03 -0800

[diff] [blame]

165

*/

Jonathan Hart