openstack/vitrage
Code Issues Proposed changes

Merge "Zabbix to Vitrage notifications"

Browse Source
This commit is contained in:
Jenkins 2016-07-27 19:12:56 +00:00 committed by Gerrit Code Review
commit 6a3f023ff1
12 changed files with 585 additions and 214 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
doc/source/zabbix_vitrage.rst Normal file
View File

@ -0,0 +1,72 @@
Zabbix-Vitrage Gateway
======================
Consolidate Zabbix alerts from across multiple sites into a single "at-a-glance" console by using a custom Zabbix [alertscript](https://www.zabbix.com/documentation/3.0/manual/config/notifications/media/script).
Installation
------------
Copy the `zabbix_vitrage.py` script into the Zabbix servers' `AlertScriptsPath` directory which is by default `/etc/zabbix/alertscripts` and make it executable:
| $ wget https://github.com/openstack/vitrage/tree/master/vitrage/datasources/zabbix/auxiliary/zabbix_vitrage.py
| $ cp zabbix_vitrage.py /etc/zabbix/alertscripts/
| $ chmod 755 /etc/zabbix/alertscripts/zabbix_vitrage.py
Configuration
-------------
To forward zabbix events to Vitrage a new media script needs to be created and associated with a user. Follow the steps below as a Zabbix Admin user:
1. Create a new media type [Admininstration > Media Types > Create Media Type]
| Name: Vitrage Notifications
| Type: Script
| Script name: zabbix_vitrage.py
| Script parameters:
| 1st: {ALERT.SENDTO}
| 2nd: {ALERT.SUBJECT}
| 3rd: {ALERT.MESSAGE}
2. Modify the Media for the Admin user [Administration > Users]
| Type: vitrage Notifications
| Send to: rabbit://rabbit_user:rabbit_pass@127.0.0.1:5672/ <--- Vitrage message bus url
| When active: 1-7,00:00-24:00
| Use if severity: (all)
| Status: Enabled
3. Configure Action [Configuration > Actions > Create Action > Action]
| Name: Forward to Vitrage
| Default Subject: {TRIGGER.STATUS}
| Default Message:
| host={HOST.NAME1}
| hostid={HOST.ID1}
| hostip={HOST.IP1}
| id={TRIGGER.ID}
| description={TRIGGER.NAME}
| rawtext={TRIGGER.NAME.ORIG}
| expression={TRIGGER.EXPRESSION}
| value={TRIGGER.VALUE}
| priority={TRIGGER.NSEVERITY}
| lastchange={EVENT.DATE} {EVENT.TIME}
For a full list of trigger macros see https://www.zabbix.com/documentation/3.0/manual/appendix/macros/supported_by_location
To send events add under the Conditions tab:
(A) Maintenance status not in `maintenance`
Finally, add an operation:
Send to Users: Admin
Send only to: Vitrage Notifications
DONE
----

5
vitrage/common/datetime_utils.py
View File

@ -23,3 +23,8 @@ def utcnow(with_timezone=True):
def change_time_str_format(timestamp_str, old_format, new_format): def change_time_str_format(timestamp_str, old_format, new_format):
utc = datetime.strptime(timestamp_str, old_format) utc = datetime.strptime(timestamp_str, old_format)
return utc.strftime(new_format) return utc.strftime(new_format)
def format_unix_timestamp(timestamp, date_format):
return datetime.fromtimestamp(float(timestamp)) \
.strftime(date_format)

3
vitrage/datasources/zabbix/__init__.py
View File

@ -39,4 +39,7 @@ OPTS = [
help='Zabbix url'), help='Zabbix url'),
cfg.StrOpt('config_file', default='/etc/vitrage/zabbix_conf.yaml', cfg.StrOpt('config_file', default='/etc/vitrage/zabbix_conf.yaml',
help='Zabbix configuration file'), help='Zabbix configuration file'),
cfg.StrOpt('notification_topic',
default='vitrage_notifications',
help='Zabbix configured notifications topic for Vitrage'),
] ]

76
vitrage/datasources/zabbix/auxiliary/readme Normal file
View File

