class MyTestCase(unittest.TestCase): def setUp(self): self.node = Node() self.substrate_node = SubstrateNode(200) self.net = Net() self.vnf = VNF('test') def test_Node(self): self.assertEqual(self.node.test_function(), "test function: Node") def test_substrate_node(self): self.assertEqual(self.substrate_node.get_id(), 200) self.assertEqual(self.substrate_node.set_cpu_capacity(100), 100) self.assertEqual(self.substrate_node.get_cpu_free(), 100) self.assertEqual(self.substrate_node.get_cpu_capacity(), 100) self.assertEqual(self.substrate_node.allocate_cpu_resource(20), 20) self.assertEqual(self.substrate_node.get_cpu_used(), 20) self.assertEqual(self.substrate_node.get_cpu_free(), 80) self.substrate_node.reset() self.assertEqual(self.substrate_node.get_cpu_capacity(), 100) self.assertEqual(self.substrate_node.get_cpu_free(), 100) self.assertEqual(self.substrate_node.get_cpu_used(), 0) self.substrate_node.reset_with_cpu_capacity(200) self.assertEqual(self.substrate_node.get_cpu_capacity(), 200) self.assertEqual(self.substrate_node.get_cpu_used(), 0) self.assertEqual(self.substrate_node.get_cpu_free(), 200) self.assertEqual(self.substrate_node.allocate_cpu_resource(150), 150) self.assertEqual(self.substrate_node.get_cpu_capacity(), 200) self.assertEqual(self.substrate_node.get_cpu_free(), 50) self.assertEqual(self.substrate_node.get_cpu_used(), 150) def test_net_node(self): self.net.init_node_cpu_capacity(1, 100) self.assertEqual(self.net.get_node_cpu_capacity(1), 100) self.assertEqual(self.net.get_node_cpu_free(1), 100) self.assertEqual(self.net.get_node_cpu_used(1), 0) self.assertTrue(self.net.allocate_cpu_resource(1, 10)) self.assertEqual(self.net.get_node_cpu_capacity(1), 100) self.assertEqual(self.net.get_node_cpu_free(1), 90) self.assertEqual(self.net.get_node_cpu_used(1), 10) self.assertFalse(self.net.allocate_cpu_resource(1, 95)) self.net.reset_node_cpu_capacity(1, 100) self.assertEqual(self.net.get_node_cpu_capacity(1), 100) self.assertEqual(self.net.get_node_cpu_free(1), 100) self.assertEqual(self.net.get_node_cpu_used(1), 0) self.assertEqual(1, 1) def test_net_link(self): self.net.init_bandwidth_capacity(1, 2, 100) self.assertEqual(self.net.get_link_bandwidth_capacity(1, 2), 100) self.assertEqual(self.net.get_link_bandwidth_used(1, 2), 0) self.assertEqual(self.net.get_link_bandwidth_free(1, 2), 100) self.assertTrue(self.net.allocate_bandwidth_resource(1, 2, 80)) self.assertFalse(self.net.allocate_bandwidth_resource(1, 2, 90)) self.assertEqual(self.net.get_link_bandwidth_used(1, 2), 80) self.assertEqual(self.net.get_link_bandwidth_free(1, 2), 20) self.net.init_link_latency(1, 2, 2) self.assertEqual(self.net.get_link_latency( 1, 2, ), 2) pass def test_vnf(self): self.vnf.set_cpu_request(100) self.assertEqual(self.vnf.get_cpu_request(), 100) self.vnf.set_income_interface_bandwidth(105) self.assertEqual(self.vnf.get_income_interface_bandwidth(), 105) self.vnf.set_outcome_interface_banwdith(110) self.assertEqual(self.vnf.get_outcome_interface_bandwidth(), 110) vnf2 = VNF(2) vnf3 = VNF(3) self.vnf.set_next_vnf(vnf2) self.vnf.set_previous_vnf(vnf3) self.assertEqual(self.vnf.get_previous_vnf().id, 3) self.assertEqual(self.vnf.get_next_vnf().id, 2) pass