From 1135f65586513c7cabb0dca063d1d2f97b07d7cd Mon Sep 17 00:00:00 2001 From: Paul Vinciguerra Date: Fri, 4 Dec 2020 17:50:26 -0500 Subject: tests: refactor ForeignAddressFactory Move ForeignAddressFactory back to test_lisp. It is the only test that uses the code and has embedded addresses specific to the test. Type: test Change-Id: If1a39f9b167e36dd7aff7481350ddb8fe17155d2 Signed-off-by: Paul Vinciguerra --- src/plugins/lisp/test/test_lisp.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'src/plugins/lisp/test/test_lisp.py') diff --git a/src/plugins/lisp/test/test_lisp.py b/src/plugins/lisp/test/test_lisp.py index 0bb42112140..0a6e7525159 100644 --- a/src/plugins/lisp/test/test_lisp.py +++ b/src/plugins/lisp/test/test_lisp.py @@ -11,7 +11,7 @@ from scapy.layers.inet6 import IPv6 from framework import VppTestCase, VppTestRunner from lisp import VppLocalMapping, VppLispAdjacency, VppLispLocator, \ VppLispLocatorSet, VppRemoteMapping, LispRemoteLocator -from util import ppp, ForeignAddressFactory +from util import ppp # From py_lispnetworking.lisp.py: # GNU General Public License v2.0 @@ -31,6 +31,19 @@ bind_layers(LISP_GPE_Header, IPv6, next_proto=2) bind_layers(LISP_GPE_Header, Ether, next_proto=3) +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) + + class Driver(metaclass=abc.ABCMeta): config_order = ['locator-sets', -- cgit 1.2.3-korg