@ -0,0 +1,76 @@
Zabbix-Vitrage Gateway
======================
Consolidate Zabbix alerts from across multiple sites into a single "at-a-glance"
console by using a custom Zabbix [alertscript].
(https://www.zabbix.com/documentation/3.0/manual/config/notifications/media/script)
Installation
------------
Copy the 'zabbix_vitrage.py' script into the Zabbix servers' 'AlertScriptsPath'
directory which is by default '/etc/zabbix/alertscripts' and make it executable:
$ wget https://github.com/openstack/vitrage/tree/master/vitrage/datasources/zabbix/auxiliary/zabbix_vitrage.py
$ cp zabbix_vitrage.py /etc/zabbix/alertscripts/
$ chmod 755 /etc/zabbix/alertscripts/zabbix_vitrage.py
Configuration
-------------
To forward zabbix events to Vitrage a new media script needs to be created and
associated with a user. Follow the steps below as a Zabbix Admin user:
1. Create a new media type [Admininstration > Media Types > Create Media Type]
Name: Vitrage Notifications
Type: Script
Script name: zabbix_vitrage.py
Script parameters:
1st: {ALERT.SENDTO}
2nd: {ALERT.SUBJECT}
3rd: {ALERT.MESSAGE}
2. Modify the Media for the Admin user [Administration > Users]
Type: Vitrage Notifications
Send to: rabbit://rabbit_user:rabbit_pass@127.0.0.1:5672/ <--- Vitrage message bus url
When active: 1-7,00:00-24:00
Use if severity: (all)
Status: Enabled
3. Configure Action [Configuration > Actions > Create Action > Action]
Name: Forward to Vitrage
Default Subject: {TRIGGER.STATUS}
Default Message:
host={HOST.NAME1}
hostid={HOST.ID1}
hostip={HOST.IP1}
id={TRIGGER.ID}
description={TRIGGER.NAME}
rawtext={TRIGGER.NAME.ORIG}
expression={TRIGGER.EXPRESSION}
value={TRIGGER.VALUE}
priority={TRIGGER.NSEVERITY}
lastchange={EVENT.DATE} {EVENT.TIME}
For a full list of trigger macros see
https://www.zabbix.com/documentation/3.0/manual/appendix/macros/supported_by_location
To send events add under the Conditions tab:
(A) Maintenance status not in `maintenance`
Finally, add an operation:
Send to Users: Admin
Send only to: Vitrage Notifications
DONE

103
vitrage/datasources/zabbix/auxiliary/zabbix_vitrage.py Normal file
View File

@ -0,0 +1,103 @@
#!/usr/bin/env python
# coding: utf-8
# Copyright 2016 - Nokia
#
# 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.
import argparse
from datetime import datetime
import logging
from oslo_config import cfg
import oslo_messaging as messaging
import six
import socket
import sys
import uuid
'''
Expected input:
Send To:
Vitrage Message Bus address e.g.
rabbit://userrabbit:passrabbit@127.0.0.1:5672/
Subject: {TRIGGER.STATUS}
Message:
host={HOST.NAME1}
hostid={HOST.ID1}
hostip={HOST.IP1}
id={TRIGGER.ID}
description={TRIGGER.NAME}
rawtext={TRIGGER.NAME.ORIG}
expression={TRIGGER.EXPRESSION}
value={TRIGGER.VALUE}
priority={TRIGGER.NSEVERITY}
lastchange={EVENT.DATE} {EVENT.TIME}
'''
LOG_FILE = '/var/log/zabbix/zabbix_vitrage.log'
LOG_FORMAT = '%(asctime)s.%(msecs).03d %(name)s[%(process)d] %(threadName)s %' \
'(levelname)s - %(message)s'
LOG_DATE_FMT = '%Y.%m.%d %H:%M:%S'
ZABBIX_EVENT_TYPE = 'zabbix.alarm'
debug = False
def main():
parser = argparse.ArgumentParser()
parser.add_argument('sendto', help='Vitrage message bus path')
parser.add_argument('topic', help='zabbix topic')
parser.add_argument('body', help='zabbix body')
args = parser.parse_args()
logging.debug('[vitrage] sendto=%s, topic=%s, body=%s',
args.sendto, args.topic, args.body)
transport_url = args.sendto
transport = messaging.get_transport(cfg.CONF, transport_url)
driver = 'messaging'
notifier = messaging.Notifier(transport,
driver=driver,
publisher_id='zabbix',
topic='vitrage_notifications')
alarm_status = args.topic.lower()
event_type = '%s.%s' % (ZABBIX_EVENT_TYPE, alarm_status)
publisher = 'zabbix_%s' % socket.gethostname()
payload = {key.lower().strip(): prop.strip() for key, prop in
(line.split('=') for line in args.body.splitlines())}
logging.debug('[vitrage] publisher: %s, event: %s, payload %s',
publisher, event_type, payload)
notifier.info(ctxt={'message_id': six.text_type(uuid.uuid4()),
'publisher_id': publisher,
'timestamp': datetime.utcnow()},
event_type=event_type,
payload=payload)
if __name__ == '__main__':
if debug:
logging.basicConfig(stream=sys.stderr, format=LOG_FORMAT,
datefmt=LOG_DATE_FMT, level=logging.DEBUG)
else:
logging.basicConfig(filename=LOG_FILE, format=LOG_FORMAT,
datefmt=LOG_DATE_FMT, level=logging.DEBUG)
main()

89
vitrage/datasources/zabbix/driver.py
View File

@ -17,11 +17,16 @@ from collections import namedtuple
from oslo_log import log from oslo_log import log
from pyzabbix import ZabbixAPI from pyzabbix import ZabbixAPI
from vitrage.common.constants import DatasourceProperties as DSProps
from vitrage.common.constants import SyncMode
from vitrage.common import file_utils from vitrage.common import file_utils
from vitrage.datasources.alarm_driver_base import AlarmDriverBase from vitrage.datasources.alarm_driver_base import AlarmDriverBase
from vitrage.datasources.zabbix.properties import ZabbixProperties \ from vitrage.datasources.zabbix.properties import ZabbixProperties \
as ZabbixProps as ZabbixProps
from vitrage.datasources.zabbix.properties import ZabbixTriggerStatus from vitrage.datasources.zabbix.properties import ZabbixTriggerStatus \
as TriggerStatus
from vitrage.datasources.zabbix.properties import ZabbixTriggerValue \
as TriggerValue
from vitrage.datasources.zabbix import ZABBIX_DATASOURCE from vitrage.datasources.zabbix import ZABBIX_DATASOURCE
LOG = log.getLogger(__name__) LOG = log.getLogger(__name__)
@ -29,12 +34,14 @@ LOG = log.getLogger(__name__)
class ZabbixDriver(AlarmDriverBase): class ZabbixDriver(AlarmDriverBase):
ServiceKey = namedtuple('ServiceKey', ['host_name', 'service']) ServiceKey = namedtuple('ServiceKey', ['host_name', 'service'])
conf_map = None
def __init__(self, conf): def __init__(self, conf):
super(ZabbixDriver, self).__init__() super(ZabbixDriver, self).__init__()
self.conf = conf self.conf = conf
self.configuration_mapping = self._configuration_mapping(conf) if ZabbixDriver.conf_map is None:
self.status_mapping = self._status_mapping() ZabbixDriver.conf_map =\
ZabbixDriver._configuration_mapping(conf)
self.client = None self.client = None
def _sync_type(self): def _sync_type(self):
@ -66,56 +73,53 @@ class ZabbixDriver(AlarmDriverBase):
self.client.login(zabbix_user, zabbix_password) self.client.login(zabbix_user, zabbix_password)
alarms = [] alarms = []
hosts = self.client.host.get() valid_hosts = (host for host in
for host in hosts: self.client.host.get(output=[ZabbixProps.HOST])
if host[ZabbixProps.HOST] in self.configuration_mapping: if host[ZabbixProps.HOST] in ZabbixDriver.conf_map)
for host in valid_hosts:
self._get_triggers_per_host(host, alarms) self._get_triggers_per_host(host, alarms)
return alarms return alarms
def _get_triggers_per_host(self, host, alarms):
host_id = host[ZabbixProps.HOST_ID]
triggers = self.client.trigger.get(hostids=host_id)
for trigger in triggers:
trigger[ZabbixProps.RESOURCE_NAME] = host[ZabbixProps.HOST]
alarms.append(trigger)
def _enrich_alarms(self, alarms): def _enrich_alarms(self, alarms):
for alarm in alarms: for alarm in alarms:
# based on zabbix configuration file, convert zabbix host name # based on zabbix configuration file, convert zabbix host name
# to vitrage resource type and name # to vitrage resource type and name
zabbix_host = alarm[ZabbixProps.RESOURCE_NAME] zabbix_host = alarm[ZabbixProps.RESOURCE_NAME]
vitrage_resource = self.configuration_mapping[zabbix_host] vitrage_resource = ZabbixDriver.conf_map[zabbix_host]
alarm[ZabbixProps.STATUS] = \ alarm[ZabbixProps.VALUE] = self._get_value(alarm)
self._get_status(alarm)
alarm[ZabbixProps.RESOURCE_TYPE] = \ alarm[ZabbixProps.RESOURCE_TYPE] = \
vitrage_resource[ZabbixProps.RESOURCE_TYPE] vitrage_resource[ZabbixProps.RESOURCE_TYPE]
alarm[ZabbixProps.RESOURCE_NAME] = \ alarm[ZabbixProps.RESOURCE_NAME] = \
vitrage_resource[ZabbixProps.RESOURCE_NAME] vitrage_resource[ZabbixProps.RESOURCE_NAME]
def _is_erroneous(self, alarm): def _is_erroneous(self, alarm):
return alarm and alarm[ZabbixProps.STATUS] != ZabbixTriggerStatus.OK return alarm and \
alarm[ZabbixProps.VALUE] == TriggerValue.PROBLEM
def _status_changed(self, alarm1, alarm2): def _status_changed(self, alarm1, alarm2):
return alarm1 and alarm2 and \ return (alarm1 and alarm2) and \
not alarm1[ZabbixProps.STATUS] == alarm2[ZabbixProps.STATUS] (alarm1[ZabbixProps.VALUE] != alarm2[ZabbixProps.VALUE] or
alarm1[ZabbixProps.PRIORITY] != alarm2[ZabbixProps.PRIORITY])
def _is_valid(self, alarm): def _is_valid(self, alarm):
return alarm[ZabbixProps.RESOURCE_TYPE] is not None and \ return alarm[ZabbixProps.RESOURCE_TYPE] is not None and \
alarm[ZabbixProps.RESOURCE_NAME] is not None alarm[ZabbixProps.RESOURCE_NAME] is not None
def _get_status(self, alarm):
if alarm[ZabbixProps.IS_ALARM_DISABLED] == '1' or \
alarm[ZabbixProps.IS_ALARM_ON] == '0':
return ZabbixTriggerStatus.OK
return self.status_mapping[alarm[ZabbixProps.SEVERITY]]
@staticmethod @staticmethod
def _status_mapping(): def _get_value(alarm):
return { if alarm[ZabbixProps.STATUS] == TriggerStatus.DISABLED:
'-1': ZabbixTriggerStatus.OK, return TriggerValue.OK
'0': ZabbixTriggerStatus.NOT_CLASSIFIED, return alarm[ZabbixProps.VALUE]
'1': ZabbixTriggerStatus.INFORMATION,
'2': ZabbixTriggerStatus.WARNING,
'3': ZabbixTriggerStatus.AVERAGE,
'4': ZabbixTriggerStatus.HIGH,
'5': ZabbixTriggerStatus.DISASTER
}
@staticmethod @staticmethod
def _configuration_mapping(conf): def _configuration_mapping(conf):
@ -136,9 +140,24 @@ class ZabbixDriver(AlarmDriverBase):
LOG.exception('failed in init %s ', e) LOG.exception('failed in init %s ', e)
return {} return {}
def _get_triggers_per_host(self, host, alarms): @staticmethod
host_ids = host[ZabbixProps.HOST_ID] def enrich_event(event, event_type):
triggers = self.client.trigger.get(hostids=host_ids) event[DSProps.EVENT_TYPE] = event_type
for trigger in triggers: if ZabbixDriver.conf_map:
trigger[ZabbixProps.RESOURCE_NAME] = host[ZabbixProps.HOST] zabbix_host = event[ZabbixProps.HOST]
alarms.append(trigger) vitrage_resource = ZabbixDriver.conf_map[zabbix_host]
event[ZabbixProps.RESOURCE_NAME] = \
vitrage_resource[ZabbixProps.RESOURCE_NAME]
event[ZabbixProps.RESOURCE_TYPE] = \
vitrage_resource[ZabbixProps.RESOURCE_TYPE]
return ZabbixDriver.make_pickleable([event], ZABBIX_DATASOURCE,
SyncMode.UPDATE)[0]
@staticmethod
def get_event_types(conf):
return ['zabbix.alarm.ok',
'zabbix.alarm.problem']
@staticmethod
def get_topic(conf):
return conf[ZABBIX_DATASOURCE].notification_topic

32
vitrage/datasources/zabbix/properties.py
View File

@ -18,19 +18,43 @@ class ZabbixProperties(object):
RESOURCE_NAME = 'resource_name' RESOURCE_NAME = 'resource_name'
DESCRIPTION = 'description' DESCRIPTION = 'description'
STATUS = 'status' STATUS = 'status'
VALUE = 'value'
HOST = 'host' HOST = 'host'
HOST_ID = 'hostid' HOST_ID = 'hostid'
IS_ALARM_DISABLED = 'status' PRIORITY = 'priority'
IS_ALARM_ON = 'value'
SEVERITY = 'priority'
LAST_CHANGE = 'lastchange' LAST_CHANGE = 'lastchange'
TIMESTAMP = 'timestamp'
ZABBIX_TIMESTAMP_FORMAT = '%Y.%m.%d %H:%M:%S'
class ZabbixTriggerValue(object):
OK = '0'
PROBLEM = '1'
class ZabbixTriggerStatus(object): class ZabbixTriggerStatus(object):
OK = 'OK' ENABLED = '0'
DISABLED = '1'
class ZabbixTriggerSeverity(object):
INFORMATION = 'INFORMATION' INFORMATION = 'INFORMATION'
WARNING = 'WARNING' WARNING = 'WARNING'
AVERAGE = 'AVERAGE' AVERAGE = 'AVERAGE'
HIGH = 'HIGH' HIGH = 'HIGH'
DISASTER = 'DISASTER' DISASTER = 'DISASTER'
NOT_CLASSIFIED = 'NOT CLASSIFIED' NOT_CLASSIFIED = 'NOT CLASSIFIED'
_SEVERITY_MAPPING = {
'0': NOT_CLASSIFIED,
'1': INFORMATION,
'2': WARNING,
'3': AVERAGE,
'4': HIGH,
'5': DISASTER
}
@staticmethod
def str(num):
return ZabbixTriggerSeverity._SEVERITY_MAPPING[num]

62
vitrage/datasources/zabbix/transformer.py
View File

@ -18,14 +18,22 @@ from vitrage.common.constants import DatasourceProperties as DSProps
from vitrage.common.constants import EdgeLabel from vitrage.common.constants import EdgeLabel
from vitrage.common.constants import EntityCategory from vitrage.common.constants import EntityCategory
from vitrage.common.constants import VertexProperties as VProps from vitrage.common.constants import VertexProperties as VProps
from vitrage.common.datetime_utils import change_time_str_format
from vitrage.common.datetime_utils import format_unix_timestamp
from vitrage.datasources.alarm_properties import AlarmProperties as AlarmProps from vitrage.datasources.alarm_properties import AlarmProperties as AlarmProps
from vitrage.datasources.alarm_transformer_base import AlarmTransformerBase from vitrage.datasources.alarm_transformer_base import AlarmTransformerBase
from vitrage.datasources.nova.host import NOVA_HOST_DATASOURCE from vitrage.datasources.nova.host import NOVA_HOST_DATASOURCE
from vitrage.datasources.static_physical import SWITCH from vitrage.datasources.static_physical import SWITCH
from vitrage.datasources import transformer_base as tbase from vitrage.datasources import transformer_base as tbase
from vitrage.datasources.transformer_base import Neighbor from vitrage.datasources.transformer_base import Neighbor
from vitrage.datasources.zabbix.properties import ZabbixProperties from vitrage.datasources.zabbix.properties import ZabbixProperties \
from vitrage.datasources.zabbix.properties import ZabbixTriggerStatus as ZabbixProps
from vitrage.datasources.zabbix.properties import ZabbixTriggerSeverity \
as TriggerSeverity
from vitrage.datasources.zabbix.properties import ZabbixTriggerValue\
as TriggerValue
import vitrage.graph.utils as graph_utils import vitrage.graph.utils as graph_utils
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -43,25 +51,23 @@ class ZabbixTransformer(AlarmTransformerBase):
return self._create_vertex(entity_event) return self._create_vertex(entity_event)
def _create_vertex(self, entity_event): def _create_vertex(self, entity_event):
# TODO(Alexey): need to check the correct format for the date self._unify_time_format(entity_event)
# update_timestamp = datetime_utils.change_time_str_format(
# entity_event[ZabbixProperties.LAST_CHANGE], update_timestamp = entity_event[ZabbixProps.TIMESTAMP]
# '%Y-%m-%d %H:%M:%S',
# tbase.TIMESTAMP_FORMAT)
update_timestamp = entity_event[ZabbixProperties.LAST_CHANGE]
sample_timestamp = entity_event[DSProps.SAMPLE_DATE] sample_timestamp = entity_event[DSProps.SAMPLE_DATE]
update_timestamp = self._format_update_timestamp(update_timestamp, update_timestamp = self._format_update_timestamp(update_timestamp,
sample_timestamp) sample_timestamp)
severity = entity_event[ZabbixProperties.STATUS] value = entity_event[ZabbixProps.VALUE]
entity_state = AlarmProps.INACTIVE_STATE if \ entity_state = AlarmProps.INACTIVE_STATE if \
severity == ZabbixTriggerStatus.OK else AlarmProps.ACTIVE_STATE value == TriggerValue.OK else AlarmProps.ACTIVE_STATE
metadata = { metadata = {
VProps.NAME: entity_event[ZabbixProperties.DESCRIPTION], VProps.NAME: entity_event[ZabbixProps.DESCRIPTION],
VProps.SEVERITY: severity VProps.SEVERITY: TriggerSeverity.str(
entity_event[ZabbixProps.PRIORITY])
} }
return graph_utils.create_vertex( return graph_utils.create_vertex(
@ -80,21 +86,18 @@ class ZabbixTransformer(AlarmTransformerBase):
return self._create_zabbix_neighbors(entity_event) return self._create_zabbix_neighbors(entity_event)
def _create_zabbix_neighbors(self, entity_event): def _create_zabbix_neighbors(self, entity_event):
vitrage_id = self._create_entity_key(entity_event) self._unify_time_format(entity_event)
# TODO(Alexey): need to check the correct format for the date
# timestamp = datetime_utils.change_time_str_format(
# entity_event[ZabbixProperties.LAST_CHANGE],
# '%Y-%m-%d %H:%M:%S',
# tbase.TIMESTAMP_FORMAT)
timestamp = entity_event[DSProps.SAMPLE_DATE]
resource_type = entity_event[ZabbixProperties.RESOURCE_TYPE] vitrage_id = self._create_entity_key(entity_event)
timestamp = entity_event[ZabbixProps.TIMESTAMP]
resource_type = entity_event[ZabbixProps.RESOURCE_TYPE]
if resource_type == NOVA_HOST_DATASOURCE or resource_type == SWITCH: if resource_type == NOVA_HOST_DATASOURCE or resource_type == SWITCH:
return [self._create_neighbor( return [self._create_neighbor(
vitrage_id, vitrage_id,
timestamp, timestamp,
resource_type, resource_type,
entity_event[ZabbixProperties.RESOURCE_NAME])] entity_event[ZabbixProps.RESOURCE_NAME])]
return [] return []
@ -125,13 +128,24 @@ class ZabbixTransformer(AlarmTransformerBase):
return None return None
def _ok_status(self, entity_event): def _ok_status(self, entity_event):
return entity_event[ZabbixProperties.STATUS] == ZabbixTriggerStatus.OK return entity_event[ZabbixProps.VALUE] == TriggerValue.OK
def _create_entity_key(self, entity_event): def _create_entity_key(self, entity_event):
sync_type = entity_event[DSProps.SYNC_TYPE] sync_type = entity_event[DSProps.SYNC_TYPE]
alarm_name = entity_event[ZabbixProperties.DESCRIPTION] alarm_name = entity_event[ZabbixProps.DESCRIPTION]
resource_name = entity_event[ZabbixProperties.RESOURCE_NAME] resource_name = entity_event[ZabbixProps.RESOURCE_NAME]
return tbase.build_key(self._key_values(sync_type, return tbase.build_key(self._key_values(sync_type,
resource_name, resource_name,
alarm_name)) alarm_name))
@staticmethod
def _unify_time_format(entity_event):
try:
entity_event[ZabbixProps.TIMESTAMP] = format_unix_timestamp(
entity_event[ZabbixProps.LAST_CHANGE], tbase.TIMESTAMP_FORMAT)
except ValueError:
entity_event[ZabbixProps.TIMESTAMP] = change_time_str_format(
entity_event[ZabbixProps.LAST_CHANGE],
ZabbixProps.ZABBIX_TIMESTAMP_FORMAT,
tbase.TIMESTAMP_FORMAT)

4
vitrage/tests/resources/mock_configurations/driver/driver_zabbix_snapshot_dynamic.json
View File

@ -3,8 +3,10 @@
"resource_type": "nova\\.host", "resource_type": "nova\\.host",
"resource_name": "compute-[1-9]", "resource_name": "compute-[1-9]",
"description": "CPU utilization|Check_MK|Uptime", "description": "CPU utilization|Check_MK|Uptime",
"name": "CPU utilization|Check_MK|Uptime",
"lastchange": "1469450551",
"timestamp": "2015.12.01 12:46:41",
"status": "0", "status": "0",
"lastchange": "2016-02-07 15:26:04",
"sample_date": "2015-12-01T12:46:41Z", "sample_date": "2015-12-01T12:46:41Z",
"value": "1", "value": "1",
"priority": "1|2|3|4|5", "priority": "1|2|3|4|5",

7
vitrage/tests/unit/datasources/zabbix/test_zabbix_configuration.py
View File

@ -87,13 +87,6 @@ class TestZabbixConfig(base.BaseTest):
for expected_mapping in self.NON_EXISTING_MAPPINGS.items(): for expected_mapping in self.NON_EXISTING_MAPPINGS.items():
self.assertFalse(self._check_contains(expected_mapping, mappings)) self.assertFalse(self._check_contains(expected_mapping, mappings))
def test_zabbix_status_mapping(self):
# Action
mappings = ZabbixDriver._status_mapping()
# Test assertions
self.assertEqual(7, len(mappings))
@staticmethod @staticmethod
def _check_contains(expected_mapping, mappings): def _check_contains(expected_mapping, mappings):
for mapping in mappings.items(): for mapping in mappings.items():

272
vitrage/tests/unit/datasources/zabbix/test_zabbix_driver.py
View File

@ -49,24 +49,24 @@ class ZabbixDriverTest(ZabbixBaseTest):
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization 1', ZabbixProps.DESCRIPTION: 'CPU utilization 1',
ZabbixProps.IS_ALARM_DISABLED: '1', ZabbixProps.STATUS: '1',
ZabbixProps.IS_ALARM_ON: '0', ZabbixProps.VALUE: '0',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization 2', ZabbixProps.DESCRIPTION: 'CPU utilization 2',
ZabbixProps.IS_ALARM_DISABLED: '1', ZabbixProps.STATUS: '1',
ZabbixProps.IS_ALARM_ON: '1', ZabbixProps.VALUE: '1',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization 3', ZabbixProps.DESCRIPTION: 'CPU utilization 3',
ZabbixProps.IS_ALARM_DISABLED: '0', ZabbixProps.STATUS: '0',
ZabbixProps.IS_ALARM_ON: '1', ZabbixProps.VALUE: '1',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1'}
alarm_data4 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data4 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization 4', ZabbixProps.DESCRIPTION: 'CPU utilization 4',
ZabbixProps.IS_ALARM_DISABLED: '0', ZabbixProps.STATUS: '0',
ZabbixProps.IS_ALARM_ON: '0', ZabbixProps.VALUE: '0',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -75,7 +75,7 @@ class ZabbixDriverTest(ZabbixBaseTest):
expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization 3', ZabbixProps.DESCRIPTION: 'CPU utilization 3',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1'}
# Action # Action
alarms = zabbix_driver._get_all_alarms() alarms = zabbix_driver._get_all_alarms()
@ -100,13 +100,13 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -122,13 +122,14 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.VALUE: '1',
ZabbixProps.PRIORITY: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -144,54 +145,58 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.VALUE: '1',
ZabbixProps.PRIORITY: '4'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.VALUE: '1',
ZabbixProps.PRIORITY: '1'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
alarm_data3]) alarm_data3])
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.VALUE: '1',
excpected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2', ZabbixProps.PRIORITY: '4'}
expected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.VALUE: '1',
ZabbixProps.PRIORITY: '1'}
alarms = zabbix_driver._get_all_alarms() alarms = zabbix_driver._get_all_alarms()
# Test assertions # Test assertions
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(2, len(alarms)) self.assertEqual(2, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
alarm_data3]) alarm_data3])
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
excpected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2', expected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
alarms = zabbix_driver._get_all_alarms() alarms = zabbix_driver._get_all_alarms()
@ -200,8 +205,8 @@ class ZabbixDriverTest(ZabbixBaseTest):
# status is OK, because they were not OK earlier # status is OK, because they were not OK earlier
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(2, len(alarms)) self.assertEqual(2, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
# Action # Action
alarms = zabbix_driver._get_all_alarms() alarms = zabbix_driver._get_all_alarms()
@ -225,13 +230,16 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '2',
ZabbixProps.VALUE: '0'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '2',
ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -247,13 +255,16 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '2',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '2',
ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -269,88 +280,102 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
alarm_data3]) alarm_data3])
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
excpected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2', ZabbixProps.VALUE: '1'}
expected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
# Test assertions # Test assertions
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(2, len(alarms)) self.assertEqual(2, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
alarm_data3]) alarm_data3])
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'host2', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
# Test assertions # Test assertions
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(1, len(alarms)) self.assertEqual(1, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '0'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
alarm_data3]) alarm_data3])
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '4',
excpected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2', ZabbixProps.VALUE: '0'}
expected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '0'}
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
# Test assertions # Test assertions
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(2, len(alarms)) self.assertEqual(2, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
# Action # Action
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
@ -368,13 +393,16 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '2',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '2',
ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -415,32 +443,37 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
alarm_data3]) alarm_data3])
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
excpected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2', ZabbixProps.VALUE: '1'}
expected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarms = zabbix_driver._get_all_alarms() alarms = zabbix_driver._get_all_alarms()
# Test assertions # Test assertions
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(2, len(alarms)) self.assertEqual(2, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
# Action # Action
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
@ -457,49 +490,57 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Calling get_all for the second time should return the same results # Calling get_all for the second time should return the same results
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(2, len(alarms)) self.assertEqual(2, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
alarm_data3]) alarm_data3])
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'host2', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
excpected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2', ZabbixProps.VALUE: '1'}
expected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
# Test assertions # Test assertions
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(2, len(alarms)) self.assertEqual(2, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -507,7 +548,8 @@ class ZabbixDriverTest(ZabbixBaseTest):
expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
@ -516,15 +558,18 @@ class ZabbixDriverTest(ZabbixBaseTest):
self.assertEqual(1, len(alarms)) self.assertEqual(1, len(alarms))
self._assert_contains(expected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
excpected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', expected_alarm1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
excpected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2', ZabbixProps.VALUE: '1'}
expected_alarm2 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
excpected_alarm3 = {ZabbixProps.RESOURCE_NAME: 'host2', excpected_alarm3 = {ZabbixProps.RESOURCE_NAME: 'host2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '4'} ZabbixProps.PRIORITY: '4',
ZabbixProps.VALUE: '1'}
# Action # Action
alarms = zabbix_driver._get_changed_alarms() alarms = zabbix_driver._get_changed_alarms()
@ -540,8 +585,8 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Calling get_all for the second time should return the same results # Calling get_all for the second time should return the same results
self.assertIsNotNone(alarms, 'No alarms returned') self.assertIsNotNone(alarms, 'No alarms returned')
self.assertEqual(3, len(alarms)) self.assertEqual(3, len(alarms))
self._assert_contains(excpected_alarm1, alarms) self._assert_contains(expected_alarm1, alarms)
self._assert_contains(excpected_alarm2, alarms) self._assert_contains(expected_alarm2, alarms)
self._assert_contains(excpected_alarm3, alarms) self._assert_contains(excpected_alarm3, alarms)
def test_delete_alarm(self): def test_delete_alarm(self):
@ -553,13 +598,14 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action # Action
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data3 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'Uptime', ZabbixProps.DESCRIPTION: 'Uptime',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, zabbix_driver.set_alarm_datas([alarm_data1,
alarm_data2, alarm_data2,
@ -575,10 +621,12 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action - delete a alarm that was OK # Action - delete a alarm that was OK
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, alarm_data2]) zabbix_driver.set_alarm_datas([alarm_data1, alarm_data2])
@ -592,7 +640,8 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action - delete a alarm that was not OK # Action - delete a alarm that was not OK
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data2]) zabbix_driver.set_alarm_datas([alarm_data2])
@ -615,10 +664,12 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action - "undelete" the alarm that was OK # Action - "undelete" the alarm that was OK
alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1', alarm_data1 = {ZabbixProps.RESOURCE_NAME: 'compute-1',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '1'}
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data1, alarm_data2]) zabbix_driver.set_alarm_datas([alarm_data1, alarm_data2])
@ -633,7 +684,8 @@ class ZabbixDriverTest(ZabbixBaseTest):
# Action - delete a alarm that was not OK and call get_changes # Action - delete a alarm that was not OK and call get_changes
alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2', alarm_data2 = {ZabbixProps.RESOURCE_NAME: 'compute-2',
ZabbixProps.DESCRIPTION: 'CPU utilization', ZabbixProps.DESCRIPTION: 'CPU utilization',
ZabbixProps.SEVERITY: '-1'} ZabbixProps.PRIORITY: '1',
ZabbixProps.VALUE: '0'}
zabbix_driver.set_alarm_datas([alarm_data2]) zabbix_driver.set_alarm_datas([alarm_data2])

