def setUpClass(cls):
template_dir_path = '%s/templates/general' % utils.get_resources_dir()
cls.def_templates_tests_path = '%s/templates/def_template_tests/' % \
utils.get_resources_dir()
cls.def_templates_dir_path = cls.def_templates_tests_path +\
'definition_templates'
cls.templates = file_utils.load_yaml_files(template_dir_path)
def_templates_list = file_utils.load_yaml_files(
cls.def_templates_dir_path)
cls.def_templates = utils.get_def_templates_dict_from_list(
def_templates_list)
cls.first_template = cls.templates[0]
cls._hide_useless_logging_messages()
def _load_templates_files(self, conf):
templates_dir = conf.evaluator.templates_dir
template_defs = file_utils.load_yaml_files(templates_dir)
for template_def in template_defs:
self.add_template(template_def)
def setUpClass(cls):
template_dir_path = '%s/templates/general' % utils.get_resources_dir()
cls.templates = file_utils.load_yaml_files(template_dir_path)
cls.first_template = cls.templates[0]
cls._hide_useless_logging_messages()
def _load_templates_files(self, conf):
templates_dir = conf.evaluator.templates_dir
template_defs = file_utils.load_yaml_files(templates_dir)
for template_def in template_defs:
self.add_template(template_def)
def setUpClass(cls):
template_dir_path = '%s/templates/general' % utils.get_resources_dir()
cls.template_yamls = file_utils.load_yaml_files(template_dir_path)
cls.first_template = cls.template_yamls[0]
cls._hide_useless_logging_messages()
def load_files(self, directory):
equivalence_defs = file_utils.load_yaml_files(directory)
for equivalence_def in equivalence_defs:
equivalence_data = EquivalenceData(equivalence_def)
for equivalence in equivalence_data.equivalences:
self._add_equivalence(equivalence)
return self.entity_equivalences
def setUp(self):
super(ScenarioRepositoryTest, self).setUp()
self.add_db()
templates_dir = utils.get_resources_dir() + '/templates/general'
self.add_templates(templates_dir)
templates_dir_path = templates_dir
self.template_defs = file_utils.load_yaml_files(templates_dir_path)
self.scenario_repository = ScenarioRepository()
def setUpClass(cls):
cls.conf = cfg.ConfigOpts()
cls.conf.register_opts(cls.OPTS, group='evaluator')
templates_dir_path = cls.conf.evaluator.templates_dir
cls.template_defs = file_utils.load_yaml_files(templates_dir_path)
cls.scenario_repository = ScenarioRepository(cls.conf)
def setUpClass(cls):
cls.conf = cfg.ConfigOpts()
cls.conf.register_opts(cls.OPTS, group='evaluator')
templates_dir_path = cls.conf.evaluator.templates_dir
cls.template_defs = file_utils.load_yaml_files(templates_dir_path)
cls.scenario_repository = ScenarioRepository(cls.conf)
def _load_def_template_files(self, conf):
if DEF_TEMPLATES_DIR_OPT in conf.evaluator:
def_templates_dir = conf.evaluator.def_templates_dir
def_templates = file_utils.load_yaml_files(def_templates_dir)
for def_template in def_templates:
self.add_def_template(def_template)
def test_static_datasources_loader(self):
# Setup
total_static_datasources = \
os.listdir(self.conf.static_physical.directory)
# Action
static_configs = file_utils.load_yaml_files(
self.conf.static_physical.directory)
# Test assertions
# -1 is because there are 2 files and a folder in static_datasource_dir
self.assertEqual(
len(total_static_datasources) - 1, len(static_configs))
def test_static_datasources_loader(self):
# Setup
total_static_datasources = \
os.listdir(self.conf.static_physical.directory)
# Action
static_configs = file_utils.load_yaml_files(
self.conf.static_physical.directory)
# Test assertions
# -1 is because there are 2 files and a folder in static_datasource_dir
self.assertEqual(len(total_static_datasources) - 1,
len(static_configs))
def setUpClass(cls):
super(TemplateSyntaxValidatorTest, cls).setUpClass()
cls.def_template_dir_path = utils.get_resources_dir() + \
'/templates/def_template_tests'
template_dir_path = '%s/templates/general' % utils.get_resources_dir()
cls.version_dir_path = '%s/templates/version/' \
% utils.get_resources_dir()
cls.template_yamls = file_utils.load_yaml_files(template_dir_path)
cls.bad_template = \
cls._load_template_file(template_dir_path + '/' + BAD_YAML_PATH)
cls.first_template = cls.template_yamls[0]
cls._hide_useless_logging_messages()
def test_basic_template_with_include(self):
# Test setup
template_path = self.DEF_TEMPLATE_TESTS_DIR +\
'/templates/%s' % self.BASIC_TEMPLATE_WITH_INCLUDE
template_definition = file_utils.load_yaml_file(template_path, True)
def_templates_path = self.DEF_TEMPLATE_TESTS_DIR + \
'/definition_templates'
def_demplates_list = file_utils.load_yaml_files(def_templates_path)
def_templates_dict = utils.get_def_templates_dict_from_list(
def_demplates_list)
template_data = \
TemplateLoader().load(template_definition, def_templates_dict)
entities = template_data.entities
relationships = template_data.relationships
scenarios = template_data.scenarios
definitions = template_definition[TFields.DEFINITIONS]
def_template = file_utils.load_yaml_file(def_templates_path +
'/basic_def_template.yaml')
def_template_entities = \
def_template[TFields.DEFINITIONS][TFields.ENTITIES]
def_template_relationships = \
def_template[TFields.DEFINITIONS][TFields.RELATIONSHIPS]
definitions[TFields.ENTITIES] += def_template_entities
definitions[TFields.RELATIONSHIPS] = def_template_relationships
# Assertions
for definition in definitions[TFields.ENTITIES]:
for key, value in definition['entity'].items():
new_key = TemplateLoader.PROPS_CONVERSION[key] if key in \
TemplateLoader.PROPS_CONVERSION else key
del definition['entity'][key]
definition['entity'][new_key] = value
self._validate_entities(entities, definitions[TFields.ENTITIES])
relate_def = def_template_relationships
self._validate_relationships(relationships, relate_def, entities)
self._validate_scenarios(scenarios, entities)
expected_entities = {
'alarm11':
Vertex(vertex_id='alarm11',
properties={
VProps.VITRAGE_CATEGORY: EntityCategory.ALARM,
VProps.VITRAGE_TYPE: NAGIOS_DATASOURCE,
VProps.NAME: 'host_problem'
}),
'resource11':
Vertex(vertex_id='resource11',
properties={
VProps.VITRAGE_CATEGORY: EntityCategory.RESOURCE,
VProps.VITRAGE_TYPE: NOVA_HOST_DATASOURCE
}),
'alarm':
Vertex(vertex_id='alarm',
properties={
VProps.VITRAGE_CATEGORY: EntityCategory.ALARM,
VProps.VITRAGE_TYPE: NAGIOS_DATASOURCE,
VProps.NAME: 'host_problem'
}),
'resource':
Vertex(vertex_id='resource',
properties={
VProps.VITRAGE_CATEGORY: EntityCategory.RESOURCE,
VProps.VITRAGE_TYPE: NOVA_HOST_DATASOURCE
})
}
expected_relationships = {
'alarm_on_host':
EdgeDescription(edge=Edge(
source_id='alarm',
target_id='resource',
label=EdgeLabel.ON,
properties={EdgeProperties.RELATIONSHIP_TYPE: EdgeLabel.ON}),
source=expected_entities['alarm'],
target=expected_entities['resource']),
}
scenario_entities = {
'alarm':
Vertex(vertex_id='alarm',
properties={
VProps.VITRAGE_CATEGORY: EntityCategory.ALARM,
VProps.VITRAGE_TYPE: NAGIOS_DATASOURCE,
VProps.NAME: 'host_problem'
}),
'resource':
Vertex(vertex_id='resource',
properties={
VProps.VITRAGE_CATEGORY: EntityCategory.RESOURCE,
VProps.VITRAGE_TYPE: NOVA_HOST_DATASOURCE
})
}
expected_scenario = Scenario(
id='basic_template_with_include-scenario0',
version=1,
condition=[[
ConditionVar(symbol_name='alarm_on_host', positive=True)
]],
actions=[
ActionSpecs(
id='basic_template_with_include-scenario0-action0',
type=ActionType.SET_STATE,
targets={'target': 'resource'},
properties={'state': OperationalResourceState.SUBOPTIMAL})
],
subgraphs=template_data.scenarios[0].subgraphs,
entities=scenario_entities,
relationships=expected_relationships)
self._validate_strict_equal(template_data, expected_entities,
expected_relationships, expected_scenario)
