class TestStaticMPLS(unittest.TestCase): # pylint: disable=invalid-name def setUp(self): self.n0 = Node("n0") self.n1 = Node("n1") self.r0 = Node("r0") self.r1 = Node("r1") self.r0.enable_ip_forwarding() self.r1.enable_ip_forwarding() def tearDown(self): delete_namespaces() ### Create interfaces and connect nodes and routers ### def test_routing(self): (eth_n0_r0, eth_r0_n0) = connect(self.n0, self.r0, "eth-n0_r0", "eth-r0_n0") (eth_r0_r1, eth_r1_r0) = connect(self.r0, self.r1, "eth-r0r1-0", "eth-r1r0-0") (eth_r1_n1, eth_n1_r1) = connect(self.r1, self.n1, "eth-r1n1-0", "eth-n1r1-0") ### Assign addresses to interfaces ### eth_n0_r0.set_address("10.0.1.1/24") eth_r0_n0.set_address("10.0.1.2/24") eth_r0_r1.set_address("10.0.2.2/24") eth_r1_r0.set_address("10.0.2.3/24") eth_r1_n1.set_address("10.0.3.3/24") eth_n1_r1.set_address("10.0.3.4/24") # Enable MPLS on all interfaces eth_n0_r0.enable_mpls() eth_r0_n0.enable_mpls() eth_r0_r1.enable_mpls() eth_r1_r0.enable_mpls() eth_r1_n1.enable_mpls() eth_n1_r1.enable_mpls() # Add routes(mpls family) self.n0.add_route_mpls_push("10.0.3.0/24", eth_r0_n0.get_address(), 101) self.r0.add_route_mpls_switch(101, eth_r1_r0.get_address(), 102) self.r1.add_route_mpls_pop(102, eth_n1_r1.get_address()) self.n1.add_route_mpls_push("10.0.1.0/24", eth_r1_n1.get_address(), 201) self.r1.add_route_mpls_switch(201, eth_r0_r1.get_address(), 202) self.r0.add_route_mpls_pop(202, eth_n0_r0.get_address()) status = self.n0.ping("10.0.3.4", verbose=True) self.assertTrue(status) status = self.n1.ping("10.0.1.1", verbose=True) self.assertTrue(status)
class TestFrr(unittest.TestCase): # pylint: disable=invalid-name # pylint: disable=too-many-instance-attributes def setUp(self): self.n0 = Node("n0") self.n1 = Node("n1") self.r0 = Node("r0") self.r1 = Node("r1") self.r0.enable_ip_forwarding() self.r1.enable_ip_forwarding() ### Create interfaces and connect nodes and routers ### (eth_p1r1, eth_r1p1) = connect(self.n0, self.r0, "eth-n1r1-0", "eth-r1n1-0") (eth_r1r2, eth_r2r1) = connect(self.r0, self.r1, "eth-r1r2-0", "eth-r2r1-0") (eth_r2p2, eth_p2r2) = connect(self.r1, self.n1, "eth-r2n2-0", "eth-n2r2-0") ### Assign addresses to interfaces ### eth_p1r1.set_address("10::1:1/122") eth_r1p1.set_address("10::1:2/122") eth_r1r2.set_address("10::2:2/122") eth_r2r1.set_address("10::2:3/122") eth_r2p2.set_address("10::3:3/122") eth_p2r2.set_address("10::3:4/122") config.set_value("routing_suite", "frr") # Use frr def tearDown(self): delete_namespaces() TopologyMap.delete_all_mapping() def test_rip_ipv6(self): RoutingHelper("rip").populate_routing_tables() status = self.n0.ping("10::3:4", verbose=False) self.assertTrue(status) status = self.n1.ping("10::1:1", verbose=False) self.assertTrue(status) def test_ospf_ipv6(self): RoutingHelper("ospf").populate_routing_tables() status = self.n0.ping("10::3:4", verbose=False) self.assertTrue(status) status = self.n1.ping("10::1:1", verbose=False) self.assertTrue(status)
class TestFrr(unittest.TestCase): # pylint: disable=invalid-name def setUp(self): self.assertTrue(isfile("/usr/lib/frr/zebra"), "Frrouting is not installed") self.n0 = Node("n0") self.n1 = Node("n1") self.r0 = Node("r0") self.r1 = Node("r1") self.r0.enable_ip_forwarding() self.r1.enable_ip_forwarding() ### Create interfaces and connect nodes and routers ### (eth_p1r1, eth_r1p1) = connect(self.n0, self.r0, "eth-n1r1-0", "eth-r1n1-0") (eth_r1r2, eth_r2r1) = connect(self.r0, self.r1, "eth-r1r2-0", "eth-r2r1-0") (eth_r2p2, eth_p2r2) = connect(self.r1, self.n1, "eth-r2n2-0", "eth-n2r2-0") ### Assign addresses to interfaces ### eth_p1r1.set_address("10.0.1.1/24") eth_r1p1.set_address("10.0.1.2/24") eth_r1r2.set_address("10.0.2.2/24") eth_r2r1.set_address("10.0.2.3/24") eth_r2p2.set_address("10.0.3.3/24") eth_p2r2.set_address("10.0.3.4/24") config.set_value("routing_suite", "frr") # Use frr def tearDown(self): delete_namespaces() TopologyMap.delete_all_mapping() def test_routing_helper(self): RoutingHelper("rip").populate_routing_tables() status = self.n0.ping("10.0.3.4", verbose=False) self.assertTrue(status) status = self.n1.ping("10.0.1.1", verbose=False) self.assertTrue(status) def test_ospf(self): RoutingHelper("ospf").populate_routing_tables() status = self.n0.ping("10.0.3.4", verbose=False) self.assertTrue(status) status = self.n1.ping("10.0.1.1", verbose=False) self.assertTrue(status)
class TestIPv4AddressHelper(unittest.TestCase): def setUp(self): self.n0 = Node("n0") self.n1 = Node("n1") self.net1 = Network("10.0.1.0/24") self.net2 = Network("10.0.2.0/24") self.net3 = Network("2001:101::/122") self.net4 = Network("2002:101::/122") def tearDown(self): delete_namespaces() TopologyMap.delete_all_mapping() def test_p2p(self): (n0_n1, n1_n0) = connect(self.n0, self.n1, network=self.net1) AddressHelper.assign_addresses() status = self.n0.ping(n1_n0.get_address(), verbose=False) self.assertTrue(status) def test_p2p_ipv6(self): (n0_n1, n1_n0) = connect(self.n0, self.n1, network=self.net3) AddressHelper.assign_addresses() status = self.n0.ping(n1_n0.get_address(), verbose=False) self.assertTrue(status) def test_prp(self): # pylint: disable=invalid-name r = Node("r") r.enable_ip_forwarding() with self.net1: (n0_r, r_n0) = connect(self.n0, r) with self.net2: (r_n1, n1_r) = connect(r, self.n1, network=self.net2) AddressHelper.assign_addresses() self.n0.add_route("DEFAULT", n0_r) self.n1.add_route("DEFAULT", n1_r) status = self.n0.ping(n1_r.get_address(), verbose=False) self.assertTrue(status) def test_prp_ipv6(self): # pylint: disable=invalid-name r = Node("r") r.enable_ip_forwarding() with self.net3: (n0_r, r_n0) = connect(self.n0, r) with self.net4: (r_n1, n1_r) = connect(r, self.n1, network=self.net4) AddressHelper.assign_addresses() self.n0.add_route("DEFAULT", n0_r) self.n1.add_route("DEFAULT", n1_r) status = self.n0.ping(n1_r.get_address(), verbose=False) self.assertTrue(status) def test_prrp(self): # pylint: disable=invalid-name r1 = Node("r") r2 = Node("r") r1.enable_ip_forwarding() r2.enable_ip_forwarding() with self.net1: (n0_r1, r1_n0) = connect(self.n0, r1) (r1_r2, r2_r1) = connect(r1, r2, network=self.net2) (r2_n1, n1_r2) = connect(r2, self.n1) AddressHelper.assign_addresses() r2_n1.set_address("10.1.3.2/24") n1_r2.set_address("10.1.3.1/24") RoutingHelper(protocol="rip").populate_routing_tables() status = self.n0.ping(n1_r2.get_address(), verbose=False) self.assertTrue(status) def test_prrp_ipv6(self): # pylint: disable=invalid-name r1 = Node("r") r2 = Node("r") r1.enable_ip_forwarding() r2.enable_ip_forwarding() with self.net3: (n0_r1, r1_n0) = connect(self.n0, r1) (r1_r2, r2_r1) = connect(r1, r2, network=self.net4) (r2_n1, n1_r2) = connect(r2, self.n1) AddressHelper.assign_addresses() r2_n1.set_address("2003:101::10:1/122") n1_r2.set_address("2003:101::10:2/122") RoutingHelper(protocol="rip").populate_routing_tables() status = self.n0.ping(n1_r2.get_address(), verbose=False) self.assertTrue(status) def test_tcp_param(self): self.n0.configure_tcp_param("ecn", "1") ecn = self.n0.read_tcp_param("ecn") self.assertEqual(ecn, "1") def test_run_inside_node(self): (n0_n1, n1_n0) = connect(self.n0, self.n1, network=self.net1) AddressHelper.assign_addresses() # Run ping from self.n0 to self.n1 with self.n0: command = f"ping -c 1 {n1_n0.address.get_addr(with_subnet=False)}" proc = subprocess.Popen(command.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE) (stdout, _) = proc.communicate() self.assertEqual(stdout[:4], b"PING", "Invalid ping output")
class TestQuagga(unittest.TestCase): # pylint: disable=invalid-name def setUp(self): self.n0 = Node("n0") self.n1 = Node("n1") self.r0 = Node("r0") self.r1 = Node("r1") self.r0.enable_ip_forwarding() self.r1.enable_ip_forwarding() ### Create interfaces and connect nodes and routers ### (eth_p1r1, eth_r1p1) = connect(self.n0, self.r0, "eth-n1r1-0", "eth-r1n1-0") (eth_r1r2, eth_r2r1) = connect(self.r0, self.r1, "eth-r1r2-0", "eth-r2r1-0") (eth_r2p2, eth_p2r2) = connect(self.r1, self.n1, "eth-r2n2-0", "eth-n2r2-0") ### Assign addresses to interfaces ### eth_p1r1.set_address("10.0.1.1/24") eth_r1p1.set_address("10.0.1.2/24") eth_r1r2.set_address("10.0.2.2/24") eth_r2r1.set_address("10.0.2.3/24") eth_r2p2.set_address("10.0.3.3/24") eth_p2r2.set_address("10.0.3.4/24") config.set_value("routing_suite", "quagga") # Use quagga def tearDown(self): delete_namespaces() TopologyMap.delete_all_mapping() def test_routing_helper(self): RoutingHelper("rip").populate_routing_tables() status = self.n0.ping("10.0.3.4", verbose=False) self.assertTrue(status) status = self.n1.ping("10.0.1.1", verbose=False) self.assertTrue(status) def test_ospf(self): RoutingHelper("ospf").populate_routing_tables() status = self.n0.ping("10.0.3.4", verbose=False) self.assertTrue(status) status = self.n1.ping("10.0.1.1", verbose=False) self.assertTrue(status) def test_isis(self): RoutingHelper("isis").populate_routing_tables() status = self.n0.ping("10.0.3.4", verbose=False) self.assertTrue(status) status = self.n1.ping("10.0.1.1", verbose=False) self.assertTrue(status) def test_logs(self): config.set_value("routing_logs", True) RoutingHelper("rip").populate_routing_tables() self.assertTrue( len(glob(f"{config.get_value('routing_suite')}-logs_*")) > 0) config.set_value("routing_logs", False) def test_custom_node_routers(self): RoutingHelper("rip", [self.n0, self.n1], [self.r0, self.r1]).populate_routing_tables() status = self.n0.ping("10.0.3.4", verbose=False) self.assertTrue(status) status = self.n1.ping("10.0.1.1", verbose=False) self.assertTrue(status) with self.assertRaises(TypeError): RoutingHelper("rip", self.n1, self.r1).populate_routing_tables() with self.assertRaises(ValueError): RoutingHelper("rip", ["n1"], ["r1"]).populate_routing_tables()
class TestTopology(unittest.TestCase): def setUp(self): self.n0 = Node("n0") self.n1 = Node("n1") def tearDown(self): delete_namespaces() TopologyMap.delete_all_mapping() def test_p2p(self): (n0_n1, n1_n0) = connect(self.n0, self.n1) n0_n1.set_address("10.0.0.1/24") n1_n0.set_address("10.0.0.2/24") status = self.n0.ping("10.0.0.2") self.assertTrue(status) def test_p2p_ipv6(self): (n0_n1, n1_n0) = connect(self.n0, self.n1) self.n0.disable_ip_dad() self.n1.disable_ip_dad() n0_n1.set_address("2001:1:1:1443::411/122") n1_n0.set_address("2001:1:1:1443::412/122") status = self.n0.ping(n1_n0.address) self.assertTrue(status) def test_prp(self): # pylint: disable=invalid-name r = Node("r") r.enable_ip_forwarding() (n0_r, r_n0) = connect(self.n0, r) (r_n1, n1_r) = connect(r, self.n1) n0_r.set_address("10.1.1.1/24") r_n0.set_address("10.1.1.2/24") r_n1.set_address("10.1.2.2/24") n1_r.set_address("10.1.2.1/24") self.n0.add_route("DEFAULT", n0_r) self.n1.add_route("DEFAULT", n1_r) status = self.n0.ping("10.1.2.1") self.assertTrue(status) def test_prp_ipv6(self): # pylint: disable=invalid-name r = Node("r") r.enable_ip_forwarding(ipv6=True) (n0_r, r_n0) = connect(self.n0, r) (r_n1, n1_r) = connect(r, self.n1) self.n0.disable_ip_dad() self.n1.disable_ip_dad() n0_r.set_address("2001:0001:0001:1443::0411/122") r_n0.set_address("2001:0001:0001:1443::0412/122") r_n1.set_address("2001:0001:0001:1444::0412/122") n1_r.set_address("2001:0001:0001:1444::0411/122") self.n0.add_route("DEFAULT", n0_r) self.n1.add_route("DEFAULT", n1_r) status = self.n0.ping(n1_r.address) self.assertTrue(status) def test_simple_lan(self): # pylint: disable=too-many-locals n0 = Node("n0") n1 = Node("n1") n2 = Node("n2") n3 = Node("n3") s0 = Switch("s0") (n0_s0, _) = connect(n0, s0) (n1_s0, _) = connect(n1, s0) (n2_s0, _) = connect(n2, s0) (n3_s0, _) = connect(n3, s0) nodes = [n0, n1, n2, n3] interface = [n0_s0, n1_s0, n2_s0, n3_s0] n0_s0.set_address("10.0.0.1/24") n1_s0.set_address("10.0.0.2/24") n2_s0.set_address("10.0.0.3/24") n3_s0.set_address("10.0.0.4/24") for x in nodes: for y in interface: status = x.ping(y.address, packets=1) if not status: break self.assertTrue(status) def test_dumbbell_lan(self): # pylint: disable=too-many-locals n0 = Node("n0") n1 = Node("n1") n2 = Node("n2") n3 = Node("n3") s0 = Switch("s0") s1 = Switch("s1") (n0_s0, _) = connect(n0, s0) (n1_s0, _) = connect(n1, s0) (n2_s1, _) = connect(n2, s1) (n3_s1, _) = connect(n3, s1) connect(s0, s1) nodes = [n0, n1, n2, n3] interface = [n0_s0, n1_s0, n2_s1, n3_s1] n0_s0.set_address("10.0.0.1/24") n1_s0.set_address("10.0.0.2/24") n2_s1.set_address("10.0.0.3/24") n3_s1.set_address("10.0.0.4/24") for x in nodes: for y in interface: status = x.ping(y.address, packets=1) if not status: break self.assertTrue(status) def test_tcp_param(self): self.n0.configure_tcp_param("ecn", "1") ecn = self.n0.read_tcp_param("ecn") self.assertEqual(ecn, "1") def test_invalid_interface_name(self): # Disable topology map config.set_value("assign_random_names", False) # Valid interface name Interface("namewith15chars") # Invalid interface name with self.assertRaises(ValueError) as cm: Interface("looonginvalidname") err = cm.exception self.assertEqual( str(err), "Device name looonginvalidname is too long. Device names " "should not exceed 15 characters", ) # Enable topology map config.set_value("assign_random_names", True) def test_invalid_veth_name(self): # Disable topology map config.set_value("assign_random_names", False) node0 = Node("longname0") node1 = Node("longname1") with self.assertRaises(ValueError) as cm: connect(node0, node1) err = cm.exception self.assertEqual( str(err), "Auto-generated device name longname0-longname1-0 is " "too long. The length of name should not exceed 15 characters.", ) # Enable topology map config.set_value("assign_random_names", True) def test_valid_veth_name(self): node0 = Node("node0") node1 = Node("node1") eth0, eth1 = connect(node0, node1) self.assertEqual(eth0.name, "node0-node1-0", "eth0 has unexpected autogenerated name") self.assertEqual(eth1.name, "node1-node0-0", "eth1 has unexpected autogenerated name") eth2, eth3 = connect(node0, node1) self.assertEqual(eth2.name, "node0-node1-1", "eth2 has unexpected autogenerated name") self.assertEqual(eth3.name, "node1-node0-1", "eth3 has unexpected autogenerated name") eth4, eth5 = connect(node0, node1, "eth4", "eth5") self.assertEqual(eth4.name, "eth4", "eth4 has unexpected user-given name") self.assertEqual(eth5.name, "eth5", "eth5 has unexpected user-given name") def test_invalid_ifb_interface_name(self): # Disable topology map config.set_value("assign_random_names", False) node0 = Node("node0") node1 = Node("node1") eth0, _ = connect(node0, node1) with self.assertRaises(ValueError) as cm: eth0.set_attributes("10mbit", "10ms", "codel") err = cm.exception self.assertEqual( str(err), "Device name ifb-node0-node1-0 is too " "long. Device names should not exceed 15 characters", ) # Enable topology map config.set_value("assign_random_names", True) def test_run_inside_node(self): (n0_n1, n1_n0) = connect(self.n0, self.n1) n0_n1.set_address("10.0.0.1/24") n1_n0.set_address("10.0.0.2/24") # Run ping from self.n0 to self.n1 with self.n0: command = f"ping -c 1 {n1_n0.address.get_addr(with_subnet=False)}" with subprocess.Popen(command.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE) as proc: (stdout, _) = proc.communicate() self.assertEqual(stdout[:4], b"PING", "Invalid ping output") def test_prp_router_api(self): # pylint: disable=invalid-name r = Router("r") (n0_r, r_n0) = connect(self.n0, r) (r_n1, n1_r) = connect(r, self.n1) n0_r.set_address("10.1.1.1/24") r_n0.set_address("10.1.1.2/24") r_n1.set_address("10.1.2.2/24") n1_r.set_address("10.1.2.1/24") self.n0.add_route("DEFAULT", n0_r) self.n1.add_route("DEFAULT", n1_r) status = self.n0.ping("10.1.2.1") self.assertTrue(status)