aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xtest/mocks/mass-pnf-sim/MassPnfSim.py55
1 files changed, 29 insertions, 26 deletions
diff --git a/test/mocks/mass-pnf-sim/MassPnfSim.py b/test/mocks/mass-pnf-sim/MassPnfSim.py
index 966f3c887..5c01704d3 100755
--- a/test/mocks/mass-pnf-sim/MassPnfSim.py
+++ b/test/mocks/mass-pnf-sim/MassPnfSim.py
@@ -72,27 +72,30 @@ def get_parser():
type=str, default='info')
return parser
-# MassPnfSim class actions decorator
-def do_action(action_string, cmd):
- def action_decorator(method):
- def action_wrap(self):
- cmd_local = cmd
- # Append instance # if action is 'stop'
- if method.__name__ == 'stop':
- cmd_local += " {}"
- # Alter looping range if action is 'tigger_custom'
- if method.__name__ == 'trigger_custom':
- iter_range = [self.args.triggerstart, self.args.triggerend+1]
- else:
- iter_range = [self.args.count]
- method(self)
- for i in range(*iter_range):
- self.logger.info(f'{action_string} {self.sim_dirname_pattern}{i} instance:')
- self._run_cmd(cmd_local.format(i), f"{self.sim_dirname_pattern}{i}")
- return action_wrap
- return action_decorator
-
-class MassPnfSim():
+class MassPnfSim:
+
+ # MassPnfSim class actions decorator
+ class _MassPnfSim_Decorators:
+
+ @staticmethod
+ def do_action(action_string, cmd):
+ def action_decorator(method):
+ def action_wrap(self):
+ cmd_local = cmd
+ # Append instance # if action is 'stop'
+ if method.__name__ == 'stop':
+ cmd_local += " {}"
+ # Alter looping range if action is 'tigger_custom'
+ if method.__name__ == 'trigger_custom':
+ iter_range = [self.args.triggerstart, self.args.triggerend+1]
+ else:
+ iter_range = [self.args.count]
+ method(self)
+ for i in range(*iter_range):
+ self.logger.info(f'{action_string} {self.sim_dirname_pattern}{i} instance:')
+ self._run_cmd(cmd_local.format(i), f"{self.sim_dirname_pattern}{i}")
+ return action_wrap
+ return action_decorator
log_lvl = logging.INFO
@@ -195,22 +198,22 @@ class MassPnfSim():
self.logger.info('Cleaning simulators workdirs')
self._run_cmd(f"rm -rf {self.sim_dirname_pattern}*")
- @do_action('Starting', './simulator.sh start')
+ @_MassPnfSim_Decorators.do_action('Starting', './simulator.sh start')
def start(self):
pass
- @do_action('Getting', './simulator.sh status')
+ @_MassPnfSim_Decorators.do_action('Getting', './simulator.sh status')
def status(self):
pass
- @do_action('Stopping', './simulator.sh stop')
+ @_MassPnfSim_Decorators.do_action('Stopping', './simulator.sh stop')
def stop(self):
pass
- @do_action('Triggering', './simulator.sh trigger-simulator')
+ @_MassPnfSim_Decorators.do_action('Triggering', './simulator.sh trigger-simulator')
def trigger(self):
self.logger.info("Triggering VES sending:")
- @do_action('Triggering', './simulator.sh trigger-simulator')
+ @_MassPnfSim_Decorators.do_action('Triggering', './simulator.sh trigger-simulator')
def trigger_custom(self):
self.logger.info("Triggering VES sending by a range of simulators:")