1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
# Copyright 2013: Mirantis Inc.
# All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
# yardstick comment: this is a modified copy of
# rally/rally/benchmark/scenarios/base.py
from stevedore import extension
import yardstick.common.utils as utils
def _iter_scenario_classes(scenario_type=None):
"""Generator over all 'Scenario' subclasses
This function will iterate over all 'Scenario' subclasses defined in this
project and will load any class introduced by any installed plugin project,
defined in 'entry_points' section, under 'yardstick.scenarios' subsection.
"""
extension.ExtensionManager(namespace='yardstick.scenarios',
invoke_on_load=False)
for scenario in utils.itersubclasses(Scenario):
if not scenario_type:
yield scenario
elif getattr(scenario, '__scenario_type__', None) == scenario_type:
yield scenario
class Scenario(object):
def setup(self):
""" default impl for scenario setup """
pass
def run(self, *args):
""" catcher for not implemented run methods in subclasses """
raise RuntimeError("run method not implemented")
def teardown(self):
""" default impl for scenario teardown """
pass
@staticmethod
def get_types():
"""return a list of known runner type (class) names"""
scenarios = []
for scenario in _iter_scenario_classes():
scenarios.append(scenario)
return scenarios
@staticmethod
def get_cls(scenario_type):
"""return class of specified type"""
for scenario in _iter_scenario_classes(scenario_type):
return scenario
raise RuntimeError("No such scenario type %s" % scenario_type)
@staticmethod
def get(scenario_type):
"""Returns instance of a scenario runner for execution type.
"""
scenario = Scenario.get_cls(scenario_type)
return scenario.__module__ + "." + scenario.__name__
@classmethod
def get_scenario_type(cls):
"""Return a string with the scenario type, if defined"""
return str(getattr(cls, '__scenario_type__', None))
@classmethod
def get_description(cls):
"""Return a single line string with the class description
This function will retrieve the class docstring and return the first
line, or 'None' if it's empty.
"""
return cls.__doc__.splitlines()[0] if cls.__doc__ else str(None)
def _push_to_outputs(self, keys, values):
return dict(zip(keys, values))
def _change_obj_to_dict(self, obj):
dic = {}
for k, v in vars(obj).items():
try:
vars(v)
except TypeError:
dic[k] = v
return dic
|