42
vitrage/tests/unit/datasources/zabbix/test_zabbix_transformer.py
View File

@ -20,13 +20,16 @@ from vitrage.common.constants import EntityCategory
from vitrage.common.constants import EventAction from vitrage.common.constants import EventAction
from vitrage.common.constants import SyncMode from vitrage.common.constants import SyncMode
from vitrage.common.constants import VertexProperties as VProps from vitrage.common.constants import VertexProperties as VProps
from vitrage.common.datetime_utils import format_unix_timestamp
from vitrage.datasources.alarm_properties import AlarmProperties as AlarmProps from vitrage.datasources.alarm_properties import AlarmProperties as AlarmProps
from vitrage.datasources.nova.host import NOVA_HOST_DATASOURCE from vitrage.datasources.nova.host import NOVA_HOST_DATASOURCE
from vitrage.datasources.nova.host.transformer import HostTransformer from vitrage.datasources.nova.host.transformer import HostTransformer
from vitrage.datasources import transformer_base as tbase
from vitrage.datasources.transformer_base import TransformerBase from vitrage.datasources.transformer_base import TransformerBase
from vitrage.datasources.zabbix.driver import ZabbixDriver from vitrage.datasources.zabbix.properties import ZabbixProperties\
from vitrage.datasources.zabbix.properties import ZabbixProperties as ZabbixProps
from vitrage.datasources.zabbix.properties import ZabbixTriggerStatus from vitrage.datasources.zabbix.properties import ZabbixTriggerSeverity
from vitrage.datasources.zabbix.properties import ZabbixTriggerValue
from vitrage.datasources.zabbix.transformer import ZabbixTransformer from vitrage.datasources.zabbix.transformer import ZabbixTransformer
from vitrage.tests import base from vitrage.tests import base
from vitrage.tests.mocks import mock_driver as mock_sync from vitrage.tests.mocks import mock_driver as mock_sync
@ -54,6 +57,7 @@ class ZabbixTransformerTest(base.BaseTest):
zabbix_alarms = mock_sync.generate_sequential_events_list(spec_list) zabbix_alarms = mock_sync.generate_sequential_events_list(spec_list)
transformer = ZabbixTransformer(self.transformers) transformer = ZabbixTransformer(self.transformers)
event = zabbix_alarms[0] event = zabbix_alarms[0]
self.enrich_event(event)
# Test action # Test action
observed_key = transformer._create_entity_key(event) observed_key = transformer._create_entity_key(event)
@ -64,9 +68,9 @@ class ZabbixTransformerTest(base.BaseTest):
self.assertEqual(EntityCategory.ALARM, observed_key_fields[0]) self.assertEqual(EntityCategory.ALARM, observed_key_fields[0])
self.assertEqual(event[DSProps.SYNC_TYPE], observed_key_fields[1]) self.assertEqual(event[DSProps.SYNC_TYPE], observed_key_fields[1])
self.assertEqual(event[ZabbixProperties.RESOURCE_NAME], self.assertEqual(event[ZabbixProps.RESOURCE_NAME],
observed_key_fields[2]) observed_key_fields[2])
self.assertEqual(event[ZabbixProperties.DESCRIPTION], self.assertEqual(event[ZabbixProps.DESCRIPTION],
observed_key_fields[3]) observed_key_fields[3])
def test_zabbix_alarm_transform(self): def test_zabbix_alarm_transform(self):
@ -77,13 +81,9 @@ class ZabbixTransformerTest(base.BaseTest):
events_num=10) events_num=10)
zabbix_alarms = mock_sync.generate_sequential_events_list(spec_list) zabbix_alarms = mock_sync.generate_sequential_events_list(spec_list)
# convert to correct status
for alar in zabbix_alarms:
alar[ZabbixProperties.STATUS] = \
ZabbixDriver._status_mapping()[alar[ZabbixProperties.SEVERITY]]
for alarm in zabbix_alarms: for alarm in zabbix_alarms:
# Test action # Test action
self.enrich_event(alarm, format_timestamp=False)
wrapper = ZabbixTransformer(self.transformers).transform(alarm) wrapper = ZabbixTransformer(self.transformers).transform(alarm)
self._validate_vertex(wrapper.vertex, alarm) self._validate_vertex(wrapper.vertex, alarm)
@ -100,7 +100,7 @@ class ZabbixTransformerTest(base.BaseTest):
def _validate_action(self, alarm, wrapper): def _validate_action(self, alarm, wrapper):
sync_mode = alarm[DSProps.SYNC_MODE] sync_mode = alarm[DSProps.SYNC_MODE]
if sync_mode in (SyncMode.SNAPSHOT, SyncMode.UPDATE): if sync_mode in (SyncMode.SNAPSHOT, SyncMode.UPDATE):
if alarm[ZabbixProperties.STATUS] == 'OK': if alarm[ZabbixProps.VALUE] == ZabbixTriggerValue.OK:
self.assertEqual(EventAction.DELETE_ENTITY, wrapper.action) self.assertEqual(EventAction.DELETE_ENTITY, wrapper.action)
else: else:
self.assertEqual(EventAction.UPDATE_ENTITY, wrapper.action) self.assertEqual(EventAction.UPDATE_ENTITY, wrapper.action)
@ -111,19 +111,21 @@ class ZabbixTransformerTest(base.BaseTest):
self.assertEqual(EntityCategory.ALARM, vertex[VProps.CATEGORY]) self.assertEqual(EntityCategory.ALARM, vertex[VProps.CATEGORY])
self.assertEqual(event[DSProps.SYNC_TYPE], vertex[VProps.TYPE]) self.assertEqual(event[DSProps.SYNC_TYPE], vertex[VProps.TYPE])
self.assertEqual(event[ZabbixProperties.DESCRIPTION], self.assertEqual(event[ZabbixProps.DESCRIPTION],
vertex[VProps.NAME]) vertex[VProps.NAME])
event_status = event[ZabbixProperties.STATUS] event_status = event[ZabbixProps.VALUE]
if event_status == ZabbixTriggerStatus.OK: if event_status == ZabbixTriggerValue.OK:
self.assertEqual(AlarmProps.INACTIVE_STATE, self.assertEqual(AlarmProps.INACTIVE_STATE,
vertex[VProps.STATE]) vertex[VProps.STATE])
else: else:
self.assertEqual(AlarmProps.ACTIVE_STATE, self.assertEqual(AlarmProps.ACTIVE_STATE,
vertex[VProps.STATE]) vertex[VProps.STATE])
self.assertEqual(event_status, vertex[VProps.SEVERITY]) event_severity = ZabbixTriggerSeverity.str(
event[ZabbixProps.PRIORITY])
self.assertEqual(event_severity, vertex[VProps.SEVERITY])
self.assertFalse(vertex[VProps.IS_DELETED]) self.assertFalse(vertex[VProps.IS_DELETED])
self.assertFalse(vertex[VProps.IS_PLACEHOLDER]) self.assertFalse(vertex[VProps.IS_PLACEHOLDER])
@ -136,13 +138,13 @@ class ZabbixTransformerTest(base.BaseTest):
self.assertEqual(EntityCategory.RESOURCE, key_fields[0]) self.assertEqual(EntityCategory.RESOURCE, key_fields[0])
self.assertEqual(NOVA_HOST_DATASOURCE, key_fields[1]) self.assertEqual(NOVA_HOST_DATASOURCE, key_fields[1])
self.assertEqual(event[ZabbixProperties.RESOURCE_NAME], key_fields[2]) self.assertEqual(event[ZabbixProps.RESOURCE_NAME], key_fields[2])
self.assertFalse(host_vertex[VProps.IS_DELETED]) self.assertFalse(host_vertex[VProps.IS_DELETED])
self.assertTrue(host_vertex[VProps.IS_PLACEHOLDER]) self.assertTrue(host_vertex[VProps.IS_PLACEHOLDER])
self.assertEqual(EntityCategory.RESOURCE, host_vertex[VProps.CATEGORY]) self.assertEqual(EntityCategory.RESOURCE, host_vertex[VProps.CATEGORY])
self.assertEqual(event[ZabbixProperties.RESOURCE_NAME], self.assertEqual(event[ZabbixProps.RESOURCE_NAME],
host_vertex[VProps.ID]) host_vertex[VProps.ID])
self.assertEqual(NOVA_HOST_DATASOURCE, host_vertex[VProps.TYPE]) self.assertEqual(NOVA_HOST_DATASOURCE, host_vertex[VProps.TYPE])
@ -153,3 +155,9 @@ class ZabbixTransformerTest(base.BaseTest):
ZabbixTransformer(self.transformers)._create_entity_key(event) ZabbixTransformer(self.transformers)._create_entity_key(event)
self.assertEqual(alarm_key, edge.source_id) self.assertEqual(alarm_key, edge.source_id)
self.assertEqual(host_vertex.vertex_id, edge.target_id) self.assertEqual(host_vertex.vertex_id, edge.target_id)
@staticmethod
def enrich_event(event, format_timestamp=True):
if format_timestamp:
event[ZabbixProps.TIMESTAMP] = format_unix_timestamp(
event[ZabbixProps.LAST_CHANGE], tbase.TIMESTAMP_FORMAT)