diff options
author | Klement Sekera <klement.sekera@gmail.com> | 2022-04-26 19:02:15 +0200 |
---|---|---|
committer | Ole Tr�an <otroan@employees.org> | 2022-05-10 18:52:08 +0000 |
commit | d9b0c6fbf7aa5bd9af84264105b39c82028a4a29 (patch) | |
tree | 4f786cfd8ebc2443cb11e11b74c8657204068898 /test/test_igmp.py | |
parent | f90348bcb4afd0af2611cefc43b17ef3042b511c (diff) |
tests: replace pycodestyle with black
Drop pycodestyle for code style checking in favor of black. Black is
much faster, stable PEP8 compliant code style checker offering also
automatic formatting. It aims to be very stable and produce smallest
diffs. It's used by many small and big projects.
Running checkstyle with black takes a few seconds with a terse output.
Thus, test-checkstyle-diff is no longer necessary.
Expand scope of checkstyle to all python files in the repo, replacing
test-checkstyle with checkstyle-python.
Also, fixstyle-python is now available for automatic style formatting.
Note: python virtualenv has been consolidated in test/Makefile,
test/requirements*.txt which will eventually be moved to a central
location. This is required to simply the automated generation of
docker executor images in the CI.
Type: improvement
Change-Id: I022a326603485f58585e879ac0f697fceefbc9c8
Signed-off-by: Klement Sekera <klement.sekera@gmail.com>
Signed-off-by: Dave Wallace <dwallacelf@gmail.com>
Diffstat (limited to 'test/test_igmp.py')
-rw-r--r-- | test/test_igmp.py | 798 |
1 files changed, 477 insertions, 321 deletions
diff --git a/test/test_igmp.py b/test/test_igmp.py index f99bdb25c8c..6e9defd4c1d 100644 --- a/test/test_igmp.py +++ b/test/test_igmp.py @@ -9,8 +9,15 @@ from scapy.contrib.igmpv3 import IGMPv3, IGMPv3gr, IGMPv3mq, IGMPv3mr from framework import tag_fixme_vpp_workers from framework import VppTestCase, VppTestRunner -from vpp_igmp import find_igmp_state, IGMP_FILTER, IgmpRecord, IGMP_MODE, \ - IgmpSG, VppHostState, wait_for_igmp_event +from vpp_igmp import ( + find_igmp_state, + IGMP_FILTER, + IgmpRecord, + IGMP_MODE, + IgmpSG, + VppHostState, + wait_for_igmp_event, +) from vpp_ip_route import find_mroute, VppIpTable @@ -21,7 +28,7 @@ class IgmpMode: @tag_fixme_vpp_workers class TestIgmp(VppTestCase): - """ IGMP Test Case """ + """IGMP Test Case""" @classmethod def setUpClass(cls): @@ -63,16 +70,16 @@ class TestIgmp(VppTestCase): self.pg_start() def test_igmp_flush(self): - """ IGMP Link Up/down and Flush """ + """IGMP Link Up/down and Flush""" # # FIX THIS. Link down. # def test_igmp_enable(self): - """ IGMP enable/disable on an interface + """IGMP enable/disable on an interface - check for the addition/removal of the IGMP mroutes """ + check for the addition/removal of the IGMP mroutes""" self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 1, IGMP_MODE.HOST) self.vapi.igmp_enable_disable(self.pg1.sw_if_index, 1, IGMP_MODE.HOST) @@ -83,10 +90,8 @@ class TestIgmp(VppTestCase): self.vapi.igmp_enable_disable(self.pg2.sw_if_index, 1, IGMP_MODE.HOST) self.vapi.igmp_enable_disable(self.pg3.sw_if_index, 1, IGMP_MODE.HOST) - self.assertTrue(find_mroute(self, "224.0.0.1", "0.0.0.0", 32, - table_id=1)) - self.assertTrue(find_mroute(self, "224.0.0.22", "0.0.0.0", 32, - table_id=1)) + self.assertTrue(find_mroute(self, "224.0.0.1", "0.0.0.0", 32, table_id=1)) + self.assertTrue(find_mroute(self, "224.0.0.22", "0.0.0.0", 32, table_id=1)) self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 0, IGMP_MODE.HOST) self.vapi.igmp_enable_disable(self.pg1.sw_if_index, 0, IGMP_MODE.HOST) self.vapi.igmp_enable_disable(self.pg2.sw_if_index, 0, IGMP_MODE.HOST) @@ -94,10 +99,8 @@ class TestIgmp(VppTestCase): self.assertTrue(find_mroute(self, "224.0.0.1", "0.0.0.0", 32)) self.assertFalse(find_mroute(self, "224.0.0.22", "0.0.0.0", 32)) - self.assertTrue(find_mroute(self, "224.0.0.1", "0.0.0.0", 32, - table_id=1)) - self.assertFalse(find_mroute(self, "224.0.0.22", "0.0.0.0", 32, - table_id=1)) + self.assertTrue(find_mroute(self, "224.0.0.1", "0.0.0.0", 32, table_id=1)) + self.assertFalse(find_mroute(self, "224.0.0.22", "0.0.0.0", 32, table_id=1)) def verify_general_query(self, p): ip = p[IP] @@ -126,8 +129,9 @@ class TestIgmp(VppTestCase): self.assertEqual(len(ip.options), 1) self.assertEqual(ip.options[0].option, 20) self.assertEqual(ip.proto, 2) - self.assertEqual(IGMPv3.igmpv3types[rx[IGMPv3].type], - "Version 3 Membership Report") + self.assertEqual( + IGMPv3.igmpv3types[rx[IGMPv3].type], "Version 3 Membership Report" + ) self.assertEqual(rx[IGMPv3mr].numgrp, len(records)) received = rx[IGMPv3mr].records @@ -140,26 +144,20 @@ class TestIgmp(VppTestCase): self.assertEqual(gr.maddr, r.sg.gaddr) self.assertEqual(len(gr.srcaddrs), len(r.sg.saddrs)) - self.assertEqual(sorted(gr.srcaddrs), - sorted(r.sg.saddrs)) + self.assertEqual(sorted(gr.srcaddrs), sorted(r.sg.saddrs)) def add_group(self, itf, sg, n_pkts=2): self.pg_enable_capture(self.pg_interfaces) self.pg_start() - hs = VppHostState(self, - IGMP_FILTER.INCLUDE, - itf.sw_if_index, - sg) + hs = VppHostState(self, IGMP_FILTER.INCLUDE, itf.sw_if_index, sg) hs.add_vpp_config() capture = itf.get_capture(n_pkts, timeout=10) # reports are transmitted twice due to default rebostness value=2 - self.verify_report(capture[0], - [IgmpRecord(sg, "Allow New Sources")]), - self.verify_report(capture[1], - [IgmpRecord(sg, "Allow New Sources")]), + self.verify_report(capture[0], [IgmpRecord(sg, "Allow New Sources")]), + self.verify_report(capture[1], [IgmpRecord(sg, "Allow New Sources")]), return hs @@ -170,18 +168,15 @@ class TestIgmp(VppTestCase): capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(hs.sg, "Block Old Sources")]) + self.verify_report(capture[0], [IgmpRecord(hs.sg, "Block Old Sources")]) def test_igmp_host(self): - """ IGMP Host functions """ + """IGMP Host functions""" # # Enable interface for host functions # - self.vapi.igmp_enable_disable(self.pg0.sw_if_index, - 1, - IGMP_MODE.HOST) + self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 1, IGMP_MODE.HOST) # # Add one S,G of state and expect a state-change event report @@ -192,8 +187,7 @@ class TestIgmp(VppTestCase): # search for the corresponding state created in VPP dump = self.vapi.igmp_dump(self.pg0.sw_if_index) self.assertEqual(len(dump), 1) - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.1", "1.1.1.1")) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.1", "1.1.1.1")) # # Send a general query (to the all router's address) @@ -201,61 +195,81 @@ class TestIgmp(VppTestCase): # Pad the query with 0 - some devices in the big wild # internet are prone to this. # - p_g = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst='224.0.0.1', tos=0xc0) / - IGMPv3(type="Membership Query", mrcode=100) / - IGMPv3mq(gaddr="0.0.0.0") / - Raw(b'\x00' * 10)) + p_g = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP(src=self.pg0.remote_ip4, dst="224.0.0.1", tos=0xC0) + / IGMPv3(type="Membership Query", mrcode=100) + / IGMPv3mq(gaddr="0.0.0.0") + / Raw(b"\x00" * 10) + ) self.send(self.pg0, p_g) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h1.sg, "Mode Is Include")]) + self.verify_report(capture[0], [IgmpRecord(h1.sg, "Mode Is Include")]) # # Group specific query # - p_gs = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst='239.1.1.1', tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Membership Query", mrcode=100) / - IGMPv3mq(gaddr="239.1.1.1")) + p_gs = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="239.1.1.1", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Membership Query", mrcode=100) + / IGMPv3mq(gaddr="239.1.1.1") + ) self.send(self.pg0, p_gs) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h1.sg, "Mode Is Include")]) + self.verify_report(capture[0], [IgmpRecord(h1.sg, "Mode Is Include")]) # # A group and source specific query, with the source matching # the source VPP has # - p_gs1 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst='239.1.1.1', tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Membership Query", mrcode=100) / - IGMPv3mq(gaddr="239.1.1.1", srcaddrs=["1.1.1.1"])) + p_gs1 = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="239.1.1.1", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Membership Query", mrcode=100) + / IGMPv3mq(gaddr="239.1.1.1", srcaddrs=["1.1.1.1"]) + ) self.send(self.pg0, p_gs1) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h1.sg, "Mode Is Include")]) + self.verify_report(capture[0], [IgmpRecord(h1.sg, "Mode Is Include")]) # # A group and source specific query that reports more sources # than the packet actually has. # - p_gs2 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst='239.1.1.1', tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Membership Query", mrcode=100) / - IGMPv3mq(gaddr="239.1.1.1", numsrc=4, srcaddrs=["1.1.1.1"])) + p_gs2 = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="239.1.1.1", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Membership Query", mrcode=100) + / IGMPv3mq(gaddr="239.1.1.1", numsrc=4, srcaddrs=["1.1.1.1"]) + ) self.send_and_assert_no_replies(self.pg0, p_gs2, timeout=10) @@ -263,12 +277,19 @@ class TestIgmp(VppTestCase): # A group and source specific query, with the source NOT matching # the source VPP has. There should be no response. # - p_gs2 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst='239.1.1.1', tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Membership Query", mrcode=100) / - IGMPv3mq(gaddr="239.1.1.1", srcaddrs=["1.1.1.2"])) + p_gs2 = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="239.1.1.1", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Membership Query", mrcode=100) + / IGMPv3mq(gaddr="239.1.1.1", srcaddrs=["1.1.1.2"]) + ) self.send_and_assert_no_replies(self.pg0, p_gs2, timeout=10) @@ -277,19 +298,24 @@ class TestIgmp(VppTestCase): # one of which matches the source VPP has. # The report should contain only the source VPP has. # - p_gs3 = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst='239.1.1.1', tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Membership Query", mrcode=100) / - IGMPv3mq(gaddr="239.1.1.1", - srcaddrs=["1.1.1.1", "1.1.1.2", "1.1.1.3"])) + p_gs3 = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="239.1.1.1", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Membership Query", mrcode=100) + / IGMPv3mq(gaddr="239.1.1.1", srcaddrs=["1.1.1.1", "1.1.1.2", "1.1.1.3"]) + ) self.send(self.pg0, p_gs3) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h1.sg, "Mode Is Include")]) + self.verify_report(capture[0], [IgmpRecord(h1.sg, "Mode Is Include")]) # # Two source and group specific queries in quick succession, the @@ -297,13 +323,11 @@ class TestIgmp(VppTestCase): # self.send(self.pg0, [p_gs2, p_gs1]) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h1.sg, "Mode Is Include")]) + self.verify_report(capture[0], [IgmpRecord(h1.sg, "Mode Is Include")]) self.send(self.pg0, [p_gs1, p_gs2]) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h1.sg, "Mode Is Include")]) + self.verify_report(capture[0], [IgmpRecord(h1.sg, "Mode Is Include")]) # # remove state, expect the report for the removal @@ -316,16 +340,15 @@ class TestIgmp(VppTestCase): # # A group with multiple sources # - h2 = self.add_group(self.pg0, - IgmpSG("239.1.1.1", - ["1.1.1.1", "1.1.1.2", "1.1.1.3"])) + h2 = self.add_group( + self.pg0, IgmpSG("239.1.1.1", ["1.1.1.1", "1.1.1.2", "1.1.1.3"]) + ) # search for the corresponding state created in VPP dump = self.vapi.igmp_dump(self.pg0.sw_if_index) self.assertEqual(len(dump), 3) for s in h2.sg.saddrs: - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.1", s)) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.1", s)) # # Send a general query (to the all router's address) # expect VPP to respond with a membership report will all sources @@ -333,53 +356,76 @@ class TestIgmp(VppTestCase): self.send(self.pg0, p_g) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h2.sg, "Mode Is Include")]) + self.verify_report(capture[0], [IgmpRecord(h2.sg, "Mode Is Include")]) # # Group and source specific query; some present some not # - p_gs = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst='239.1.1.1', tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Membership Query", mrcode=100) / - IGMPv3mq(gaddr="239.1.1.1", - srcaddrs=["1.1.1.1", "1.1.1.2", "1.1.1.4"])) + p_gs = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="239.1.1.1", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Membership Query", mrcode=100) + / IGMPv3mq(gaddr="239.1.1.1", srcaddrs=["1.1.1.1", "1.1.1.2", "1.1.1.4"]) + ) self.send(self.pg0, p_gs) capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord( - IgmpSG('239.1.1.1', ["1.1.1.1", "1.1.1.2"]), - "Mode Is Include")]) + self.verify_report( + capture[0], + [ + IgmpRecord( + IgmpSG("239.1.1.1", ["1.1.1.1", "1.1.1.2"]), "Mode Is Include" + ) + ], + ) # # add loads more groups # - h3 = self.add_group(self.pg0, - IgmpSG("239.1.1.2", - ["2.1.1.1", "2.1.1.2", "2.1.1.3"])) - h4 = self.add_group(self.pg0, - IgmpSG("239.1.1.3", - ["3.1.1.1", "3.1.1.2", "3.1.1.3"])) - h5 = self.add_group(self.pg0, - IgmpSG("239.1.1.4", - ["4.1.1.1", "4.1.1.2", "4.1.1.3"])) - h6 = self.add_group(self.pg0, - IgmpSG("239.1.1.5", - ["5.1.1.1", "5.1.1.2", "5.1.1.3"])) - h7 = self.add_group(self.pg0, - IgmpSG("239.1.1.6", - ["6.1.1.1", "6.1.1.2", - "6.1.1.3", "6.1.1.4", - "6.1.1.5", "6.1.1.6", - "6.1.1.7", "6.1.1.8", - "6.1.1.9", "6.1.1.10", - "6.1.1.11", "6.1.1.12", - "6.1.1.13", "6.1.1.14", - "6.1.1.15", "6.1.1.16"])) + h3 = self.add_group( + self.pg0, IgmpSG("239.1.1.2", ["2.1.1.1", "2.1.1.2", "2.1.1.3"]) + ) + h4 = self.add_group( + self.pg0, IgmpSG("239.1.1.3", ["3.1.1.1", "3.1.1.2", "3.1.1.3"]) + ) + h5 = self.add_group( + self.pg0, IgmpSG("239.1.1.4", ["4.1.1.1", "4.1.1.2", "4.1.1.3"]) + ) + h6 = self.add_group( + self.pg0, IgmpSG("239.1.1.5", ["5.1.1.1", "5.1.1.2", "5.1.1.3"]) + ) + h7 = self.add_group( + self.pg0, + IgmpSG( + "239.1.1.6", + [ + "6.1.1.1", + "6.1.1.2", + "6.1.1.3", + "6.1.1.4", + "6.1.1.5", + "6.1.1.6", + "6.1.1.7", + "6.1.1.8", + "6.1.1.9", + "6.1.1.10", + "6.1.1.11", + "6.1.1.12", + "6.1.1.13", + "6.1.1.14", + "6.1.1.15", + "6.1.1.16", + ], + ), + ) # # general query. @@ -390,39 +436,59 @@ class TestIgmp(VppTestCase): capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h3.sg, "Mode Is Include"), - IgmpRecord(h2.sg, "Mode Is Include"), - IgmpRecord(h6.sg, "Mode Is Include"), - IgmpRecord(h4.sg, "Mode Is Include"), - IgmpRecord(h5.sg, "Mode Is Include"), - IgmpRecord(h7.sg, "Mode Is Include")]) + self.verify_report( + capture[0], + [ + IgmpRecord(h3.sg, "Mode Is Include"), + IgmpRecord(h2.sg, "Mode Is Include"), + IgmpRecord(h6.sg, "Mode Is Include"), + IgmpRecord(h4.sg, "Mode Is Include"), + IgmpRecord(h5.sg, "Mode Is Include"), + IgmpRecord(h7.sg, "Mode Is Include"), + ], + ) # # modify a group to add and remove some sources # - h7.sg = IgmpSG("239.1.1.6", - ["6.1.1.1", "6.1.1.2", - "6.1.1.5", "6.1.1.6", - "6.1.1.7", "6.1.1.8", - "6.1.1.9", "6.1.1.10", - "6.1.1.11", "6.1.1.12", - "6.1.1.13", "6.1.1.14", - "6.1.1.15", "6.1.1.16", - "6.1.1.17", "6.1.1.18"]) + h7.sg = IgmpSG( + "239.1.1.6", + [ + "6.1.1.1", + "6.1.1.2", + "6.1.1.5", + "6.1.1.6", + "6.1.1.7", + "6.1.1.8", + "6.1.1.9", + "6.1.1.10", + "6.1.1.11", + "6.1.1.12", + "6.1.1.13", + "6.1.1.14", + "6.1.1.15", + "6.1.1.16", + "6.1.1.17", + "6.1.1.18", + ], + ) self.pg_enable_capture(self.pg_interfaces) self.pg_start() h7.add_vpp_config() capture = self.pg0.get_capture(1, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(IgmpSG("239.1.1.6", - ["6.1.1.17", "6.1.1.18"]), - "Allow New Sources"), - IgmpRecord(IgmpSG("239.1.1.6", - ["6.1.1.3", "6.1.1.4"]), - "Block Old Sources")]) + self.verify_report( + capture[0], + [ + IgmpRecord( + IgmpSG("239.1.1.6", ["6.1.1.17", "6.1.1.18"]), "Allow New Sources" + ), + IgmpRecord( + IgmpSG("239.1.1.6", ["6.1.1.3", "6.1.1.4"]), "Block Old Sources" + ), + ], + ) # # add an additional groups with many sources so that each group @@ -435,27 +501,26 @@ class TestIgmp(VppTestCase): for i in range(128): src_list.append("10.1.1.%d" % i) - h8 = self.add_group(self.pg0, - IgmpSG("238.1.1.1", src_list)) - h9 = self.add_group(self.pg0, - IgmpSG("238.1.1.2", src_list)) + h8 = self.add_group(self.pg0, IgmpSG("238.1.1.1", src_list)) + h9 = self.add_group(self.pg0, IgmpSG("238.1.1.2", src_list)) self.send(self.pg0, p_g) capture = self.pg0.get_capture(4, timeout=10) - self.verify_report(capture[0], - [IgmpRecord(h3.sg, "Mode Is Include"), - IgmpRecord(h2.sg, "Mode Is Include"), - IgmpRecord(h6.sg, "Mode Is Include"), - IgmpRecord(h4.sg, "Mode Is Include"), - IgmpRecord(h5.sg, "Mode Is Include")]) - self.verify_report(capture[1], - [IgmpRecord(h8.sg, "Mode Is Include")]) - self.verify_report(capture[2], - [IgmpRecord(h7.sg, "Mode Is Include")]) - self.verify_report(capture[3], - [IgmpRecord(h9.sg, "Mode Is Include")]) + self.verify_report( + capture[0], + [ + IgmpRecord(h3.sg, "Mode Is Include"), + IgmpRecord(h2.sg, "Mode Is Include"), + IgmpRecord(h6.sg, "Mode Is Include"), + IgmpRecord(h4.sg, "Mode Is Include"), + IgmpRecord(h5.sg, "Mode Is Include"), + ], + ) + self.verify_report(capture[1], [IgmpRecord(h8.sg, "Mode Is Include")]) + self.verify_report(capture[2], [IgmpRecord(h7.sg, "Mode Is Include")]) + self.verify_report(capture[3], [IgmpRecord(h9.sg, "Mode Is Include")]) # # drop the MTU further (so a 128 sized group won't fit) @@ -465,10 +530,12 @@ class TestIgmp(VppTestCase): self.pg_enable_capture(self.pg_interfaces) self.pg_start() - h10 = VppHostState(self, - IGMP_FILTER.INCLUDE, - self.pg0.sw_if_index, - IgmpSG("238.1.1.3", src_list)) + h10 = VppHostState( + self, + IGMP_FILTER.INCLUDE, + self.pg0.sw_if_index, + IgmpSG("238.1.1.3", src_list), + ) h10.add_vpp_config() capture = self.pg0.get_capture(2, timeout=10) @@ -498,32 +565,51 @@ class TestIgmp(VppTestCase): # ADD STATE ON MORE INTERFACES # - self.vapi.igmp_enable_disable(self.pg0.sw_if_index, - 0, - IGMP_MODE.HOST) + self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 0, IGMP_MODE.HOST) def test_igmp_router(self): - """ IGMP Router Functions """ + """IGMP Router Functions""" # # Drop reports when not enabled # - p_j = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, ttl=1, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Allow New Sources", - maddr="239.1.1.1", srcaddrs=["10.1.1.1", "10.1.1.2"])) - p_l = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Block Old Sources", - maddr="239.1.1.1", srcaddrs=["10.1.1.1", "10.1.1.2"])) + p_j = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + ttl=1, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr( + rtype="Allow New Sources", + maddr="239.1.1.1", + srcaddrs=["10.1.1.1", "10.1.1.2"], + ) + ) + p_l = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr( + rtype="Block Old Sources", + maddr="239.1.1.1", + srcaddrs=["10.1.1.1", "10.1.1.2"], + ) + ) self.send(self.pg0, p_j) self.assertFalse(self.vapi.igmp_dump()) @@ -539,9 +625,7 @@ class TestIgmp(VppTestCase): # self.pg_enable_capture(self.pg_interfaces) self.pg_start() - self.vapi.igmp_enable_disable(self.pg0.sw_if_index, - 1, - IGMP_MODE.ROUTER) + self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 1, IGMP_MODE.ROUTER) self.vapi.want_igmp_events(1) # @@ -559,26 +643,28 @@ class TestIgmp(VppTestCase): # self.send(self.pg0, p_j) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.1", 1)) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.2", 1)) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.1", 1) + ) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.2", 1) + ) dump = self.vapi.igmp_dump(self.pg0.sw_if_index) self.assertEqual(len(dump), 2) - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.1", "10.1.1.1")) - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.1", "10.1.1.2")) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.1", "10.1.1.1")) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.1", "10.1.1.2")) # # wait for the per-source timer to expire # the state should be reaped # VPP sends a notification that the group has been left # - self.assertTrue(wait_for_igmp_event(self, 4, self.pg0, - "239.1.1.1", "10.1.1.1", 0)) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.2", 0)) + self.assertTrue( + wait_for_igmp_event(self, 4, self.pg0, "239.1.1.1", "10.1.1.1", 0) + ) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.2", 0) + ) self.assertFalse(self.vapi.igmp_dump()) # @@ -589,10 +675,12 @@ class TestIgmp(VppTestCase): # expired in 3 seconds. # self.send(self.pg0, p_j) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.1", 1)) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.2", 1)) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.1", 1) + ) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.2", 1) + ) dump = self.vapi.igmp_dump(self.pg0.sw_if_index) self.assertEqual(len(dump), 2) @@ -600,33 +688,43 @@ class TestIgmp(VppTestCase): self.verify_general_query(capture[0]) self.verify_general_query(capture[1]) - p_cs = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Mode Is Include", - maddr="239.1.1.1", srcaddrs=["10.1.1.1", "10.1.1.2"])) + p_cs = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr( + rtype="Mode Is Include", + maddr="239.1.1.1", + srcaddrs=["10.1.1.1", "10.1.1.2"], + ) + ) self.send(self.pg0, p_cs) self.sleep(2) dump = self.vapi.igmp_dump(self.pg0.sw_if_index) self.assertEqual(len(dump), 2) - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.1", "10.1.1.1")) - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.1", "10.1.1.2")) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.1", "10.1.1.1")) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.1", "10.1.1.2")) # # wait for the per-source timer to expire # the state should be reaped # - self.assertTrue(wait_for_igmp_event(self, 4, self.pg0, - "239.1.1.1", "10.1.1.1", 0)) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.2", 0)) + self.assertTrue( + wait_for_igmp_event(self, 4, self.pg0, "239.1.1.1", "10.1.1.1", 0) + ) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.2", 0) + ) self.assertFalse(self.vapi.igmp_dump()) # @@ -635,203 +733,261 @@ class TestIgmp(VppTestCase): # self.send(self.pg0, p_j) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.1", 1)) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.2", 1)) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.1", 1) + ) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.2", 1) + ) dump = self.vapi.igmp_dump(self.pg0.sw_if_index) self.assertEqual(len(dump), 2) self.send(self.pg0, p_l) capture = self.pg0.get_capture(1, timeout=3) - self.verify_group_query(capture[0], "239.1.1.1", - ["10.1.1.1", "10.1.1.2"]) + self.verify_group_query(capture[0], "239.1.1.1", ["10.1.1.1", "10.1.1.2"]) # # the group specific query drops the timeout to leave (=1) seconds # - self.assertTrue(wait_for_igmp_event(self, 2, self.pg0, - "239.1.1.1", "10.1.1.1", 0)) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.1", "10.1.1.2", 0)) + self.assertTrue( + wait_for_igmp_event(self, 2, self.pg0, "239.1.1.1", "10.1.1.1", 0) + ) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.1", "10.1.1.2", 0) + ) self.assertFalse(self.vapi.igmp_dump()) self.assertFalse(self.vapi.igmp_dump()) # # a TO_EX({}) / IN_EX({}) is treated like a (*,G) join # - p_j = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, ttl=1, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Change To Exclude Mode", maddr="239.1.1.2")) + p_j = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + ttl=1, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr(rtype="Change To Exclude Mode", maddr="239.1.1.2") + ) self.send(self.pg0, p_j) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.2", "0.0.0.0", 1)) - - p_j = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, ttl=1, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Mode Is Exclude", maddr="239.1.1.3")) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.2", "0.0.0.0", 1) + ) + + p_j = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + ttl=1, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr(rtype="Mode Is Exclude", maddr="239.1.1.3") + ) self.send(self.pg0, p_j) - self.assertTrue(wait_for_igmp_event(self, 1, self.pg0, - "239.1.1.3", "0.0.0.0", 1)) + self.assertTrue( + wait_for_igmp_event(self, 1, self.pg0, "239.1.1.3", "0.0.0.0", 1) + ) # # A 'allow sources' for {} should be ignored as it should # never be sent. # - p_j = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, ttl=1, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Allow New Sources", maddr="239.1.1.4")) + p_j = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + ttl=1, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr(rtype="Allow New Sources", maddr="239.1.1.4") + ) self.send(self.pg0, p_j) dump = self.vapi.igmp_dump(self.pg0.sw_if_index) - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.2", "0.0.0.0")) - self.assertTrue(find_igmp_state(dump, self.pg0, - "239.1.1.3", "0.0.0.0")) - self.assertFalse(find_igmp_state(dump, self.pg0, - "239.1.1.4", "0.0.0.0")) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.2", "0.0.0.0")) + self.assertTrue(find_igmp_state(dump, self.pg0, "239.1.1.3", "0.0.0.0")) + self.assertFalse(find_igmp_state(dump, self.pg0, "239.1.1.4", "0.0.0.0")) # # a TO_IN({}) and IS_IN({}) are treated like a (*,G) leave # self.vapi.cli("set logging class igmp level debug") - p_l = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, ttl=1, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Change To Include Mode", maddr="239.1.1.2")) + p_l = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + ttl=1, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr(rtype="Change To Include Mode", maddr="239.1.1.2") + ) self.send(self.pg0, p_l) - self.assertTrue(wait_for_igmp_event(self, 2, self.pg0, - "239.1.1.2", "0.0.0.0", 0)) - - p_l = (Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) / - IP(src=self.pg0.remote_ip4, dst="224.0.0.22", tos=0xc0, ttl=1, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype="Mode Is Include", maddr="239.1.1.3")) + self.assertTrue( + wait_for_igmp_event(self, 2, self.pg0, "239.1.1.2", "0.0.0.0", 0) + ) + + p_l = ( + Ether(dst=self.pg0.local_mac, src=self.pg0.remote_mac) + / IP( + src=self.pg0.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + ttl=1, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr(rtype="Mode Is Include", maddr="239.1.1.3") + ) self.send(self.pg0, p_l) - self.assertTrue(wait_for_igmp_event(self, 2, self.pg0, - "239.1.1.3", "0.0.0.0", 0)) + self.assertTrue( + wait_for_igmp_event(self, 2, self.pg0, "239.1.1.3", "0.0.0.0", 0) + ) self.assertFalse(self.vapi.igmp_dump(self.pg0.sw_if_index)) # # disable router config # - self.vapi.igmp_enable_disable(self.pg0.sw_if_index, - 0, - IGMP_MODE.ROUTER) + self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 0, IGMP_MODE.ROUTER) def _create_igmpv3_pck(self, itf, rtype, maddr, srcaddrs): - p = (Ether(dst=itf.local_mac, src=itf.remote_mac) / - IP(src=itf.remote_ip4, dst="224.0.0.22", tos=0xc0, ttl=1, - options=[IPOption(copy_flag=1, optclass="control", - option="router_alert")]) / - IGMPv3(type="Version 3 Membership Report") / - IGMPv3mr(numgrp=1) / - IGMPv3gr(rtype=rtype, - maddr=maddr, srcaddrs=srcaddrs)) + p = ( + Ether(dst=itf.local_mac, src=itf.remote_mac) + / IP( + src=itf.remote_ip4, + dst="224.0.0.22", + tos=0xC0, + ttl=1, + options=[ + IPOption(copy_flag=1, optclass="control", option="router_alert") + ], + ) + / IGMPv3(type="Version 3 Membership Report") + / IGMPv3mr(numgrp=1) + / IGMPv3gr(rtype=rtype, maddr=maddr, srcaddrs=srcaddrs) + ) return p def test_igmp_proxy_device(self): - """ IGMP proxy device """ + """IGMP proxy device""" self.pg2.admin_down() self.pg2.unconfig_ip4() self.pg2.set_table_ip4(0) self.pg2.config_ip4() self.pg2.admin_up() - self.vapi.cli('test igmp timers query 10 src 3 leave 1') + self.vapi.cli("test igmp timers query 10 src 3 leave 1") # enable IGMP self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 1, IGMP_MODE.HOST) - self.vapi.igmp_enable_disable(self.pg1.sw_if_index, 1, - IGMP_MODE.ROUTER) - self.vapi.igmp_enable_disable(self.pg2.sw_if_index, 1, - IGMP_MODE.ROUTER) + self.vapi.igmp_enable_disable(self.pg1.sw_if_index, 1, IGMP_MODE.ROUTER) + self.vapi.igmp_enable_disable(self.pg2.sw_if_index, 1, IGMP_MODE.ROUTER) # create IGMP proxy device self.vapi.igmp_proxy_device_add_del(0, self.pg0.sw_if_index, 1) - self.vapi.igmp_proxy_device_add_del_interface(0, - self.pg1.sw_if_index, 1) - self.vapi.igmp_proxy_device_add_del_interface(0, - self.pg2.sw_if_index, 1) + self.vapi.igmp_proxy_device_add_del_interface(0, self.pg1.sw_if_index, 1) + self.vapi.igmp_proxy_device_add_del_interface(0, self.pg2.sw_if_index, 1) # send join on pg1. join should be proxied by pg0 - p_j = self._create_igmpv3_pck(self.pg1, "Allow New Sources", - "239.1.1.1", ["10.1.1.1", "10.1.1.2"]) + p_j = self._create_igmpv3_pck( + self.pg1, "Allow New Sources", "239.1.1.1", ["10.1.1.1", "10.1.1.2"] + ) self.send(self.pg1, p_j) capture = self.pg0.get_capture(1, timeout=1) - self.verify_report(capture[0], [IgmpRecord(IgmpSG("239.1.1.1", - ["10.1.1.1", "10.1.1.2"]), "Allow New Sources")]) + self.verify_report( + capture[0], + [ + IgmpRecord( + IgmpSG("239.1.1.1", ["10.1.1.1", "10.1.1.2"]), "Allow New Sources" + ) + ], + ) self.assertTrue(find_mroute(self, "239.1.1.1", "0.0.0.0", 32)) # send join on pg2. join should be proxied by pg0. # the group should contain only 10.1.1.3 as # 10.1.1.1 was already reported - p_j = self._create_igmpv3_pck(self.pg2, "Allow New Sources", - "239.1.1.1", ["10.1.1.1", "10.1.1.3"]) + p_j = self._create_igmpv3_pck( + self.pg2, "Allow New Sources", "239.1.1.1", ["10.1.1.1", "10.1.1.3"] + ) self.send(self.pg2, p_j) capture = self.pg0.get_capture(1, timeout=1) - self.verify_report(capture[0], [IgmpRecord(IgmpSG("239.1.1.1", - ["10.1.1.3"]), "Allow New Sources")]) + self.verify_report( + capture[0], + [IgmpRecord(IgmpSG("239.1.1.1", ["10.1.1.3"]), "Allow New Sources")], + ) self.assertTrue(find_mroute(self, "239.1.1.1", "0.0.0.0", 32)) # send leave on pg2. leave for 10.1.1.3 should be proxyed # as pg2 was the only interface interested in 10.1.1.3 - p_l = self._create_igmpv3_pck(self.pg2, "Block Old Sources", - "239.1.1.1", ["10.1.1.3"]) + p_l = self._create_igmpv3_pck( + self.pg2, "Block Old Sources", "239.1.1.1", ["10.1.1.3"] + ) self.send(self.pg2, p_l) capture = self.pg0.get_capture(1, timeout=2) - self.verify_report(capture[0], [IgmpRecord(IgmpSG("239.1.1.1", - ["10.1.1.3"]), "Block Old Sources")]) + self.verify_report( + capture[0], + [IgmpRecord(IgmpSG("239.1.1.1", ["10.1.1.3"]), "Block Old Sources")], + ) self.assertTrue(find_mroute(self, "239.1.1.1", "0.0.0.0", 32)) # disable igmp on pg1 (also removes interface from proxy device) # proxy leave for 10.1.1.2. pg2 is still interested in 10.1.1.1 self.pg_enable_capture(self.pg_interfaces) - self.vapi.igmp_enable_disable(self.pg1.sw_if_index, 0, - IGMP_MODE.ROUTER) + self.vapi.igmp_enable_disable(self.pg1.sw_if_index, 0, IGMP_MODE.ROUTER) capture = self.pg0.get_capture(1, timeout=1) - self.verify_report(capture[0], [IgmpRecord(IgmpSG("239.1.1.1", - ["10.1.1.2"]), "Block Old Sources")]) + self.verify_report( + capture[0], + [IgmpRecord(IgmpSG("239.1.1.1", ["10.1.1.2"]), "Block Old Sources")], + ) self.assertTrue(find_mroute(self, "239.1.1.1", "0.0.0.0", 32)) # disable IGMP on pg0 and pg1. # disabling IGMP on pg0 (proxy device upstream interface) # removes this proxy device self.vapi.igmp_enable_disable(self.pg0.sw_if_index, 0, IGMP_MODE.HOST) - self.vapi.igmp_enable_disable(self.pg2.sw_if_index, 0, - IGMP_MODE.ROUTER) + self.vapi.igmp_enable_disable(self.pg2.sw_if_index, 0, IGMP_MODE.ROUTER) self.assertFalse(find_mroute(self, "239.1.1.1", "0.0.0.0", 32)) -if __name__ == '__main__': +if __name__ == "__main__": unittest.main(testRunner=VppTestRunner) |