def test_process_lifecycle_reports_on_error(self):
# set working_dir and conduit. This is required by process_lifecycle
step = PluginStep('parent', working_dir=self.working_dir, conduit=self.conduit)
step.process = Mock(side_effect=Exception('Foo'))
step.report_progress = Mock()
self.assertRaises(Exception, step.process_lifecycle)
step.report_progress.assert_called_once_with(force=True)
def test_process_lifecycle(self):
# set working_dir and conduit. This is required by process_lifecycle
step = PluginStep('parent', working_dir=self.working_dir, conduit=self.conduit)
step.process = Mock()
child_step = PluginStep('child', working_dir=self.working_dir, conduit=self.conduit)
child_step.process = Mock()
step.add_child(child_step)
step.report_progress = Mock()
step.process_lifecycle()
step.process.assert_called_once_with()
child_step.process.assert_called_once_with()
step.report_progress.assert_called_once_with(force=True)
def test_report_progress(self):
plugin_step = PluginStep('foo_step')
plugin_step.parent = Mock()
plugin_step.report_progress()
plugin_step.parent.report_progress.assert_called_once_with(False)
