summaryrefslogtreecommitdiffstats
path: root/test/template_bd.py
diff options
context:
space:
mode:
authorEyal Bari <ebari@cisco.com>2016-12-20 18:36:46 +0200
committerJohn Lo <loj@cisco.com>2017-01-06 20:29:33 +0000
commitc4aaee11468aa5ed7af01d0747d912493cff002d (patch)
tree86d5d920af04e2627c5ccaaa9cd1ded004c13783 /test/template_bd.py
parentef6e54de4a52b1f8b37c6f074b6803e54d5db327 (diff)
Added basic tests for multicast vxlan tunnels
unicast flood test - test headend replication multicast flood test - test flooding when a multicast vxlan tunnel is present in BD multicast receive test - verify that multicast packet are received on their corresponding unicast tunnels and that unmatched packets are dropped all tests run after adding and removing 200 mcast tunnels to test stability Change-Id: Ia05108c39ac35096a5b633cf52480a9ba87c14df Signed-off-by: Eyal Bari <ebari@cisco.com>
Diffstat (limited to 'test/template_bd.py')
-rw-r--r--test/template_bd.py119
1 files changed, 92 insertions, 27 deletions
diff --git a/test/template_bd.py b/test/template_bd.py
index d70648b41db..6c384922bbd 100644
--- a/test/template_bd.py
+++ b/test/template_bd.py
@@ -11,23 +11,28 @@ class BridgeDomain(object):
__metaclass__ = ABCMeta
@property
- def frame_pg0_to_pg1(self):
- """ Ethernet frame sent from pg0 and expected to arrive at pg1 """
+ def frame_request(self):
+ """ Ethernet frame modeling a generic request """
return (Ether(src='00:00:00:00:00:01', dst='00:00:00:00:00:02') /
IP(src='1.2.3.4', dst='4.3.2.1') /
UDP(sport=10000, dport=20000) /
Raw('\xa5' * 100))
@property
- def frame_pg1_to_pg0(self):
- """ Ethernet frame sent from pg1 and expected to arrive at pg0 """
+ def frame_reply(self):
+ """ Ethernet frame modeling a generic reply """
return (Ether(src='00:00:00:00:00:02', dst='00:00:00:00:00:01') /
IP(src='4.3.2.1', dst='1.2.3.4') /
UDP(sport=20000, dport=10000) /
Raw('\xa5' * 100))
@abstractmethod
- def encapsulate(self, pkt):
+ def encap_mcast(self, pkt, src_ip, src_mac, vni):
+ """ Encapsulate mcast packet """
+ pass
+
+ @abstractmethod
+ def encapsulate(self, pkt, vni):
""" Encapsulate packet """
pass
@@ -37,17 +42,30 @@ class BridgeDomain(object):
pass
@abstractmethod
- def check_encapsulation(self, pkt):
+ def check_encapsulation(self, pkt, vni, local_only=False):
""" Verify the encapsulation """
pass
+ def assert_eq_pkts(self, pkt1, pkt2):
+ """ Verify the Ether, IP, UDP, payload are equal in both
+ packets
+ """
+ self.assertEqual(pkt1[Ether].src, pkt2[Ether].src)
+ self.assertEqual(pkt1[Ether].dst, pkt2[Ether].dst)
+ self.assertEqual(pkt1[IP].src, pkt2[IP].src)
+ self.assertEqual(pkt1[IP].dst, pkt2[IP].dst)
+ self.assertEqual(pkt1[UDP].sport, pkt2[UDP].sport)
+ self.assertEqual(pkt1[UDP].dport, pkt2[UDP].dport)
+ self.assertEqual(pkt1[Raw], pkt2[Raw])
+
def test_decap(self):
""" Decapsulation test
Send encapsulated frames from pg0
Verify receipt of decapsulated frames on pg1
"""
- encapsulated_pkt = self.encapsulate(self.frame_pg0_to_pg1)
+ encapsulated_pkt = self.encapsulate(self.frame_request,
+ self.single_tunnel_bd)
self.pg0.add_stream([encapsulated_pkt, ])
@@ -55,25 +73,18 @@ class BridgeDomain(object):
self.pg_start()
- # Pick first received frame and check if it's the non-encapsulated frame
+ # Pick first received frame and check if it's the
+ # non-encapsulated frame
out = self.pg1.get_capture(1)
pkt = out[0]
-
- # TODO: add error messages
- self.assertEqual(pkt[Ether].src, self.frame_pg0_to_pg1[Ether].src)
- self.assertEqual(pkt[Ether].dst, self.frame_pg0_to_pg1[Ether].dst)
- self.assertEqual(pkt[IP].src, self.frame_pg0_to_pg1[IP].src)
- self.assertEqual(pkt[IP].dst, self.frame_pg0_to_pg1[IP].dst)
- self.assertEqual(pkt[UDP].sport, self.frame_pg0_to_pg1[UDP].sport)
- self.assertEqual(pkt[UDP].dport, self.frame_pg0_to_pg1[UDP].dport)
- self.assertEqual(pkt[Raw], self.frame_pg0_to_pg1[Raw])
+ self.assert_eq_pkts(pkt, self.frame_request)
def test_encap(self):
""" Encapsulation test
Send frames from pg1
Verify receipt of encapsulated frames on pg0
"""
- self.pg1.add_stream([self.frame_pg1_to_pg0])
+ self.pg1.add_stream([self.frame_reply])
self.pg0.enable_capture()
@@ -82,14 +93,68 @@ class BridgeDomain(object):
# Pick first received frame and check if it's corectly encapsulated.
out = self.pg0.get_capture(1)
pkt = out[0]
- self.check_encapsulation(pkt)
+ self.check_encapsulation(pkt, self.single_tunnel_bd)
payload = self.decapsulate(pkt)
- # TODO: add error messages
- self.assertEqual(payload[Ether].src, self.frame_pg1_to_pg0[Ether].src)
- self.assertEqual(payload[Ether].dst, self.frame_pg1_to_pg0[Ether].dst)
- self.assertEqual(payload[IP].src, self.frame_pg1_to_pg0[IP].src)
- self.assertEqual(payload[IP].dst, self.frame_pg1_to_pg0[IP].dst)
- self.assertEqual(payload[UDP].sport, self.frame_pg1_to_pg0[UDP].sport)
- self.assertEqual(payload[UDP].dport, self.frame_pg1_to_pg0[UDP].dport)
- self.assertEqual(payload[Raw], self.frame_pg1_to_pg0[Raw])
+ self.assert_eq_pkts(payload, self.frame_reply)
+
+ def test_ucast_flood(self):
+ """ Unicast flood test
+ Send frames from pg3
+ Verify receipt of encapsulated frames on pg0
+ """
+ self.pg3.add_stream([self.frame_reply])
+
+ self.pg0.enable_capture()
+
+ self.pg_start()
+
+ # Get packet from each tunnel and assert it's corectly encapsulated.
+ out = self.pg0.get_capture(10)
+ for pkt in out:
+ self.check_encapsulation(pkt, self.ucast_flood_bd, True)
+ payload = self.decapsulate(pkt)
+ self.assert_eq_pkts(payload, self.frame_reply)
+
+ def test_mcast_flood(self):
+ """ Multicast flood test
+ Send frames from pg2
+ Verify receipt of encapsulated frames on pg0
+ """
+ self.pg2.add_stream([self.frame_reply])
+
+ self.pg0.enable_capture()
+
+ self.pg_start()
+
+ # Pick first received frame and check if it's corectly encapsulated.
+ out = self.pg0.get_capture(1)
+ pkt = out[0]
+ self.check_encapsulation(pkt, self.mcast_flood_bd, True)
+
+ payload = self.decapsulate(pkt)
+ self.assert_eq_pkts(payload, self.frame_reply)
+
+ @staticmethod
+ def ipn_to_ip(ipn):
+ return '.'.join(str(i) for i in bytearray(ipn))
+
+ def test_mcast_rcv(self):
+ """ Multicast receive test
+ Send 20 encapsulated frames from pg0 only 10 match unicast tunnels
+ Verify receipt of 10 decap frames on pg2
+ """
+ mac = self.pg0.remote_mac
+ ip_range_start = 10
+ ip_range_end = 30
+ mcast_stream = [
+ self.encap_mcast(self.frame_request, self.ipn_to_ip(ip), mac,
+ self.mcast_flood_bd)
+ for ip in self.ip4_range(self.pg0.remote_ip4n,
+ ip_range_start, ip_range_end)]
+ self.pg0.add_stream(mcast_stream)
+ self.pg2.enable_capture()
+ self.pg_start()
+ out = self.pg2.get_capture(10)
+ for pkt in out:
+ self.assert_eq_pkts(pkt, self.frame_request)