From bb8249ec4200471659d443bdec4befbac2ba01aa Mon Sep 17 00:00:00 2001
From: Bartek Grzybowski <b.grzybowski@partner.samsung.com>
Date: Mon, 18 May 2020 04:11:15 -0700
Subject: Create a boilerplate decorator for MassPnfSim class actions

Change-Id: Ia095a608d204e3da6dae101afa04f2e66a9bceeb
Issue-ID: INT-1577
Signed-off-by: Bartek Grzybowski <b.grzybowski@partner.samsung.com>
---
 test/mocks/mass-pnf-sim/MassPnfSim.py              | 45 ++++++++++++++--------
 .../mass-pnf-sim/pnf-sim-lightweight/simulator.sh  |  2 +-
 2 files changed, 29 insertions(+), 18 deletions(-)

(limited to 'test/mocks/mass-pnf-sim')

diff --git a/test/mocks/mass-pnf-sim/MassPnfSim.py b/test/mocks/mass-pnf-sim/MassPnfSim.py
index f38da71e3..966f3c887 100755
--- a/test/mocks/mass-pnf-sim/MassPnfSim.py
+++ b/test/mocks/mass-pnf-sim/MassPnfSim.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python3
 import logging
 import subprocess
-import time
 import argparse
 import ipaddress
 from sys import exit
@@ -73,6 +72,26 @@ 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():
 
     log_lvl = logging.INFO
@@ -176,30 +195,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')
     def start(self):
-        for i in range(self.args.count):
-            self.logger.info(f'Starting {self.sim_dirname_pattern}{i} instance:')
-            self._run_cmd('./simulator.sh start', f"{self.sim_dirname_pattern}{i}")
-            time.sleep(5)
+        pass
 
+    @do_action('Getting', './simulator.sh status')
     def status(self):
-        for i in range(self.args.count):
-            self.logger.info(f'Getting {self.sim_dirname_pattern}{i} status:')
-            self._run_cmd('./simulator.sh status', f"{self.sim_dirname_pattern}{i}")
+        pass
 
+    @do_action('Stopping', './simulator.sh stop')
     def stop(self):
-        for i in range(self.args.count):
-            self.logger.info(f'Stopping {self.sim_dirname_pattern}{i} instance:')
-            self._run_cmd(f'./simulator.sh stop {i}', f"{self.sim_dirname_pattern}{i}")
+        pass
 
+    @do_action('Triggering', './simulator.sh trigger-simulator')
     def trigger(self):
         self.logger.info("Triggering VES sending:")
-        for i in range(self.args.count):
-            self.logger.info(f'Triggering {self.sim_dirname_pattern}{i} instance:')
-            self._run_cmd(f'./simulator.sh trigger-simulator', f"{self.sim_dirname_pattern}{i}")
 
+    @do_action('Triggering', './simulator.sh trigger-simulator')
     def trigger_custom(self):
         self.logger.info("Triggering VES sending by a range of simulators:")
-        for i in range(self.args.triggerstart, self.args.triggerend+1):
-            self.logger.info(f'Triggering {self.sim_dirname_pattern}{i} instance:')
-            self._run_cmd(f'./simulator.sh trigger-simulator', f"{self.sim_dirname_pattern}{i}")
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 50ef3214d..8c5390ed5 100755
--- a/test/mocks/mass-pnf-sim/pnf-sim-lightweight/simulator.sh
+++ b/test/mocks/mass-pnf-sim/pnf-sim-lightweight/simulator.sh
@@ -30,7 +30,7 @@ function main(){
             if [[ -z ${2+x} ]]
             then
                echo "Error: action 'stop' requires the instance identifier"
-               exit
+               exit 1
             fi
             stop $2;;
         "run-simulator")
-- 
cgit 1.2.3-korg