super().setUp()
self.config = Config()
- @mock.patch('multiprocessing.Process')
- @mock.patch.object(Evaluator, "notify_alarm")
+ @mock.patch('concurrent.futures.ProcessPoolExecutor.submit')
+ @mock.patch.object(Evaluator, "_notify_alarm")
@mock.patch.object(EvaluatorService, "evaluate_alarms")
- @mock.patch('osm_mon.core.database.db')
- def test_evaluate(self, db, evaluate_alarms, notify_alarm, process):
+ def test_evaluate(self, evaluate_alarms, _notify_alarm, futures):
mock_alarm = mock.Mock()
mock_alarm.operation = 'gt'
mock_alarm.threshold = 50.0
evaluator = Evaluator(self.config)
evaluator.evaluate()
- process.assert_called_with(target=notify_alarm, args=(mock_alarm, AlarmStatus.ALARM))
+ futures.assert_called_with(Evaluator._notify_alarm, self.config, mock_alarm, AlarmStatus.ALARM)