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