71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
# Copyright (c) 2014 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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.
|
|
|
|
"""Tests the notification class."""
|
|
|
|
import json
|
|
from mon_notification import notification
|
|
|
|
|
|
def test_json():
|
|
"""Test the to_json method to verify it behaves as expected.
|
|
"""
|
|
alarm = {'alarmId': 'alarmId',
|
|
'alarmName': 'alarmName',
|
|
'timestamp': 'timestamp',
|
|
'stateChangeReason': 'stateChangeReason',
|
|
'newState': 'newState',
|
|
'tenantId': 'tenantId'}
|
|
test_notification = notification.Notification('ntype', 'src_partition', 'src_offset', 'name', 'address', alarm)
|
|
|
|
expected_dict = {u'name': u'name',
|
|
u'notification_timestamp': None,
|
|
u'tenant_id': u'tenantId',
|
|
u'alarm_name': u'alarmName',
|
|
u'alarm_id': u'alarmId',
|
|
u'state': u'newState',
|
|
u'alarm_timestamp': u'timestamp',
|
|
u'address': u'address',
|
|
u'message': u'stateChangeReason'}
|
|
# Compare as dicts so ordering is not an issue
|
|
assert json.loads(test_notification.to_json()) == expected_dict
|
|
|
|
|
|
def test_equal():
|
|
alarm = {'alarmId': 'alarmId',
|
|
'alarmName': 'alarmName',
|
|
'timestamp': 'timestamp',
|
|
'stateChangeReason': 'stateChangeReason',
|
|
'newState': 'newState',
|
|
'tenantId': 'tenantId'}
|
|
test_notification = notification.Notification('ntype', 'src_partition', 'src_offset', 'name', 'address', alarm)
|
|
test_notification2 = notification.Notification('ntype', 'src_partition', 'src_offset', 'name', 'address', alarm)
|
|
|
|
assert(test_notification == test_notification2)
|
|
|
|
|
|
def test_unequal():
|
|
alarm = {'alarmId': 'alarmId',
|
|
'alarmName': 'alarmName',
|
|
'timestamp': 'timestamp',
|
|
'stateChangeReason': 'stateChangeReason',
|
|
'newState': 'newState',
|
|
'tenantId': 'tenantId'}
|
|
test_notification = notification.Notification('ntype', 'src_partition', 'src_offset', 'name', 'address', alarm)
|
|
test_notification2 = notification.Notification('ntype', 'src_partition', 'src_offset', 'name', 'address', alarm)
|
|
test_notification2.alarm_id = None
|
|
|
|
assert(test_notification != test_notification2)
|