def test_celery_none(self):
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.const.DEFAULT_CONFIG_FILE_PATH',
self.panoptes_test_conf_file):
mock_start = create_autospec(PanoptesPluginScheduler.start, return_value=None)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesPluginScheduler.start',
mock_start):
with self.assertRaises(SystemExit):
start_enrichment_plugin_scheduler()
def test_enrichment_plugin_scheduler_agent_config_error(self):
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.const.DEFAULT_CONFIG_FILE_PATH',
self.panoptes_test_conf_file):
mock_config = MagicMock(side_effect=Exception)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesCeleryEnrichmentAgentConfig',
mock_config):
with self.assertRaises(SystemExit):
start_enrichment_plugin_scheduler()
def test_enrichment_plugin_scheduler_task_config_error(self):
celery_app = self._scheduler.start()
celery_beat_service = Service(celery_app, max_interval=None, schedule_filename=None,
scheduler_cls=PanoptesCeleryPluginScheduler)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.const.DEFAULT_CONFIG_FILE_PATH',
self.panoptes_test_conf_file):
mock_getmtime = MagicMock(side_effect=Exception)
with patch('yahoo_panoptes.framework.plugins.panoptes_base_plugin.os.path.getmtime', mock_getmtime):
start_enrichment_plugin_scheduler()
enrichment_plugin_scheduler_task(celery_beat_service)
def test_basic_operations(self):
celery_app = self._scheduler.start()
celery_beat_service = Service(celery_app, max_interval=None, schedule_filename=None,
scheduler_cls=PanoptesCeleryPluginScheduler)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.const.DEFAULT_CONFIG_FILE_PATH',
self.panoptes_test_conf_file):
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesResourceCache.get_resources',
mock_get_resources):
start_enrichment_plugin_scheduler()
enrichment_plugin_scheduler_task(celery_beat_service)
def test_error_messages(self):
celery_app = self._scheduler.start()
celery_beat_service = Service(celery_app, max_interval=None, schedule_filename=None,
scheduler_cls=PanoptesCeleryPluginScheduler)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.const.DEFAULT_CONFIG_FILE_PATH',
self.panoptes_test_conf_file):
mock_plugin_manager = MagicMock(side_effect=Exception)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesPluginManager',
mock_plugin_manager):
start_enrichment_plugin_scheduler()
enrichment_plugin_scheduler_task(celery_beat_service)
def test_enrichment_plugin_scheduler_task_bad_plugin(self):
celery_app = self._scheduler.start()
celery_beat_service = Service(celery_app, max_interval=None, schedule_filename=None,
scheduler_cls=PanoptesCeleryPluginScheduler)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.const.DEFAULT_CONFIG_FILE_PATH',
self.panoptes_test_conf_file):
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesEnrichmentPluginInfo.'
'execute_frequency', 0):
start_enrichment_plugin_scheduler()
enrichment_plugin_scheduler_task(celery_beat_service)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesEnrichmentPluginInfo.'
'resource_filter', None):
start_enrichment_plugin_scheduler()
enrichment_plugin_scheduler_task(celery_beat_service)
def test_resource_cache_error(self):
celery_app = self._scheduler.start()
celery_beat_service = Service(celery_app, max_interval=None, schedule_filename=None,
scheduler_cls=PanoptesCeleryPluginScheduler)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.const.DEFAULT_CONFIG_FILE_PATH',
self.panoptes_test_conf_file):
mock_cache = MagicMock(side_effect=Exception)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesResourceCache', mock_cache):
start_enrichment_plugin_scheduler()
enrichment_plugin_scheduler_task(celery_beat_service)
mock_get_resources_exception = MagicMock(side_effect=Exception)
with patch('yahoo_panoptes.enrichment.enrichment_plugin_scheduler.PanoptesResourceCache.get_resources',
mock_get_resources_exception):
start_enrichment_plugin_scheduler()
enrichment_plugin_scheduler_task(celery_beat_service)
