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
.evaluator
import AlarmStatus
, Evaluator
29 from osm_mon
.evaluator
.service
import EvaluatorService
32 @mock.patch
.object(CommonDbClient
, "__init__", lambda *args
, **kwargs
: None)
33 @mock.patch
.object(MessageBusClient
, "__init__", lambda *args
, **kwargs
: None)
34 class EvaluatorTest(TestCase
):
37 self
.config
= Config()
39 @mock.patch('concurrent.futures.ProcessPoolExecutor.submit')
40 @mock.patch
.object(Evaluator
, "_notify_alarm")
41 @mock.patch
.object(EvaluatorService
, "evaluate_alarms")
42 def test_evaluate(self
, evaluate_alarms
, _notify_alarm
, futures
):
43 mock_alarm
= mock
.Mock()
44 mock_alarm
.operation
= 'gt'
45 mock_alarm
.threshold
= 50.0
46 evaluate_alarms
.return_value
= [(mock_alarm
, AlarmStatus
.ALARM
)]
48 evaluator
= Evaluator(self
.config
)
51 futures
.assert_called_with(Evaluator
._notify
_alarm
, self
.config
, mock_alarm
, AlarmStatus
.ALARM
)