4edfa862564123291a78af382219106d400981e1
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
21 # For those usages not covered by the Apache License, Version 2.0 please
22 # contact: bdiaz@whitestack.com or glavado@whitestack.com
25 from unittest
import TestCase
, mock
27 from osm_policy_module
.alarming
.service
import AlarmingService
28 from osm_policy_module
.common
.common_db_client
import CommonDbClient
29 from osm_policy_module
.common
.lcm_client
import LcmClient
30 from osm_policy_module
.common
.mon_client
import MonClient
31 from osm_policy_module
.core
.config
import Config
32 from osm_policy_module
.core
.database
import VnfAlarmRepository
35 @mock.patch
.object(LcmClient
, "__init__", lambda *args
, **kwargs
: None)
36 @mock.patch
.object(MonClient
, "__init__", lambda *args
, **kwargs
: None)
37 @mock.patch
.object(CommonDbClient
, "__init__", lambda *args
, **kwargs
: None)
38 class TestAlarmingService(TestCase
):
41 self
.config
= Config()
42 self
.loop
= asyncio
.new_event_loop()
44 @mock.patch
.object(VnfAlarmRepository
, 'get')
45 @mock.patch('requests.post')
46 @mock.patch('osm_policy_module.core.database.db')
47 def test_handle_alarm(self
, database
, requests_post
, get_alarm
):
48 mock_alarm
= self
._build
_mock
_alarm
('test_id')
49 get_alarm
.return_value
= mock_alarm
51 service
= AlarmingService(self
.config
)
52 self
.loop
.run_until_complete(service
.handle_alarm('test_id', 'alarm', {}))
53 requests_post
.assert_called_once_with(json
='{}', url
='http://alarm-url/')
55 requests_post
.reset_mock()
56 self
.loop
.run_until_complete(service
.handle_alarm('test_id', 'ok', {}))
57 requests_post
.assert_called_once_with(json
='{}', url
='http://ok-url/')
59 requests_post
.reset_mock()
60 self
.loop
.run_until_complete(service
.handle_alarm('test_id', 'insufficient-data', {}))
61 requests_post
.assert_called_once_with(json
='{}', url
='http://insufficient-data-url/')
63 @mock.patch
.object(VnfAlarmRepository
, 'get')
64 @mock.patch('requests.post')
65 @mock.patch('osm_policy_module.core.database.db')
66 def test_handle_alarm_unknown_status(self
, database
, requests_post
, get_alarm
):
67 mock_alarm
= self
._build
_mock
_alarm
('test_id')
68 get_alarm
.return_value
= mock_alarm
70 service
= AlarmingService(self
.config
)
71 self
.loop
.run_until_complete(service
.handle_alarm('test_id', 'unknown', {}))
72 requests_post
.assert_not_called()
74 def _build_mock_alarm(self
,
76 alarm_url
='http://alarm-url/',
77 insufficient_data_url
='http://insufficient-data-url/',
78 ok_url
='http://ok-url/'):
79 mock_alarm
= mock
.Mock()
80 mock_alarm
.alarm_id
= alarm_id
81 insufficient_data_action
= mock
.Mock()
82 insufficient_data_action
.type = 'insufficient-data'
83 insufficient_data_action
.url
= insufficient_data_url
84 alarm_action
= mock
.Mock()
85 alarm_action
.type = 'alarm'
86 alarm_action
.url
= alarm_url
87 ok_action
= mock
.Mock()
89 ok_action
.url
= ok_url
90 mock_alarm
.actions
= [insufficient_data_action
, alarm_action
, ok_action
]