def test_topo_discovery():
tree = PhysicalTopology()
node1 = make_node("tester1", "IP_ADDR", 1234)
tree.addNode(node1, None, None)
printTest(1)
printPhyTopoTree(tree.root)
node2 = make_node("tester2", "IP_ADDR", 1235)
tree.addNode(node2, node1.name, 2)
printTest(2)
printPhyTopoTree(tree.root)
node3 = make_node("tester3", "IP_ADDR", 1236)
tree.addNode(node3, node1.name, 4)
printTest(3)
printPhyTopoTree(tree.root)
node4 = make_node("tester4", "IP_ADDR", 1237)
tree.addNode(node4, node3.name, 2)
printTest(4)
printPhyTopoTree(tree.root)
node5 = make_node("tester5", "IP_ADDR", 1238)
tree.addNode(node5, node3.name, 6)
printTest(5)
printPhyTopoTree(tree.root)
node6 = make_node("tester6", "IP_ADDR", 1239)
tree.addNode(node6, node3.name, 6)
printTest(6)
printPhyTopoTree(tree.root)
def test_operations():
tree = PhysicalTopology('topo.txt')
ovly = OverlayTopology(tree)
printTest(1)
printPhyTopoTree(tree.root)
printTest(2)
printOvlyTopoTree(ovly.root)
printTest(3)
print "Adding one node into the tree"
ovly.addNode(tree.findNodeByName('s34'))
printTest(4)
printOvlyTopoTree(ovly.root)
printTest(5)
print "Trying to register same node twice"
ovly.addNode(tree.findNodeByName('s34'))
printTest(6)
printOvlyTopoTree(ovly.root)
printTest(7)
print "Trying to register something else"
t = OverlayTopology(tree)
t.addNode(tree.findNodeByName('s31'))
t.addNode(tree.findNodeByName('s32'))
t.addNode(tree.findNodeByName('s33'))
t.addNode(tree.findNodeByName('s37'))
t.addNode(tree.findNodeByName('s35'))
t.addNode(tree.findNodeByName('s36'))
printTest(8)
printOvlyTopoTree(t.root)
printTest(9)
print "Trying to register something else"
t = OverlayTopology(tree)
t.addNode(tree.findNodeByName('s11'))
t.addNode(tree.findNodeByName('s21'))
t.addNode(tree.findNodeByName('s12'))
t.addNode(tree.findNodeByName('s22'))
t.addNode(tree.findNodeByName('s13'))
t.addNode(tree.findNodeByName('s23'))
t.addNode(tree.findNodeByName('s33'))
t.addNode(tree.findNodeByName('s32'))
printTest(10)
printOvlyTopoTree(t.root)
printTest(11)
print "Trying to remove something"
t.removeNode(tree.findNodeByName('s21'))
printTest(12)
printOvlyTopoTree(t.root)
printTest(13)
print "Trying to remove something"
t.removeNode(tree.findNodeByName('s23'))
printOvlyTopoTree(t.root)
t.removeNode(tree.findNodeByName('s22'))
printTest(14)
printOvlyTopoTree(t.root)
