Add basic 4o4 LISP unit test
Change-Id: I2d812153d7afe7980346382b525af89b3c47e796 Signed-off-by: Filip Tehlar <ftehlar@cisco.com>
This commit is contained in:
Filip Tehlar
committed by
Damjan Marion
Damjan Marion
parent
057bb8c3a4
commit
770e89e6b9
@ -12,7 +12,7 @@ UNITTEST_EXTRA_OPTS="-f"
|
||||
endif
|
||||
|
||||
PYTHON_VENV_PATH=$(VPP_PYTHON_PREFIX)/virtualenv
|
||||
PYTHON_DEPENDS=scapy==2.3.3 pexpect subprocess32
|
||||
PYTHON_DEPENDS=scapy==2.3.3 pexpect subprocess32 git+https://github.com/klement/py-lispnetworking@setup
|
||||
SCAPY_SOURCE=$(PYTHON_VENV_PATH)/lib/python2.7/site-packages/
|
||||
BUILD_COV_DIR = $(BR)/test-cov
|
||||
|
||||
|
||||
326
test/lisp.py
Normal file
326
test/lisp.py
Normal file
File diff suppressed because it is too large
Load Diff
171
test/test_lisp.py
Normal file
171
test/test_lisp.py
Normal file
@ -0,0 +1,171 @@
|
||||
#!/usr/bin/env python
|
||||
import unittest
|
||||
|
||||
from scapy.packet import Raw
|
||||
from scapy.layers.inet import IP, UDP, Ether
|
||||
from py_lispnetworking.lisp import LISP_GPE_Header
|
||||
|
||||
from util import ppp, ForeignAddressFactory
|
||||
from framework import VppTestCase, VppTestRunner
|
||||
from lisp import *
|
||||
|
||||
|
||||
class Driver(object):
|
||||
|
||||
config_order = ['locator-sets',
|
||||
'locators',
|
||||
'local-mappings',
|
||||
'remote-mappings',
|
||||
'adjacencies']
|
||||
|
||||
""" Basic class for data driven testing """
|
||||
def __init__(self, test, test_cases):
|
||||
self._test_cases = test_cases
|
||||
self._test = test
|
||||
|
||||
@property
|
||||
def test_cases(self):
|
||||
return self._test_cases
|
||||
|
||||
@property
|
||||
def test(self):
|
||||
return self._test
|
||||
|
||||
def create_packet(self, src_if, dst_if, deid, payload=''):
|
||||
"""
|
||||
Create IPv4 packet
|
||||
|
||||
param: src_if
|
||||
param: dst_if
|
||||
"""
|
||||
packet = (Ether(dst=src_if.local_mac, src=src_if.remote_mac) /
|
||||
IP(src=src_if.remote_ip4, dst=deid) /
|
||||
Raw(payload))
|
||||
return packet
|
||||
|
||||
@abstractmethod
|
||||
def run(self):
|
||||
""" testing procedure """
|
||||
pass
|
||||
|
||||
|
||||
class SimpleDriver(Driver):
|
||||
""" Implements simple test procedure """
|
||||
def __init__(self, test, test_cases):
|
||||
super(SimpleDriver, self).__init__(test, test_cases)
|
||||
|
||||
def verify_capture(self, src_loc, dst_loc, capture):
|
||||
"""
|
||||
Verify captured packet
|
||||
|
||||
:param src_loc: source locator address
|
||||
:param dst_loc: destination locator address
|
||||
:param capture: list of captured packets
|
||||
"""
|
||||
self.test.assertEqual(len(capture), 1, "Unexpected number of "
|
||||
"packets! Expected 1 but {} received"
|
||||
.format(len(capture)))
|
||||
packet = capture[0]
|
||||
try:
|
||||
ip_hdr = packet[IP]
|
||||
# assert the values match
|
||||
self.test.assertEqual(ip_hdr.src, src_loc, "IP source address")
|
||||
self.test.assertEqual(ip_hdr.dst, dst_loc,
|
||||
"IP destination address")
|
||||
gpe_hdr = packet[LISP_GPE_Header]
|
||||
self.test.assertEqual(gpe_hdr.next_proto, 1,
|
||||
"next_proto is not ipv4!")
|
||||
ih = gpe_hdr[IP]
|
||||
self.test.assertEqual(ih.src, self.test.pg0.remote_ip4,
|
||||
"unexpected source EID!")
|
||||
self.test.assertEqual(ih.dst, self.test.deid_ip4,
|
||||
"unexpected dest EID!")
|
||||
except:
|
||||
self.test.logger.error(ppp("Unexpected or invalid packet:",
|
||||
packet))
|
||||
raise
|
||||
|
||||
def configure_tc(self, tc):
|
||||
for config_item in self.config_order:
|
||||
for vpp_object in tc[config_item]:
|
||||
vpp_object.add_vpp_config()
|
||||
|
||||
def run(self, dest):
|
||||
""" Send traffic for each test case and verify that it
|
||||
is encapsulated """
|
||||
for tc in enumerate(self.test_cases):
|
||||
self.test.logger.info('Running {}'.format(tc[1]['name']))
|
||||
self.configure_tc(tc[1])
|
||||
|
||||
print self.test.vapi.cli("sh lisp loc")
|
||||
print self.test.vapi.cli("sh lisp eid")
|
||||
print self.test.vapi.cli("sh lisp adj vni 0")
|
||||
print self.test.vapi.cli("sh lisp gpe entry")
|
||||
|
||||
packet = self.create_packet(self.test.pg0, self.test.pg1, dest,
|
||||
'data')
|
||||
self.test.pg0.add_stream(packet)
|
||||
self.test.pg0.enable_capture()
|
||||
self.test.pg1.enable_capture()
|
||||
self.test.pg_start()
|
||||
capture = self.test.pg1.get_capture(1)
|
||||
self.verify_capture(self.test.pg1.local_ip4,
|
||||
self.test.pg1.remote_ip4, capture)
|
||||
self.test.pg0.assert_nothing_captured()
|
||||
|
||||
|
||||
class TestLisp(VppTestCase):
|
||||
""" Basic LISP test """
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
super(TestLisp, cls).setUpClass()
|
||||
cls.faf = ForeignAddressFactory()
|
||||
cls.create_pg_interfaces(range(2)) # create pg0 and pg1
|
||||
for i in cls.pg_interfaces:
|
||||
i.admin_up() # put the interface upsrc_if
|
||||
i.config_ip4() # configure IPv4 address on the interface
|
||||
i.resolve_arp() # resolve ARP, so that we know VPP MAC
|
||||
|
||||
def setUp(self):
|
||||
super(TestLisp, self).setUp()
|
||||
self.vapi.lisp_enable_disable(is_enabled=1)
|
||||
|
||||
def test_lisp_basic_encap(self):
|
||||
"""Test case for basic encapsulation"""
|
||||
|
||||
self.deid_ip4_net = self.faf.net
|
||||
self.deid_ip4 = self.faf.get_ip4()
|
||||
self.seid_ip4 = '{}/{}'.format(self.pg0.local_ip4, 32)
|
||||
self.rloc_ip4 = self.pg1.remote_ip4n
|
||||
|
||||
test_cases = [
|
||||
{
|
||||
'name': 'basic ip4 over ip4',
|
||||
'locator-sets': [VppLispLocatorSet(self, 'ls-4o4')],
|
||||
'locators': [
|
||||
VppLispLocator(self, self.pg1.sw_if_index, 'ls-4o4')
|
||||
],
|
||||
'local-mappings': [
|
||||
VppLocalMapping(self, self.seid_ip4, 'ls-4o4')
|
||||
],
|
||||
'remote-mappings': [
|
||||
VppRemoteMapping(self, self.deid_ip4_net,
|
||||
[{
|
||||
"is_ip4": 1,
|
||||
"priority": 1,
|
||||
"weight": 1,
|
||||
"addr": self.rloc_ip4
|
||||
}])
|
||||
],
|
||||
'adjacencies': [
|
||||
VppLispAdjacency(self, self.seid_ip4, self.deid_ip4_net)
|
||||
]
|
||||
}
|
||||
]
|
||||
self.test_driver = SimpleDriver(self, test_cases)
|
||||
self.test_driver.run(self.deid_ip4)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main(testRunner=VppTestRunner)
|
||||
13
test/util.py
13
test/util.py
@ -100,3 +100,16 @@ class Host(object):
|
||||
self._mac = mac
|
||||
self._ip4 = ip4
|
||||
self._ip6 = ip6
|
||||
|
||||
|
||||
class ForeignAddressFactory(object):
|
||||
count = 0
|
||||
prefix_len = 24
|
||||
net_template = '10.10.10.{}'
|
||||
net = net_template.format(0) + '/' + str(prefix_len)
|
||||
|
||||
def get_ip4(self):
|
||||
if self.count > 255:
|
||||
raise Exception("Network host address exhaustion")
|
||||
self.count += 1
|
||||
return self.net_template.format(self.count)
|
||||
|
||||
@ -1,4 +1,5 @@
|
||||
import os
|
||||
import socket
|
||||
import fnmatch
|
||||
import time
|
||||
from hook import Hook
|
||||
@ -1301,3 +1302,137 @@ class VppPapiProvider(object):
|
||||
|
||||
def ip_mfib_dump(self):
|
||||
return self.api(self.papi.ip_mfib_dump, {})
|
||||
|
||||
def lisp_enable_disable(self, is_enabled):
|
||||
return self.api(
|
||||
self.papi.lisp_enable_disable,
|
||||
{
|
||||
'is_en': is_enabled,
|
||||
})
|
||||
|
||||
def lisp_locator_set(self,
|
||||
ls_name,
|
||||
is_add=1):
|
||||
return self.api(
|
||||
self.papi.lisp_add_del_locator_set,
|
||||
{
|
||||
'is_add': is_add,
|
||||
'locator_set_name': ls_name
|
||||
})
|
||||
|
||||
def lisp_locator_set_dump(self):
|
||||
return self.api(self.papi.lisp_locator_set_dump, {})
|
||||
|
||||
def lisp_locator(self,
|
||||
ls_name,
|
||||
sw_if_index,
|
||||
priority=1,
|
||||
weight=1,
|
||||
is_add=1):
|
||||
return self.api(
|
||||
self.papi.lisp_add_del_locator,
|
||||
{
|
||||
'is_add': is_add,
|
||||
'locator_set_name': ls_name,
|
||||
'sw_if_index': sw_if_index,
|
||||
'priority': priority,
|
||||
'weight': weight
|
||||
})
|
||||
|
||||
def lisp_locator_dump(self, is_index_set, ls_name=None, ls_index=0):
|
||||
return self.api(
|
||||
self.papi.lisp_locator_dump,
|
||||
{
|
||||
'is_index_set': is_index_set,
|
||||
'ls_name': ls_name,
|
||||
'ls_index': ls_index,
|
||||
})
|
||||
|
||||
def lisp_local_mapping(self,
|
||||
ls_name,
|
||||
eid_type,
|
||||
eid,
|
||||
prefix_len,
|
||||
vni=0,
|
||||
key_id=0,
|
||||
key="",
|
||||
is_add=1):
|
||||
return self.api(
|
||||
self.papi.lisp_add_del_local_eid,
|
||||
{
|
||||
'locator_set_name': ls_name,
|
||||
'is_add': is_add,
|
||||
'eid_type': eid_type,
|
||||
'eid': eid,
|
||||
'prefix_len': prefix_len,
|
||||
'vni': vni,
|
||||
'key_id': key_id,
|
||||
'key': key
|
||||
})
|
||||
|
||||
def lisp_eid_table_dump(self,
|
||||
eid_set=0,
|
||||
prefix_length=0,
|
||||
vni=0,
|
||||
eid_type=0,
|
||||
eid=None,
|
||||
filter_opt=0):
|
||||
return self.api(
|
||||
self.papi.lisp_eid_table_dump,
|
||||
{
|
||||
'eid_set': eid_set,
|
||||
'prefix_length': prefix_length,
|
||||
'vni': vni,
|
||||
'eid_type': eid_type,
|
||||
'eid': eid,
|
||||
'filter': filter_opt,
|
||||
})
|
||||
|
||||
def lisp_remote_mapping(self,
|
||||
eid_type,
|
||||
eid,
|
||||
eid_prefix_len=0,
|
||||
vni=0,
|
||||
rlocs=None,
|
||||
rlocs_num=0,
|
||||
is_src_dst=0,
|
||||
is_add=1):
|
||||
return self.api(
|
||||
self.papi.lisp_add_del_remote_mapping,
|
||||
{
|
||||
'is_add': is_add,
|
||||
'eid_type': eid_type,
|
||||
'eid': eid,
|
||||
'eid_len': eid_prefix_len,
|
||||
'rloc_num': rlocs_num,
|
||||
'rlocs': rlocs,
|
||||
'vni': vni,
|
||||
'is_src_dst': is_src_dst,
|
||||
})
|
||||
|
||||
def lisp_adjacency(self,
|
||||
leid,
|
||||
reid,
|
||||
leid_len,
|
||||
reid_len,
|
||||
eid_type,
|
||||
is_add=1,
|
||||
vni=0):
|
||||
return self.api(
|
||||
self.papi.lisp_add_del_adjacency,
|
||||
{
|
||||
'is_add': is_add,
|
||||
'vni': vni,
|
||||
'eid_type': eid_type,
|
||||
'leid': leid,
|
||||
'reid': reid,
|
||||
'leid_len': leid_len,
|
||||
'reid_len': reid_len,
|
||||
})
|
||||
|
||||
def lisp_adjacencies_get(self, vni=0):
|
||||
return self.api(
|
||||
self.papi.lisp_adjacencies_get,
|
||||
{
|
||||
'vni': vni
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user