diff options
Diffstat (limited to 'test/test_srv6_un.py')
-rw-r--r-- | test/test_srv6_un.py | 121 |
1 files changed, 56 insertions, 65 deletions
diff --git a/test/test_srv6_un.py b/test/test_srv6_un.py index 73285bbac87..a5651f52edb 100644 --- a/test/test_srv6_un.py +++ b/test/test_srv6_un.py @@ -1,14 +1,13 @@ #!/usr/bin/env python3 from framework import VppTestCase -from ipaddress import IPv4Address from ipaddress import IPv6Address from scapy.contrib.gtp import * from scapy.all import * class TestSRv6uSIDSRH(VppTestCase): - """ SRv6 End.uSID w/ SRH """ + """SRv6 End.uSID w/ SRH""" @classmethod def setUpClass(cls): @@ -35,7 +34,6 @@ class TestSRv6uSIDSRH(VppTestCase): raise def create_packets(self, inner): - ip6_dst = IPv6Address(str(self.ip6_dst)) ip6_src = IPv6Address(str(self.ip6_src)) @@ -45,30 +43,30 @@ class TestSRv6uSIDSRH(VppTestCase): pkts = list() for d, s in inner: - pkt = (Ether() / - IPv6(dst=str(ip6_dst), src=str(ip6_src)) / - IPv6ExtHdrSegmentRouting(segleft=1, - lastentry=0, - tag=0, - addresses=[ - "a1::1", - "1111:2222:aaaa:bbbb::"]) / - IPv6(dst=d, src=s) / - UDP(sport=1000, dport=23)) + pkt = ( + Ether() + / IPv6(dst=str(ip6_dst), src=str(ip6_src)) + / IPv6ExtHdrSegmentRouting( + segleft=1, + lastentry=0, + tag=0, + addresses=["a1::1", "1111:2222:aaaa:bbbb::"], + ) + / IPv6(dst=d, src=s) + / UDP(sport=1000, dport=23) + ) self.logger.info(pkt.show2(dump=True)) pkts.append(pkt) return pkts def test_srv6_usid_srh(self): - """ test_srv6_usid_srh """ + """test_srv6_usid_srh""" pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")]) self.vapi.cli("set sr encaps source addr A1::1") - self.vapi.cli( - "sr localsid prefix 1111:2222:aaaa::/48 behavior un 16") - self.vapi.cli( - "ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) + self.vapi.cli("sr localsid prefix 1111:2222:aaaa::/48 behavior un 16") + self.vapi.cli("ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) self.logger.info(self.vapi.cli("show sr localsids")) @@ -85,12 +83,11 @@ class TestSRv6uSIDSRH(VppTestCase): for pkt in capture: self.logger.info(pkt.show2(dump=True)) - self.assertEqual( - pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") + self.assertEqual(pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") class TestSRv6uSID(VppTestCase): - """ SRv6 End.uSID w/o SRH """ + """SRv6 End.uSID w/o SRH""" @classmethod def setUpClass(cls): @@ -117,7 +114,6 @@ class TestSRv6uSID(VppTestCase): raise def create_packets(self, inner): - ip6_dst = IPv6Address(str(self.ip6_dst)) ip6_src = IPv6Address(str(self.ip6_src)) @@ -127,24 +123,24 @@ class TestSRv6uSID(VppTestCase): pkts = list() for d, s in inner: - pkt = (Ether() / - IPv6(dst=str(ip6_dst), src=str(ip6_src)) / - IPv6(dst=d, src=s) / - UDP(sport=1000, dport=23)) + pkt = ( + Ether() + / IPv6(dst=str(ip6_dst), src=str(ip6_src)) + / IPv6(dst=d, src=s) + / UDP(sport=1000, dport=23) + ) self.logger.info(pkt.show2(dump=True)) pkts.append(pkt) return pkts def test_srv6_usid(self): - """ test_srv6_usid """ + """test_srv6_usid""" pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")]) self.vapi.cli("set sr encaps source addr A1::1") - self.vapi.cli( - "sr localsid prefix 1111:2222:aaaa::/48 behavior un 16") - self.vapi.cli( - "ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) + self.vapi.cli("sr localsid prefix 1111:2222:aaaa::/48 behavior un 16") + self.vapi.cli("ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) self.logger.info(self.vapi.cli("show sr localsids")) @@ -161,12 +157,11 @@ class TestSRv6uSID(VppTestCase): for pkt in capture: self.logger.info(pkt.show2(dump=True)) - self.assertEqual( - pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") + self.assertEqual(pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") class TestSRv6uSIDFlexSRH(VppTestCase): - """ SRv6 End.uSID.Flex w/ SRH """ + """SRv6 End.uSID.Flex w/ SRH""" @classmethod def setUpClass(cls): @@ -193,7 +188,6 @@ class TestSRv6uSIDFlexSRH(VppTestCase): raise def create_packets(self, inner): - ip6_dst = IPv6Address(str(self.ip6_dst)) ip6_src = IPv6Address(str(self.ip6_src)) @@ -203,30 +197,30 @@ class TestSRv6uSIDFlexSRH(VppTestCase): pkts = list() for d, s in inner: - pkt = (Ether() / - IPv6(dst=str(ip6_dst), src=str(ip6_src)) / - IPv6ExtHdrSegmentRouting(segleft=1, - lastentry=0, - tag=0, - addresses=[ - "a1::1", - "1111:2222:aaaa:bbbb::"]) / - IPv6(dst=d, src=s) / - UDP(sport=1000, dport=23)) + pkt = ( + Ether() + / IPv6(dst=str(ip6_dst), src=str(ip6_src)) + / IPv6ExtHdrSegmentRouting( + segleft=1, + lastentry=0, + tag=0, + addresses=["a1::1", "1111:2222:aaaa:bbbb::"], + ) + / IPv6(dst=d, src=s) + / UDP(sport=1000, dport=23) + ) self.logger.info(pkt.show2(dump=True)) pkts.append(pkt) return pkts def test_srv6_usid_flex_srh(self): - """ test_srv6_usid_flex_srh """ + """test_srv6_usid_flex_srh""" pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")]) self.vapi.cli("set sr encaps source addr A1::1") - self.vapi.cli( - "sr localsid prefix 1111:2222:aaaa::/48 behavior un.flex 16") - self.vapi.cli( - "ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) + self.vapi.cli("sr localsid prefix 1111:2222:aaaa::/48 behavior un.flex 16") + self.vapi.cli("ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) self.logger.info(self.vapi.cli("show sr localsids")) @@ -243,12 +237,11 @@ class TestSRv6uSIDFlexSRH(VppTestCase): for pkt in capture: self.logger.info(pkt.show2(dump=True)) - self.assertEqual( - pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") + self.assertEqual(pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") class TestSRv6uSIDFlex(VppTestCase): - """ SRv6 End.uSID.Flex w/o SRH """ + """SRv6 End.uSID.Flex w/o SRH""" @classmethod def setUpClass(cls): @@ -275,7 +268,6 @@ class TestSRv6uSIDFlex(VppTestCase): raise def create_packets(self, inner): - ip6_dst = IPv6Address(str(self.ip6_dst)) ip6_src = IPv6Address(str(self.ip6_src)) @@ -285,24 +277,24 @@ class TestSRv6uSIDFlex(VppTestCase): pkts = list() for d, s in inner: - pkt = (Ether() / - IPv6(dst=str(ip6_dst), src=str(ip6_src)) / - IPv6(dst=d, src=s) / - UDP(sport=1000, dport=23)) + pkt = ( + Ether() + / IPv6(dst=str(ip6_dst), src=str(ip6_src)) + / IPv6(dst=d, src=s) + / UDP(sport=1000, dport=23) + ) self.logger.info(pkt.show2(dump=True)) pkts.append(pkt) return pkts def test_srv6_usid_flex(self): - """ test_srv6_usid_flex """ + """test_srv6_usid_flex""" pkts = self.create_packets([("A::1", "B::1"), ("C::1", "D::1")]) self.vapi.cli("set sr encaps source addr A1::1") - self.vapi.cli( - "sr localsid prefix 1111:2222:aaaa::/48 behavior un.flex 16") - self.vapi.cli( - "ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) + self.vapi.cli("sr localsid prefix 1111:2222:aaaa::/48 behavior un.flex 16") + self.vapi.cli("ip route add 1111:2222:bbbb::/48 via {}".format(self.ip6_nhop)) self.logger.info(self.vapi.cli("show sr localsids")) @@ -319,5 +311,4 @@ class TestSRv6uSIDFlex(VppTestCase): for pkt in capture: self.logger.info(pkt.show2(dump=True)) - self.assertEqual( - pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") + self.assertEqual(pkt[IPv6].dst, "1111:2222:bbbb:cccc:dddd:eeee:ffff:0") |