diff options
author | Paul Vinciguerra <pvinci@vinciconsulting.com> | 2020-12-04 17:50:26 -0500 |
---|---|---|
committer | Florin Coras <florin.coras@gmail.com> | 2020-12-06 18:54:07 +0000 |
commit | 1135f65586513c7cabb0dca063d1d2f97b07d7cd (patch) | |
tree | aca2a38091d2995598055f478871e16631defee4 /src | |
parent | 24841867d17f0b57abcce6353e23c9a668e70517 (diff) |
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 <pvinci@vinciconsulting.com>
Diffstat (limited to 'src')
-rw-r--r-- | src/plugins/lisp/test/test_lisp.py | 15 |
1 files changed, 14 insertions, 1 deletions
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', |