diff options
Diffstat (limited to 'test/test_l2bd.py')
| -rw-r--r-- | test/test_l2bd.py | 223 |
1 files changed, 146 insertions, 77 deletions
diff --git a/test/test_l2bd.py b/test/test_l2bd.py index 3c65cc1e0a8..46ba2e49c13 100644 --- a/test/test_l2bd.py +++ b/test/test_l2bd.py @@ -1,7 +1,6 @@ #!/usr/bin/env python import unittest -from logging import * import random from scapy.packet import Raw @@ -9,78 +8,111 @@ from scapy.layers.l2 import Ether, Dot1Q from scapy.layers.inet import IP, UDP from framework import VppTestCase, VppTestRunner -from vpp_sub_interface import VppDot1QSubint from util import Host +from vpp_sub_interface import VppDot1QSubint, VppDot1ADSubint class TestL2bd(VppTestCase): """ L2BD Test Case """ - # Test variables - bd_id = 1 # Bridge domain ID - mac_entries_count = 100 # Number of MAC entries for bridge-domain to learn - dot1q_sub_id = 100 # SubID of dot1q sub-interface - dot1q_tag = 100 # VLAN tag for dot1q sub-interface - dot1ad_sub_id = 200 # SubID of dot1ad sub-interface - dot1ad_outer_tag = 200 # VLAN S-tag for dot1ad sub-interface - dot1ad_inner_tag = 300 # VLAN C-tag for dot1ad sub-interface - pkts_per_burst = 257 # Number of packets per burst - @classmethod def setUpClass(cls): + """ + Perform standard class setup (defined by class method setUpClass in + class VppTestCase) before running the test case, set test case related + variables and configure VPP. + + :var int bd_id: Bridge domain ID. + :var int mac_entries_count: Number of MAC entries for bridge-domain to + learn. + :var int dot1q_tag: VLAN tag for dot1q sub-interface. + :var int dot1ad_sub_id: SubID of dot1ad sub-interface. + :var int dot1ad_outer_tag: VLAN S-tag for dot1ad sub-interface. + :var int dot1ad_inner_tag: VLAN C-tag for dot1ad sub-interface. + :var int sl_pkts_per_burst: Number of packets in burst for single-loop + test. + :var int dl_pkts_per_burst: Number of packets in burst for dual-loop + test. + """ super(TestL2bd, cls).setUpClass() - def setUp(self): - super(TestL2bd, self).setUp() + # Test variables + cls.bd_id = 1 + cls.mac_entries_count = 100 + # cls.dot1q_sub_id = 100 + cls.dot1q_tag = 100 + cls.dot1ad_sub_id = 20 + cls.dot1ad_outer_tag = 200 + cls.dot1ad_inner_tag = 300 + cls.sl_pkts_per_burst = 2 + cls.dl_pkts_per_burst = 257 + + try: + # create 3 pg interfaces + cls.create_pg_interfaces(range(3)) - # create 3 pg interfaces - self.create_pg_interfaces(range(3)) + # create 2 sub-interfaces for pg1 and pg2 + cls.sub_interfaces = [ + VppDot1QSubint(cls, cls.pg1, cls.dot1q_tag), + VppDot1ADSubint(cls, cls.pg2, cls.dot1ad_sub_id, + cls.dot1ad_outer_tag, cls.dot1ad_inner_tag)] - # create 2 sub-interfaces for pg1 and pg2 - self.sub_interfaces = [ - VppDot1QSubint(self, self.pg1, TestL2bd.dot1q_sub_id), - VppDot1QSubint(self, self.pg2, TestL2bd.dot1ad_sub_id)] + # packet flows mapping pg0 -> pg1, pg2, etc. + cls.flows = dict() + cls.flows[cls.pg0] = [cls.pg1, cls.pg2] + cls.flows[cls.pg1] = [cls.pg0, cls.pg2] + cls.flows[cls.pg2] = [cls.pg0, cls.pg1] - # packet flows mapping pg0 -> pg1, pg2, etc. - self.flows = dict() - self.flows[self.pg0] = [self.pg1, self.pg2] - self.flows[self.pg1] = [self.pg0, self.pg2] - self.flows[self.pg2] = [self.pg0, self.pg1] + # packet sizes + cls.pg_if_packet_sizes = [64, 512, 1518, 9018] + cls.sub_if_packet_sizes = [64, 512, 1518 + 4, 9018 + 4] - # packet sizes - self.pg_if_packet_sizes = [64, 512, 1518, 9018] - self.sub_if_packet_sizes = [64, 512, 1518 + 4, 9018 + 4] + cls.interfaces = list(cls.pg_interfaces) + cls.interfaces.extend(cls.sub_interfaces) - self.interfaces = list(self.pg_interfaces) - self.interfaces.extend(self.sub_interfaces) + # Create BD with MAC learning enabled and put interfaces and + # sub-interfaces to this BD + for pg_if in cls.pg_interfaces: + sw_if_index = pg_if.sub_if.sw_if_index \ + if hasattr(pg_if, 'sub_if') else pg_if.sw_if_index + cls.vapi.sw_interface_set_l2_bridge(sw_if_index, + bd_id=cls.bd_id) - # Create BD with MAC learning enabled and put interfaces and - # sub-interfaces to this BD - for pg_if in self.pg_interfaces: - sw_if_index = pg_if.sub_if.sw_if_index if hasattr(pg_if, 'sub_if') \ - else pg_if.sw_if_index - self.vapi.sw_interface_set_l2_bridge(sw_if_index, - bd_id=TestL2bd.bd_id) + # setup all interfaces + for i in cls.interfaces: + i.admin_up() - # setup all interfaces - for i in self.interfaces: - i.admin_up() + # mapping between packet-generator index and lists of test hosts + cls.hosts_by_pg_idx = dict() - # mapping between packet-generator index and lists of test hosts - self.hosts_by_pg_idx = dict() + # create test host entries and inject packets to learn MAC entries + # in the bridge-domain + cls.create_hosts_and_learn(cls.mac_entries_count) + cls.logger.info(cls.vapi.ppcli("show l2fib")) - # create test host entries and inject packets to learn MAC entries in - # the bridge-domain - self.create_hosts_and_learn(TestL2bd.mac_entries_count) - info(self.vapi.cli("show l2fib")) + except Exception: + super(TestL2bd, cls).tearDownClass() + raise + + def setUp(self): + """ + Clear trace and packet infos before running each test. + """ + super(TestL2bd, self).setUp() + self.packet_infos = {} def tearDown(self): + """ + Show various debug prints after each test. + """ super(TestL2bd, self).tearDown() if not self.vpp_dead: - info(self.vapi.cli("show l2fib verbose")) - info(self.vapi.cli("show bridge-domain %s detail" % self.bd_id)) + self.logger.info(self.vapi.ppcli("show l2fib verbose")) + self.logger.info(self.vapi.ppcli("show bridge-domain %s detail" % + self.bd_id)) - def create_hosts_and_learn(self, count): + @classmethod + def create_hosts_and_learn(cls, count): """ Create required number of host MAC addresses and distribute them among interfaces. Create host IPv4 address for every host MAC address. Create @@ -89,15 +121,15 @@ class TestL2bd(VppTestCase): :param count: Integer number of hosts to create MAC/IPv4 addresses for. """ - n_int = len(self.pg_interfaces) + n_int = len(cls.pg_interfaces) macs_per_if = count / n_int i = -1 - for pg_if in self.pg_interfaces: + for pg_if in cls.pg_interfaces: i += 1 start_nr = macs_per_if * i end_nr = count if i == (n_int - 1) else macs_per_if * (i + 1) - self.hosts_by_pg_idx[pg_if.sw_if_index] = [] - hosts = self.hosts_by_pg_idx[pg_if.sw_if_index] + cls.hosts_by_pg_idx[pg_if.sw_if_index] = [] + hosts = cls.hosts_by_pg_idx[pg_if.sw_if_index] packets = [] for j in range(start_nr, end_nr): host = Host( @@ -109,12 +141,20 @@ class TestL2bd(VppTestCase): packet = pg_if.sub_if.add_dot1_layer(packet) packets.append(packet) pg_if.add_stream(packets) - info("Sending broadcast eth frames for MAC learning") - self.pg_start() + cls.logger.info("Sending broadcast eth frames for MAC learning") + cls.pg_start() + + def create_stream(self, src_if, packet_sizes, packets_per_burst): + """ + Create input packet stream for defined interface. - def create_stream(self, src_if, packet_sizes): + :param object src_if: Interface to create packet stream for. + :param list packet_sizes: List of required packet sizes. + :param int packets_per_burst: Number of packets in burst. + :return: Stream of packets. + """ pkts = [] - for i in range(0, TestL2bd.pkts_per_burst): + for i in range(0, packets_per_burst): dst_if = self.flows[src_if][i % 2] dst_host = random.choice(self.hosts_by_pg_idx[dst_if.sw_if_index]) src_host = random.choice(self.hosts_by_pg_idx[src_if.sw_if_index]) @@ -128,12 +168,18 @@ class TestL2bd(VppTestCase): pkt_info.data = p.copy() if hasattr(src_if, 'sub_if'): p = src_if.sub_if.add_dot1_layer(p) - size = packet_sizes[(i / 2) % len(packet_sizes)] + size = random.choice(packet_sizes) self.extend_packet(p, size) pkts.append(p) return pkts def verify_capture(self, pg_if, capture): + """ + Verify captured input packet stream for defined interface. + + :param object pg_if: Interface to verify captured packet stream for. + :param list capture: Captured packet stream. + """ last_info = dict() for i in self.pg_interfaces: last_info[i.sw_if_index] = None @@ -156,8 +202,8 @@ class TestL2bd(VppTestCase): udp = packet[UDP] packet_index = payload_info.index self.assertEqual(payload_info.dst, dst_sw_if_index) - debug("Got packet on port %s: src=%u (id=%u)" % - (pg_if.name, payload_info.src, packet_index)) + self.logger.debug("Got packet on port %s: src=%u (id=%u)" % + (pg_if.name, payload_info.src, packet_index)) next_info = self.get_next_packet_info_for_interface2( payload_info.src, dst_sw_if_index, last_info[payload_info.src]) @@ -171,8 +217,8 @@ class TestL2bd(VppTestCase): self.assertEqual(udp.sport, saved_packet[UDP].sport) self.assertEqual(udp.dport, saved_packet[UDP].dport) except: - error("Unexpected or invalid packet:") - error(packet.show()) + self.logger.error("Unexpected or invalid packet:") + self.logger.error(packet.show()) raise for i in self.pg_interfaces: remaining_packet = self.get_next_packet_info_for_interface2( @@ -182,25 +228,14 @@ class TestL2bd(VppTestCase): "Port %u: Packet expected from source %u didn't arrive" % (dst_sw_if_index, i.sw_if_index)) - def test_l2bd(self): - """ L2BD MAC learning test - - 1.config - MAC learning enabled - learn 100 MAC enries - 3 interfaces: untagged, dot1q, dot1ad (dot1q used instead of dot1ad - in the first version) - - 2.sending l2 eth pkts between 3 interface - 64B, 512B, 1518B, 9200B (ether_size) - burst of 257 pkts per interface - """ + def run_l2bd_test(self, pkts_per_burst): + """ L2BD MAC learning test """ # Create incoming packet streams for packet-generator interfaces for i in self.pg_interfaces: packet_sizes = self.sub_if_packet_sizes if hasattr(i, 'sub_if') \ else self.pg_if_packet_sizes - pkts = self.create_stream(i, packet_sizes) + pkts = self.create_stream(i, packet_sizes, pkts_per_burst) i.add_stream(pkts) # Enable packet capture and start packet sending @@ -210,9 +245,43 @@ class TestL2bd(VppTestCase): # Verify outgoing packet streams per packet-generator interface for i in self.pg_interfaces: capture = i.get_capture() - info("Verifying capture on interface %s" % i.name) + self.logger.info("Verifying capture on interface %s" % i.name) self.verify_capture(i, capture) + def test_l2bd_sl(self): + """ L2BD MAC learning single-loop test + + Test scenario: + 1.config + MAC learning enabled + learn 100 MAC enries + 3 interfaces: untagged, dot1q, dot1ad (dot1q used instead of + dot1ad in the first version) + + 2.sending l2 eth pkts between 3 interface + 64B, 512B, 1518B, 9200B (ether_size) + burst of 2 pkts per interface + """ + + self.run_l2bd_test(self.sl_pkts_per_burst) + + def test_l2bd_dl(self): + """ L2BD MAC learning dual-loop test + + Test scenario: + 1.config + MAC learning enabled + learn 100 MAC enries + 3 interfaces: untagged, dot1q, dot1ad (dot1q used instead of + dot1ad in the first version) + + 2.sending l2 eth pkts between 3 interface + 64B, 512B, 1518B, 9200B (ether_size) + burst of 257 pkts per interface + """ + + self.run_l2bd_test(self.dl_pkts_per_burst) + if __name__ == '__main__': unittest.main(testRunner=VppTestRunner) |