X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Ftemplate_bd.py;h=55aaa5a8f4c69d21f993abccfee49524e2b526ba;hb=56e17cf7a23831b353bc1836def47dad7f869dad;hp=91d5dd719f42b856b16d7c19cf8b07ca5e054bbd;hpb=d81da8c0b5094de6635ad9575a2a2bb6f44070dc;p=vpp.git diff --git a/test/template_bd.py b/test/template_bd.py index 91d5dd719f4..55aaa5a8f4c 100644 --- a/test/template_bd.py +++ b/test/template_bd.py @@ -1,55 +1,62 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 -from abc import abstractmethod, ABCMeta +import abc -from scapy.layers.l2 import Ether, Raw +from scapy.layers.l2 import Ether +from scapy.packet import Raw from scapy.layers.inet import IP, UDP -from util import ip4_range - -class BridgeDomain(object): - """ Bridge domain abstraction """ - __metaclass__ = ABCMeta +class BridgeDomain(metaclass=abc.ABCMeta): + """Bridge domain abstraction""" @property 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)) + """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_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)) + """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) + ) + + @abc.abstractmethod + def ip_range(self, start, end): + """range of remote ip's""" + pass - @abstractmethod + @abc.abstractmethod def encap_mcast(self, pkt, src_ip, src_mac, vni): - """ Encapsulate mcast packet """ + """Encapsulate mcast packet""" pass - @abstractmethod + @abc.abstractmethod def encapsulate(self, pkt, vni): - """ Encapsulate packet """ + """Encapsulate packet""" pass - @abstractmethod + @abc.abstractmethod def decapsulate(self, pkt): - """ Decapsulate packet """ + """Decapsulate packet""" pass - @abstractmethod + @abc.abstractmethod def check_encapsulation(self, pkt, vni, local_only=False): - """ Verify the encapsulation """ + """Verify the encapsulation""" pass def assert_eq_pkts(self, pkt1, pkt2): - """ Verify the Ether, IP, UDP, payload are equal in both + """Verify the Ether, IP, UDP, payload are equal in both packets """ self.assertEqual(pkt1[Ether].src, pkt2[Ether].src) @@ -61,28 +68,31 @@ class BridgeDomain(object): self.assertEqual(pkt1[Raw], pkt2[Raw]) def test_decap(self): - """ Decapsulation test + """Decapsulation test Send encapsulated frames from pg0 Verify receipt of decapsulated frames on pg1 """ - encapsulated_pkt = self.encapsulate(self.frame_request, - self.single_tunnel_bd) + encapsulated_pkt = self.encapsulate(self.frame_request, self.single_tunnel_vni) - self.pg0.add_stream([encapsulated_pkt, ]) + self.pg0.add_stream( + [ + encapsulated_pkt, + ] + ) self.pg1.enable_capture() 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] self.assert_eq_pkts(pkt, self.frame_request) def test_encap(self): - """ Encapsulation test + """Encapsulation test Send frames from pg1 Verify receipt of encapsulated frames on pg0 """ @@ -92,16 +102,16 @@ class BridgeDomain(object): self.pg_start() - # Pick first received frame and check if it's corectly encapsulated. + # Pick first received frame and check if it's correctly encapsulated. out = self.pg0.get_capture(1) pkt = out[0] - self.check_encapsulation(pkt, self.single_tunnel_bd) + self.check_encapsulation(pkt, self.single_tunnel_vni) payload = self.decapsulate(pkt) self.assert_eq_pkts(payload, self.frame_reply) def test_ucast_flood(self): - """ Unicast flood test + """Unicast flood test Send frames from pg3 Verify receipt of encapsulated frames on pg0 """ @@ -111,7 +121,7 @@ class BridgeDomain(object): self.pg_start() - # Get packet from each tunnel and assert it's corectly encapsulated. + # Get packet from each tunnel and assert it's correctly encapsulated. out = self.pg0.get_capture(self.n_ucast_tunnels) for pkt in out: self.check_encapsulation(pkt, self.ucast_flood_bd, True) @@ -119,7 +129,7 @@ class BridgeDomain(object): self.assert_eq_pkts(payload, self.frame_reply) def test_mcast_flood(self): - """ Multicast flood test + """Multicast flood test Send frames from pg2 Verify receipt of encapsulated frames on pg0 """ @@ -129,16 +139,18 @@ class BridgeDomain(object): self.pg_start() - # Pick first received frame and check if it's corectly encapsulated. + # Pick first received frame and check if it's correctly encapsulated. out = self.pg0.get_capture(1) pkt = out[0] - self.check_encapsulation(pkt, self.mcast_flood_bd, True) + self.check_encapsulation( + pkt, self.mcast_flood_bd, local_only=False, mcast_pkt=True + ) payload = self.decapsulate(pkt) self.assert_eq_pkts(payload, self.frame_reply) def test_mcast_rcv(self): - """ Multicast receive test + """Multicast receive test Send 20 encapsulated frames from pg0 only 10 match unicast tunnels Verify receipt of 10 decap frames on pg2 """ @@ -147,8 +159,8 @@ class BridgeDomain(object): ip_range_end = 30 mcast_stream = [ self.encap_mcast(self.frame_request, ip, mac, self.mcast_flood_bd) - for ip in ip4_range(self.pg0.remote_ip4, - ip_range_start, ip_range_end)] + for ip in self.ip_range(ip_range_start, ip_range_end) + ] self.pg0.add_stream(mcast_stream) self.pg2.enable_capture() self.pg_start()