test:vxlan over ipv6 tests

[vpp.git] / test / test_gtpu.py

diff --git a/test/test_gtpu.py b/test/test_gtpu.py
index b0c0a74..ed4e689 100644 (file)
--- a/test/test_gtpu.py
+++ b/test/test_gtpu.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 
 import socket
-from util import ip4n_range
+from util import ip4n_range, ip4_range
 import unittest
 from framework import VppTestCase, VppTestRunner
 from template_bd import BridgeDomain
@@ -30,6 +30,10 @@ class TestGtpu(BridgeDomain, VppTestCase):
                 GTP_U_Header(teid=vni, gtp_type=self.gtp_type, length=150) /
                 pkt)
 
+    def ip_range(self, start, end):
+        """ range of remote ip's """
+        return ip4_range(self.pg0.remote_ip4, start, end)
+
     def encap_mcast(self, pkt, src_ip, src_mac, vni):
         """
         Encapsulate the original payload frame by adding GTPU header with its