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.threshold = 50.0
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))