def test_add_new(self): """Add a new plugin to the scheduler.""" CONF.set_override('enable', True, 'scheduler') self.assertIsNone(scheduler.SCHEDULER) scheduler.initialize_scheduler() mock_plugin = MockPlugin(dict()) mock_plugin_name = mock_plugin.get_name() mock_extensions = [ Extension(mock_plugin_name, None, None, mock_plugin) ] loader = StoryboardPluginLoader.make_test_instance( mock_extensions, namespace='storyboard.plugin.testing' ) test_list = list() loader.map(scheduler.add_plugins, test_list) self.assertTrue(test_list.index(mock_plugin_name) == 0) self.assertIsNotNone(scheduler.SCHEDULER.get_job(mock_plugin_name)) scheduler.shutdown_scheduler() self.assertIsNone(scheduler.SCHEDULER) CONF.clear_override('enable', 'scheduler')
def test_add_plugins_reschedule(self): """Assert that the test_add_plugins will reschedule existing plugins. """ CONF.set_override('enable', True, 'scheduler') self.assertIsNone(scheduler.SCHEDULER) scheduler.initialize_scheduler() mock_plugin = MockPlugin(dict()) mock_plugin_name = mock_plugin.get_name() mock_extensions = [ Extension(mock_plugin_name, None, None, mock_plugin) ] loader = StoryboardPluginLoader.make_test_instance( mock_extensions, namespace='storyboard.plugin.testing' ) test_list = list() loader.map(scheduler.add_plugins, test_list) self.assertTrue(test_list.index(mock_plugin_name) == 0) first_run_job = scheduler.SCHEDULER.get_job(mock_plugin_name) first_run_trigger = first_run_job.trigger self.assertEqual(mock_plugin._trigger.run_date, first_run_trigger.run_date) # Update the plugin's interval and re-run new_date = datetime.datetime.now() + datetime.timedelta(days=2) mock_plugin._trigger = DateTrigger(run_date=new_date) test_list = list() loader.map(scheduler.add_plugins, test_list) # make sure the plugin is only loaded once. self.assertTrue(test_list.index(mock_plugin_name) == 0) self.assertEquals(len(test_list), 1) # Get the job. second_run_job = scheduler.SCHEDULER.get_job(mock_plugin_name) second_run_trigger = second_run_job.trigger self.assertNotEqual(second_run_trigger.run_date, first_run_trigger.run_date) scheduler.shutdown_scheduler() self.assertIsNone(scheduler.SCHEDULER) CONF.clear_override('enable', 'scheduler')