1 # -*- coding: utf-8 -*-
3 # Copyright 2018 Whitestack, LLC
4 # *************************************************************
6 # This file is part of OSM Monitoring module
7 # All Rights Reserved to Whitestack, LLC
9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
10 # not use this file except in compliance with the License. You may obtain
11 # a copy of the License at
13 # http://www.apache.org/licenses/LICENSE-2.0
15 # Unless required by applicable law or agreed to in writing, software
16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
18 # License for the specific language governing permissions and limitations
20 # For those usages not covered by the Apache License, Version 2.0 please
21 # contact: bdiaz@whitestack.com or glavado@whitestack.com
23 from unittest
import TestCase
, mock
25 from osm_mon
.core
.common_db
import CommonDbClient
26 from osm_mon
.core
.config
import Config
27 from osm_mon
.core
.message_bus_client
import MessageBusClient
28 from osm_mon
.evaluator
.backends
.prometheus
import PrometheusBackend
29 from osm_mon
.evaluator
.evaluator
import AlarmStatus
30 from osm_mon
.evaluator
.service
import EvaluatorService
33 "_id": "0d9d06ad-3fc2-418c-9934-465e815fafe2",
34 "ip-address": "192.168.160.2",
35 "created-time": 1535392482.0044956,
36 "vim-account-id": "be48ae31-1d46-4892-a4b4-d69abd55714b",
41 "mac-address": "fa:16:3e:71:fd:b8",
43 "ip-address": "192.168.160.2"
47 "vim-id": "63a65636-9fc8-4022-b070-980823e6266a",
48 "name": "cirros_ns-1-cirros_vnfd-VM-1",
49 "status-detailed": None,
50 "ip-address": "192.168.160.2",
51 "vdu-id-ref": "cirros_vnfd-VM"
54 "id": "0d9d06ad-3fc2-418c-9934-465e815fafe2",
55 "vnfd-ref": "cirros_vdu_scaling_vnf",
56 "vnfd-id": "63f44c41-45ee-456b-b10d-5f08fb1796e0",
58 "created": 1535392482.0067868,
62 "modified": 1535392482.0067868,
67 "nsr-id-ref": "87776f33-b67c-417a-8119-cb08e4098951",
68 "member-vnf-index-ref": "1",
73 "connection-point-id": None
79 "_id": "63f44c41-45ee-456b-b10d-5f08fb1796e0",
80 "id": "cirros_vdu_scaling_vnf",
82 "product-name": "cirros_vdu_scaling_vnf",
86 "id": "cirros_vnfd-VM",
87 "name": "cirros_vnfd-VM",
90 "virtual-network-interface-requirement": [
98 "virtual-compute-desc": "cirros_vnfd-VM-compute",
99 "virtual-storage-desc": [
100 "cirros_vnfd-VM-storage"
102 "sw-image-desc": "cirros034",
103 "monitoring-parameter": [
105 "id": "cirros_vnfd-VM_memory_util",
106 "name": "cirros_vnfd-VM_memory_util",
107 "performance-metric": "average_memory_utilization"
112 "virtual-compute-desc": [
114 "id": "cirros_vnfd-VM-compute",
123 "virtual-storage-desc": [
125 "id": "cirros_vnfd-VM-storage",
139 "vdu-id": "cirros_vnfd-VM",
140 "cpd": "vdu-eth0-int"
150 "id": "cirros_vnfd-VM"
153 "instantiation-level": [
155 "id": "default-instantiation-level",
158 "vdu-id": "cirros_vnfd-VM"
165 "description": "Simple VNF example with a cirros and a scaling group descriptor",
166 "mgmt-cp": "vnf-cp0-ext"
170 @mock.patch
.object(CommonDbClient
, "__init__", lambda *args
, **kwargs
: None)
171 @mock.patch
.object(MessageBusClient
, "__init__", lambda *args
, **kwargs
: None)
172 class EvaluatorTest(TestCase
):
175 self
.config
= Config()
177 @mock.patch
.object(EvaluatorService
, "_get_metric_value")
178 def test_evaluate_metric(self
, get_metric_value
):
179 mock_alarm
= mock
.Mock()
180 mock_alarm
.operation
= 'gt'
181 mock_alarm
.threshold
= 50.0
182 mock_alarm
.metric
= 'metric_name'
183 get_metric_value
.return_value
= 100.0
185 service
= EvaluatorService(self
.config
)
186 service
.queue
= mock
.Mock()
187 service
._evaluate
_metric
(mock_alarm
)
188 service
.queue
.put
.assert_called_with((mock_alarm
, AlarmStatus
.ALARM
))
189 service
.queue
.reset_mock()
191 mock_alarm
.operation
= 'lt'
192 service
._evaluate
_metric
(mock_alarm
)
193 service
.queue
.put
.assert_called_with((mock_alarm
, AlarmStatus
.OK
))
194 service
.queue
.reset_mock()
196 get_metric_value
.return_value
= None
197 service
._evaluate
_metric
(mock_alarm
)
198 service
.queue
.put
.assert_called_with((mock_alarm
, AlarmStatus
.INSUFFICIENT
))
200 @mock.patch('multiprocessing.Process')
201 @mock.patch
.object(EvaluatorService
, "_evaluate_metric")
202 @mock.patch
.object(CommonDbClient
, "get_vnfd")
203 @mock.patch
.object(CommonDbClient
, "get_vnfr")
204 @mock.patch
.object(CommonDbClient
, "get_alarms")
205 def test_evaluate_alarms(self
, alarm_list
, get_vnfr
, get_vnfd
, evaluate_metric
, process
):
206 mock_alarm
= mock
.Mock()
207 mock_alarm
.vdur_name
= 'cirros_ns-1-cirros_vnfd-VM-1'
208 mock_alarm
.monitoring_param
= 'cirros_vnf_memory_util'
209 mock_alarm
.tags
= {'name': 'value'}
210 alarm_list
.return_value
= [mock_alarm
]
211 get_vnfr
.return_value
= vnfr_record_mock
212 get_vnfd
.return_value
= vnfd_record_mock
214 evaluator
= EvaluatorService(self
.config
)
215 evaluator
.evaluate_alarms()
217 process
.assert_called_with(target
=evaluate_metric
, args
=(mock_alarm
,))
219 @mock.patch
.object(PrometheusBackend
, "get_metric_value")
220 def test_get_metric_value_prometheus(self
, get_metric_value
):
221 self
.config
.set('evaluator', 'backend', 'prometheus')
222 evaluator
= EvaluatorService(self
.config
)
223 evaluator
._get
_metric
_value
('test', {})
225 get_metric_value
.assert_called_with('test', {})