libraries/cluster/testutils/ippool.py

   1 # Copyright 2019 Nokia
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 import itertools
  16
  17
  18 class IPPool(object):
  19     def __init__(self):
  20         self._internal_ip = self._ip_gen(base='192.168.1')
  21         self._external_ip = self._ip_gen(base='10.1.1')
  22
  23     def get_internal_ip(self):
  24         return next(self._internal_ip)
  25
  26     def get_external_ip(self):
  27         return next(self._external_ip)
  28
  29     @staticmethod
  30     def _ip_gen(base):
  31         for i in itertools.count(start=2):  # pragma: no branch
  32             yield '{base}.{i}'.format(base=base, i=i)