test/test_l2bd_learnlimit.py

   1 #!/usr/bin/env python3
   2
   3 import unittest
   4 import random
   5
   6 from scapy.packet import Raw
   7 from scapy.layers.l2 import Ether
   8 from scapy.layers.inet import IP, UDP
   9
  10 from framework import VppTestCase, VppTestRunner
  11 from util import Host, ppp
  12
  13
  14 class TestL2LearnLimit(VppTestCase):
  15     """L2 Learn no limit Test Case"""
  16
  17     @classmethod
  18     def setUpClass(self):
  19         super(TestL2LearnLimit, self).setUpClass()
  20         self.create_pg_interfaces(range(3))
  21
  22     @classmethod
  23     def tearDownClass(cls):
  24         super(TestL2LearnLimit, cls).tearDownClass()
  25
  26     def create_hosts(self, pg_if, n_hosts_per_if, subnet):
  27         """
  28         Create required number of host MAC addresses and distribute them among
  29         interfaces. Create host IPv4 address for every host MAC address.
  30
  31         :param int n_hosts_per_if: Number of per interface hosts to
  32             create MAC/IPv4 addresses for.
  33         """
  34
  35         hosts = dict()
  36         swif = pg_if.sw_if_index
  37
  38         def mac(j):
  39             return "00:00:%02x:ff:%02x:%02x" % (subnet, swif, j)
  40
  41         def ip(j):
  42             return "172.%02u.1%02x.%u" % (subnet, swif, j)
  43
  44         def h(j):
  45             return Host(mac(j), ip(j))
  46
  47         hosts[swif] = [h(j) for j in range(n_hosts_per_if)]
  48
  49         return hosts
  50
  51     def learn_hosts(self, pg_if, bd_id, hosts):
  52         """
  53         Create and send per interface L2 MAC broadcast packet stream to
  54         let the bridge domain learn these MAC addresses.
  55
  56         :param int bd_id: BD to teach
  57         :param dict hosts: dict of hosts per interface
  58         """
  59
  60         self.vapi.bridge_flags(bd_id=bd_id, is_set=1, flags=1)
  61
  62         swif = pg_if.sw_if_index
  63         packets = [Ether(dst="ff:ff:ff:ff:ff:ff", src=host.mac) for host in hosts[swif]]
  64         pg_if.add_stream(packets)
  65         self.logger.info("Sending broadcast eth frames for MAC learning")
  66         self.pg_start()
  67
  68     def test_l2bd_learnlimit(self):
  69         """L2BD test without learn Limit"""
  70         hosts = self.create_hosts(self.pg_interfaces[0], 20, 1)
  71         self.learn_hosts(self.pg_interfaces[0], 1, hosts)
  72         lfs = self.vapi.l2_fib_table_dump(1)
  73
  74         # check that 20 macs are learned.
  75         self.assertEqual(len(lfs), 20)
  76
  77     def setUp(self):
  78         super(TestL2LearnLimit, self).setUp()
  79
  80         self.vapi.bridge_domain_add_del(bd_id=1)
  81         self.vapi.bridge_domain_add_del(bd_id=2)
  82
  83         self.vapi.sw_interface_set_l2_bridge(self.pg_interfaces[0].sw_if_index, bd_id=1)
  84         self.vapi.sw_interface_set_l2_bridge(self.pg_interfaces[1].sw_if_index, bd_id=2)
  85
  86     def tearDown(self):
  87         super(TestL2LearnLimit, self).tearDown()
  88         self.vapi.sw_interface_set_l2_bridge(
  89             rx_sw_if_index=self.pg_interfaces[0].sw_if_index, bd_id=1, enable=0
  90         )
  91         self.vapi.sw_interface_set_l2_bridge(
  92             rx_sw_if_index=self.pg_interfaces[1].sw_if_index, bd_id=2, enable=0
  93         )
  94         self.vapi.bridge_domain_add_del(bd_id=1, is_add=0)
  95         self.vapi.bridge_domain_add_del(bd_id=2, is_add=0)
  96
  97
  98 if __name__ == "__main__":
  99     unittest.main(testRunner=VppTestRunner)