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)