def test_process_step_failure_reported_on_metadata_finalized(self, mock_get_units):
self.pluginstep.repo.content_unit_counts = {'FOO_TYPE': 1}
mock_get_units.return_value = ['mock_unit']
step = PluginStep('foo_step')
step.parent = self.pluginstep
step.finalize = Mock(side_effect=Exception())
self.assertRaises(Exception, step.process)
self.assertEquals(step.state, reporting_constants.STATE_FAILED)
self.assertEquals(step.progress_successes, 1)
self.assertEquals(step.progress_failures, 1)
self.assertEquals(step.total_units, 1)
def test_record_failure(self):
plugin_step = PluginStep('foo_step')
plugin_step.parent = self.pluginstep
error_msg = 'Too bad, so sad'
try:
raise Exception(error_msg)
except Exception, e:
tb = sys.exc_info()[2]
plugin_step._record_failure(e, tb)
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)
def test_get_conduit_from_parent(self):
step = PluginStep('foo_step')
step.conduit = 'foo'
step.parent = Mock()
step.parent.get_conduit.return_value = 'foo'
self.assertEquals('foo', step.get_conduit())
