Added support to run cpu folding workload for specific time

Here to run the CPU fold workload for a specific duration added runtime parameter using .yaml

Signed-off-by: Samir Mulani <[email protected]>

workload/cpu_fold.py

@@ -36,14 +36,22 @@ def setUp(self):
         for pkg in ['util-linux', 'powerpc-utils', 'numactl']:
             if not sm.check_installed(pkg) and not sm.install(pkg):
                 self.cancel("%s is required to continue..." % pkg)
+        self.runtime = self.params.get('runtime', default='')
 
     def test_cpu_start(self):
         """
         Start the CPU Workload
         """
         relative_path = 'cpu_fold.py.data/cpu.sh'
         absolute_path = os.path.abspath(relative_path)
-        cpu_folding = "bash " + absolute_path + " &> /tmp/cpu_folding.log &"
+        cpu_folding = ""
+        if self.runtime == "":
+            cpu_folding = "bash " + absolute_path + \
+                " &> /tmp/cpu_folding.log &"
+        else:
+            cpu_folding = "timeout " + str(self.runtime) + "m" + " bash " + \
+                absolute_path + " &> /tmp/cpu_folding.log &"
+
         process.run(cpu_folding, ignore_status=True, sudo=True, shell=True)
         self.log.info("CPU Workload started--!!")
 

workload/cpu_fold.py.data/cpu_fold.yaml

@@ -0,0 +1 @@
+runtime: 1 # min