def test_check(monkeypatch): reload(plugin_manager) plugin_manager.init_plugin_manager() # init plugin manager MainTask.configure({}) task = MainTask() monkeypatch.setattr(task, '_get_check_result', MagicMock()) monkeypatch.setattr(task, '_store_check_result', MagicMock()) monkeypatch.setattr(task, 'send_metrics', MagicMock()) req = {'check_id': 123, 'entity': {'id': 'myent'}} task.check(req)
def test_check_result_size_violation(monkeypatch, fx_big_result): config, result = fx_big_result reload(plugin_manager) plugin_manager.init_plugin_manager() # init plugin manager monkeypatch.setattr('zmon_worker_monitor.zmon_worker.tasks.main.MAX_RESULT_SIZE', 2) # Lower default limit to 2K MainTask.configure(config) task = MainTask() get_result = MagicMock() get_result.return_value = {'value': result} monkeypatch.setattr(task, '_get_check_result_internal', get_result) monkeypatch.setattr(task, '_store_check_result', MagicMock()) monkeypatch.setattr(task, 'send_metrics', MagicMock()) req = {'check_id': 123, 'entity': {'id': 'myent'}} with pytest.raises(ResultSizeError): task.check(req)