add get_workloads to utils

Author: msrba

scotty/core/components.py

@@ -72,8 +72,10 @@ def __init__(self):
         self.module = None
         self.parent_module_name = 'scotty.workload'
         self.state = WorkloadState.PREPARE
+        self.result = None
         self._setaccess('params')
         self._setaccess('resources')
+        self._setaccess('result')
 
     @property
     def module_path(self):
@@ -143,6 +145,8 @@ def __init__(self):
         self.module = None
         self.parent_module_name = 'scotty.systemcollector'
         self.state = CommonComponentState.PREPARE
+        self.result = None
+        self._setaccess('result')
 
     @property
     def module_path(self):

scotty/core/executor.py

@@ -110,6 +110,11 @@ def submit_systemcollectors(self, experiment):
             logger.info(msg.format(systemcollector.name))
             self.submit(experiment, systemcollector, 'collect')
 
+    def collect_results(self):
+        for future in as_completed(self._future_to_component):
+            systemcollector = self._future_to_component[future]
+            systemcollector.result = future.result()
+
 
 class ResultStoreSubmitExecutor(ComponentExecutor):
     def submit_resultstores(self, experiment):

scotty/tests/unit/__init__.py

@@ -0,0 +1,21 @@
+import unittest
+import mock
+
+from scotty.core.executor import ComponentExecutor
+from scotty import utils 
+
+class ExperimentHelperTest(unittest.TestCase):
+
+    @mock.patch('scotty.core.context.Context')
+    def test_get_workloads(self, context_mock):
+        experiment = mock.Mock()
+        experiment.components = {
+            "workload":{ 
+                "wl_1":mock.Mock(),
+                "wl_2":mock.Mock()
+            }
+        }
+        context_mock.v1._ContextV1__experiment = experiment
+        experiment_helper = utils.ExperimentHelper(context_mock)
+        workloads = experiment_helper.get_workloads()
+        self.assertEqual(workloads, experiment.components['workload'])

scotty/tests/unit/core/__init__.py

@@ -9,6 +9,7 @@
 
 class ExperimentHelper(object):
     def __init__(self, context):
+        # TODO validate context - is from scotty and not a fake from customer component
         self.context = context
         self.__experiment = context.v1._ContextV1__experiment
 
@@ -26,3 +27,9 @@ def get_resource(self, resource_name):
             raise ScottyException(
                 'Can not find resource ({})'.format(resource_name))
         return resource
+
+    # restrict function to a list of context types (component list)
+    # @restrict_to(component, component2, ...)
+    def get_workloads(self):
+        workloads = self.__experiment.components['workload']
+        return workloads