1 ##############################################################################
2 # Copyright (c) 2015 Ericsson AB and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
10 '''A runner that every run changes a specified input value to the scenario.
11 The input value in the sequence is specified in a list in the input file.
15 import multiprocessing
20 from yardstick.benchmark.runners import base
22 LOG = logging.getLogger(__name__)
25 def _worker_process(queue, cls, method_name, scenario_cfg,
26 context_cfg, aborted):
30 runner_cfg = scenario_cfg['runner']
32 interval = runner_cfg.get("interval", 1)
33 arg_name = runner_cfg.get('scenario_option_name')
34 sequence_values = runner_cfg.get('sequence')
36 if 'options' not in scenario_cfg:
37 scenario_cfg['options'] = {}
39 options = scenario_cfg['options']
41 runner_cfg['runner_id'] = os.getpid()
43 LOG.info("worker START, sequence_values(%s, %s), class %s",
44 arg_name, sequence_values, cls)
46 benchmark = cls(scenario_cfg, context_cfg)
48 method = getattr(benchmark, method_name)
50 queue.put({'runner_id': runner_cfg['runner_id'],
51 'scenario_cfg': scenario_cfg,
52 'context_cfg': context_cfg})
55 if "sla" in scenario_cfg:
56 sla_action = scenario_cfg["sla"].get("action", "assert")
58 for value in sequence_values:
59 options[arg_name] = value
61 LOG.debug("runner=%(runner)s seq=%(sequence)s START" %
62 {"runner": runner_cfg["runner_id"], "sequence": sequence})
69 except AssertionError as assertion:
70 # SLA validation failed in scenario, determine what to do now
71 if sla_action == "assert":
73 elif sla_action == "monitor":
74 LOG.warning("SLA validation failed: %s" % assertion.args)
75 errors = assertion.args
76 except Exception as e:
77 errors = traceback.format_exc()
83 'timestamp': time.time(),
89 record = {'runner_id': runner_cfg['runner_id'],
90 'benchmark': benchmark_output}
94 LOG.debug("runner=%(runner)s seq=%(sequence)s END" %
95 {"runner": runner_cfg["runner_id"], "sequence": sequence})
99 if (errors and sla_action is None) or aborted.is_set():
103 LOG.info("worker END")
106 class SequenceRunner(base.Runner):
107 '''Run a scenario by changing an input value defined in a list
110 interval - time to wait between each scenario invocation
114 scenario_option_name - name of the option that is increased each invocation
118 sequence - list of values which are executed in their respective scenarios
124 __execution_type__ = 'Sequence'
126 def _run_benchmark(self, cls, method, scenario_cfg, context_cfg):
127 self.process = multiprocessing.Process(
128 target=_worker_process,
129 args=(self.result_queue, cls, method, scenario_cfg,
130 context_cfg, self.aborted))