def test_process_step_failure_reported_on_metadata_finalized(self, mock_get_units):
self.publisher.repo.content_unit_counts = {'FOO_TYPE': 1}
mock_get_units.return_value = ['mock_unit']
step = PublishStep('foo_step')
step.parent = self.publisher
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):
publish_step = PublishStep('foo_step')
publish_step.parent = self.publisher
error_msg = 'Too bad, so sad'
try:
raise Exception(error_msg)
except Exception, e:
tb = sys.exc_info()[2]
publish_step._record_failure(e, tb)
def test_report_progress(self):
publish_step = PublishStep('foo_step')
publish_step.parent = Mock()
publish_step.report_progress()
publish_step.parent.report_progress.assert_called_once_with(False)
def test_get_conduit_from_parent(self):
step = PublishStep('foo_step')
step.conduit = 'foo'
step.parent = Mock()
step.parent.get_conduit.return_value = 'foo'
self.assertEquals('foo', step.get_conduit())
def test_get_distributor_type_from_parent(self):
step = PublishStep('foo_step')
step.conduit = 'foo'
step.parent = Mock()
step.parent.get_plugin_type.return_value = 'foo'
self.assertEquals('foo', step.get_distributor_type())
def test_get_repo_from_parent(self):
step = PublishStep('foo_step')
step.publish_conduit = 'foo'
step.parent = mock.Mock()
step.parent.get_repo.return_value = 'foo'
self.assertEquals('foo', step.get_repo())
