diff options
-rwxr-xr-x | test/mocks/mass-pnf-sim/MassPnfSim.py | 11 | ||||
-rwxr-xr-x | test/mocks/mass-pnf-sim/pnf-sim-lightweight/simulator.sh | 1 | ||||
-rwxr-xr-x | test/mocks/mass-pnf-sim/setup.py | 3 |
3 files changed, 13 insertions, 2 deletions
diff --git a/test/mocks/mass-pnf-sim/MassPnfSim.py b/test/mocks/mass-pnf-sim/MassPnfSim.py index d15b69551..b69e0fcac 100755 --- a/test/mocks/mass-pnf-sim/MassPnfSim.py +++ b/test/mocks/mass-pnf-sim/MassPnfSim.py @@ -7,6 +7,7 @@ from sys import exit from os import chdir, getcwd, path from shutil import copytree, rmtree from json import dumps +from yaml import load, SafeLoader from glob import glob from requests import get from requests.exceptions import MissingSchema, InvalidSchema, InvalidURL, ConnectionError, ConnectTimeout @@ -104,6 +105,7 @@ class MassPnfSim: return action_decorator log_lvl = logging.INFO + sim_config = 'config/config.yml' def __init__(self, args): self.args = args @@ -150,6 +152,15 @@ class MassPnfSim: '''Helper method that returns bootstraped simulator instances count''' return len(glob(f"{self.sim_dirname_pattern}[0-9]*")) + def _get_sim_instance_data(self, instance_id): + '''Helper method that returns specific instance data''' + oldpwd = getcwd() + chdir(f"{self.sim_dirname_pattern}{instance_id}") + with open(self.sim_config) as cfg: + yml = load(cfg, Loader=SafeLoader) + chdir(oldpwd) + return yml['ippnfsim'] + def bootstrap(self): self.logger.info("Bootstrapping PNF instances") diff --git a/test/mocks/mass-pnf-sim/pnf-sim-lightweight/simulator.sh b/test/mocks/mass-pnf-sim/pnf-sim-lightweight/simulator.sh index 8c5390ed5..b957279af 100755 --- a/test/mocks/mass-pnf-sim/pnf-sim-lightweight/simulator.sh +++ b/test/mocks/mass-pnf-sim/pnf-sim-lightweight/simulator.sh @@ -187,7 +187,6 @@ EndOfMessage } function get_status(){ - get_pnfsim_ip if [[ $(running_containers) ]]; then print_status else diff --git a/test/mocks/mass-pnf-sim/setup.py b/test/mocks/mass-pnf-sim/setup.py index 662976b03..1d3dd7293 100755 --- a/test/mocks/mass-pnf-sim/setup.py +++ b/test/mocks/mass-pnf-sim/setup.py @@ -31,6 +31,7 @@ setuptools.setup( install_requires=[ 'argparse', 'ipaddress', - 'requests' + 'requests', + 'pyyaml' ] ) |