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())