osm_mon/test/OpenStack/unit/test_alarm_req.py

   1 # Copyright 2017 iIntel Research and Development Ireland Limited
   2 # **************************************************************
   3
   4 # This file is part of OSM Monitoring module
   5 # All Rights Reserved to Intel Corporation
   6
   7 # Licensed under the Apache License, Version 2.0 (the 'License'); you may
   8 # not use this file except in compliance with the License. You may obtain
   9 # a copy of the License at
  10
  11 #         http://www.apache.org/licenses/LICENSE-2.0
  12
  13 # Unless required by applicable law or agreed to in writing, software
  14 # distributed under the License is distributed on an 'AS IS' BASIS, WITHOUT
  15 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  16 # License for the specific language governing permissions and limitations
  17 # under the License.
  18
  19 # For those usages not covered by the Apache License, Version 2.0 please
  20 # contact: helena.mcgough@intel.com or adrian.hoban@intel.com
  21 ##
  22 """Tests for all alarm request message keys."""
  23
  24 import json
  25
  26 import logging
  27
  28 import unittest
  29
  30 import mock
  31
  32 from osm_mon.core.auth import AuthManager
  33 from osm_mon.core.database import VimCredentials, DatabaseManager
  34 from osm_mon.core.message_bus.producer import KafkaProducer
  35 from osm_mon.plugins.OpenStack.Aodh import alarming as alarm_req
  36 from osm_mon.plugins.OpenStack.common import Common
  37
  38 log = logging.getLogger(__name__)
  39
  40 mock_creds = VimCredentials()
  41 mock_creds.config = '{}'
  42
  43
  44 class Message(object):
  45     """A class to mock a message object value for alarm requests."""
  46
  47     def __init__(self):
  48         """Initialize a mocked message instance."""
  49         self.topic = 'alarm_request'
  50         self.key = None
  51         self.value = json.dumps({'mock_value': 'mock_details'})
  52
  53
  54 @mock.patch.object(KafkaProducer, 'publish', mock.Mock())
  55 class TestAlarmKeys(unittest.TestCase):
  56     """Integration test for alarm request keys."""
  57
  58     def setUp(self):
  59         """Setup the tests for alarm request keys."""
  60         super(TestAlarmKeys, self).setUp()
  61         self.alarming = alarm_req.Alarming()
  62         self.alarming.common = Common()
  63
  64     @mock.patch.object(AuthManager, 'get_credentials')
  65     @mock.patch.object(Common, 'get_endpoint')
  66     @mock.patch.object(Common, 'get_auth_token')
  67     def test_alarming_authentication(self, get_token, get_endpoint, get_creds):
  68         """Test getting an auth_token and endpoint for alarm requests."""
  69         # if auth_token is None environment variables are used to authenticate
  70         message = Message()
  71
  72         get_creds.return_value = mock_creds
  73
  74         self.alarming.alarming(message, 'test_id')
  75
  76         get_token.assert_called_with('test_id', verify_ssl=True)
  77         get_endpoint.assert_any_call('alarming', 'test_id', verify_ssl=True)
  78
  79     @mock.patch.object(Common, 'get_endpoint', mock.Mock())
  80     @mock.patch.object(Common, 'get_auth_token', mock.Mock())
  81     @mock.patch.object(AuthManager, 'get_credentials')
  82     @mock.patch.object(alarm_req.Alarming, 'delete_alarm')
  83     def test_delete_alarm_key(self, del_alarm, get_creds):
  84         """Test the functionality for a create alarm request."""
  85         # Mock a message value and key
  86         message = Message()
  87         message.key = 'delete_alarm_request'
  88         message.value = json.dumps({'alarm_delete_request': {
  89             'correlation_id': 1,
  90             'alarm_uuid': 'my_alarm_id'
  91         }})
  92
  93         get_creds.return_value = mock_creds
  94         del_alarm.return_value = {}
  95
  96         # Call the alarming functionality and check delete request
  97         self.alarming.alarming(message, 'test_id')
  98         del_alarm.assert_called_with(mock.ANY, mock.ANY, 'my_alarm_id', True)
  99
 100     @mock.patch.object(Common, 'get_endpoint', mock.Mock())
 101     @mock.patch.object(Common, 'get_auth_token', mock.Mock())
 102     @mock.patch.object(AuthManager, 'get_credentials')
 103     @mock.patch.object(alarm_req.Alarming, 'list_alarms')
 104     def test_list_alarm_key(self, list_alarm, get_creds):
 105         """Test the functionality for a list alarm request."""
 106         # Mock a message with list alarm key and value
 107         message = Message()
 108         message.key = 'list_alarm_request'
 109         message.value = json.dumps({'alarm_list_request': {'correlation_id': 1}})
 110
 111         get_creds.return_value = mock_creds
 112
 113         list_alarm.return_value = []
 114
 115         # Call the alarming functionality and check list functionality
 116         self.alarming.alarming(message, 'test_id')
 117         list_alarm.assert_called_with(mock.ANY, mock.ANY, {'correlation_id': 1}, True)
 118
 119     @mock.patch.object(Common, 'get_auth_token', mock.Mock())
 120     @mock.patch.object(Common, 'get_endpoint', mock.Mock())
 121     @mock.patch.object(AuthManager, 'get_credentials')
 122     @mock.patch.object(alarm_req.Alarming, 'update_alarm_state')
 123     def test_ack_alarm_key(self, ack_alarm, get_creds):
 124         """Test the functionality for an acknowledge alarm request."""
 125         # Mock a message with acknowledge alarm key and value
 126         message = Message()
 127         message.key = 'acknowledge_alarm'
 128         message.value = json.dumps({'ack_details':
 129                                         {'alarm_uuid': 'my_alarm_id'}})
 130
 131         get_creds.return_value = mock_creds
 132
 133         # Call alarming functionality and check acknowledge functionality
 134         self.alarming.alarming(message, 'test_id')
 135         ack_alarm.assert_called_with(mock.ANY, mock.ANY, 'my_alarm_id', True)
 136
 137     @mock.patch.object(Common, 'get_auth_token', mock.Mock())
 138     @mock.patch.object(Common, 'get_endpoint', mock.Mock())
 139     @mock.patch.object(DatabaseManager, 'save_alarm', mock.Mock())
 140     @mock.patch.object(Common, "perform_request")
 141     @mock.patch.object(AuthManager, 'get_credentials')
 142     @mock.patch.object(alarm_req.Alarming, 'configure_alarm')
 143     def test_config_alarm_key(self, config_alarm, get_creds, perf_req):
 144         """Test the functionality for a create alarm request."""
 145         # Mock a message with config alarm key and value
 146         message = Message()
 147         message.key = 'create_alarm_request'
 148         message.value = json.dumps({'alarm_create_request': {'correlation_id': 1, 'threshold_value': 50,
 149                                                              'operation': 'GT', 'metric_name': 'cpu_utilization',
 150                                                              'vdu_name': 'vdu',
 151                                                              'vnf_member_index': '1',
 152                                                              'ns_id': '1',
 153                                                              'resource_uuid': '123'}})
 154         mock_perf_req_return_value = {"metrics": {"cpu_util": 123}}
 155         perf_req.return_value = type('obj', (object,), {'text': json.dumps(mock_perf_req_return_value, sort_keys=True)})
 156         get_creds.return_value = mock_creds
 157
 158         # Call alarming functionality and check config alarm call
 159         config_alarm.return_value = 'my_alarm_id'
 160         self.alarming.alarming(message, 'test_id')
 161         config_alarm.assert_called_with(mock.ANY, mock.ANY, {'correlation_id': 1, 'threshold_value': 50,
 162                                                              'operation': 'GT',
 163                                                              'metric_name': 'cpu_utilization',
 164                                                              'vdu_name': 'vdu',
 165                                                              'vnf_member_index': '1', 'ns_id': '1',
 166                                                              'resource_uuid': '123'}, {}, True)