vpp/test/test_arping.py
Klement Sekera d9b0c6fbf7 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>
2022-05-10 18:52:08 +00:00

262 lines
8.1 KiB
Python

from scapy.layers.l2 import ARP
from scapy.layers.inet6 import ICMPv6ND_NS, ICMPv6ND_NA, IPv6
from framework import VppTestCase
""" TestArping is a subclass of VPPTestCase classes.
Basic test for sanity check of arping.
"""
class TestArping(VppTestCase):
"""Arping Test Case"""
@classmethod
def setUpClass(cls):
super(TestArping, cls).setUpClass()
try:
cls.create_pg_interfaces(range(2))
cls.interfaces = list(cls.pg_interfaces)
for i in cls.interfaces:
i.admin_up()
i.config_ip4()
i.config_ip6()
i.disable_ipv6_ra()
i.resolve_arp()
i.resolve_ndp()
except Exception:
super(TestArping, cls).tearDownClass()
raise
@classmethod
def tearDownClass(cls):
super(TestArping, cls).tearDownClass()
def tearDown(self):
super(TestArping, self).tearDown()
def show_commands_at_teardown(self):
self.logger.info(self.vapi.cli("show hardware"))
def verify_arping_request(self, p, src, dst):
arp = p[ARP]
self.assertEqual(arp.hwtype, 0x0001)
self.assertEqual(arp.ptype, 0x0800)
self.assertEqual(arp.hwlen, 6)
self.assertEqual(arp.op, 1)
self.assertEqual(arp.psrc, src)
self.assertEqual(arp.pdst, dst)
def verify_arping_ip6_ns(self, p, src, dst):
icmpv6 = p[ICMPv6ND_NS]
self.assertEqual(icmpv6.type, 135)
self.assertEqual(icmpv6.tgt, dst)
ipv6 = p[IPv6]
self.assertEqual(src, ipv6.src)
def verify_arping_ip6_na(self, p, src, dst):
icmpv6 = p[ICMPv6ND_NA]
self.assertEqual(icmpv6.type, 136)
self.assertEqual(icmpv6.tgt, dst)
ipv6 = p[IPv6]
self.assertEqual(src, ipv6.src)
def test_arping_ip4_arp_request_cli(self):
"""arping IP4 arp request CLI test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
remote_ip4 = self.pg1.remote_ip4
ping_cmd = "arping " + remote_ip4 + "pg1 repeat 5 interval 0.1"
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
ping_cmd = "arping " + remote_ip4 + "pg1"
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_request(p, self.pg1.local_ip4, self.pg1.remote_ip4)
finally:
self.vapi.cli("show error")
def test_arping_ip4_garp_cli(self):
"""arping ip4 gratuitous arp CLI test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
ping_cmd = (
"arping gratuitous" + self.pg1.local_ip4 + "pg1 repeat 5 interval 0.1"
)
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
ping_cmd = "arping gratuitous" + self.pg1.local_ip4 + "pg1"
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_request(p, self.pg1.local_ip4, self.pg1.local_ip4)
finally:
self.vapi.cli("show error")
def test_arping_ip4_arp_request_api(self):
"""arping ip4 arp request API test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
remote_ip4 = self.pg1.remote_ip4
ret = self.vapi.arping(
address=remote_ip4,
sw_if_index=self.pg1.sw_if_index,
is_garp=0,
repeat=5,
interval=0.1,
)
self.logger.info(ret)
ret = self.vapi.arping(
address=remote_ip4, sw_if_index=self.pg1.sw_if_index, is_garp=0
)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_request(p, self.pg1.local_ip4, self.pg1.remote_ip4)
finally:
self.vapi.cli("show error")
def test_arping_ip4_garp_api(self):
"""arping ip4 gratuitous arp API test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
ret = self.vapi.arping(
address=self.pg1.local_ip4,
sw_if_index=self.pg1.sw_if_index,
is_garp=1,
repeat=5,
interval=0.1,
)
self.logger.info(ret)
ret = self.vapi.arping(
address=self.pg1.local_ip4, sw_if_index=self.pg1.sw_if_index, is_garp=1
)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_request(p, self.pg1.local_ip4, self.pg1.local_ip4)
finally:
self.vapi.cli("show error")
def test_arping_ip6_ns_cli(self):
"""arping IP6 neighbor solicitation CLI test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
remote_ip6 = self.pg1.remote_ip6
ping_cmd = "arping " + remote_ip6 + "pg1 repeat 5 interval 0.1"
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
ping_cmd = "arping " + remote_ip6 + "pg1"
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_ip6_ns(p, self.pg1.local_ip6, self.pg1.remote_ip6)
finally:
self.vapi.cli("show error")
def test_arping_ip6_ns_api(self):
"""arping ip6 neighbor solicitation API test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
remote_ip6 = self.pg1.remote_ip6
ret = self.vapi.arping(
address=remote_ip6,
sw_if_index=self.pg1.sw_if_index,
is_garp=0,
repeat=5,
interval=0.1,
)
self.logger.info(ret)
ret = self.vapi.arping(
address=remote_ip6, sw_if_index=self.pg1.sw_if_index, is_garp=0
)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_ip6_ns(p, self.pg1.local_ip6, self.pg1.remote_ip6)
finally:
self.vapi.cli("show error")
def test_arping_ip6_na_cli(self):
"""arping ip6 neighbor advertisement CLI test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
ping_cmd = (
"arping gratuitous" + self.pg1.local_ip6 + "pg1 repeat 5 interval 0.1"
)
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
ping_cmd = "arping gratuitous" + self.pg1.local_ip6 + "pg1"
ret = self.vapi.cli(ping_cmd)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_ip6_na(p, self.pg1.local_ip6, self.pg1.local_ip6)
finally:
self.vapi.cli("show error")
def test_arping_ip6_na_api(self):
"""arping ip6 neighbor advertisement API test"""
try:
self.pg_enable_capture(self.pg_interfaces)
self.pg_start()
ret = self.vapi.arping(
address=self.pg1.local_ip6,
sw_if_index=self.pg1.sw_if_index,
is_garp=1,
repeat=5,
interval=0.1,
)
self.logger.info(ret)
ret = self.vapi.arping(
address=self.pg1.local_ip6, sw_if_index=self.pg1.sw_if_index, is_garp=1
)
self.logger.info(ret)
out = self.pg1.get_capture(6)
for p in out:
self.verify_arping_ip6_na(p, self.pg1.local_ip6, self.pg1.local_ip6)
finally:
self.vapi.cli("show error")
if __name__ == "__main__":
unittest.main(testRunner=VppTestRunner)