CSIT-1148: VPPAPIGEN 2nd level import of .api ignored

[vpp.git] / test / test_lb.py

diff --git a/test/test_lb.py b/test/test_lb.py
index 79a9598..d2e7185 100644 (file)
--- a/test/test_lb.py
+++ b/test/test_lb.py
@@ -4,6 +4,7 @@ from scapy.layers.inet import IP, UDP
 from scapy.layers.inet6 import IPv6
 from scapy.layers.l2 import Ether, GRE
 from scapy.packet import Raw
+from scapy.data import IP_PROTOS
 
 from framework import VppTestCase
 from util import ppp
@@ -145,6 +146,11 @@ class TestLB(VppTestCase):
                     self.assertEqual(ip.dst, "10.0.0.%u" % asid)
                     self.assertEqual(ip.tos, 0x1c)
                     self.assertEqual(len(ip.options), 0)
+                    self.assert_ip_checksum_valid(p)
+                    if ip.proto == IP_PROTOS.tcp:
+                        self.assert_tcp_checksum_valid(p)
+                    elif ip.proto == IP_PROTOS.udp:
+                        self.assert_udp_checksum_valid(p)
                 elif (encap == 'nat4'):
                     ip = p[IP]
                     asid = int(ip.dst.split(".")[3])