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)