def test_process_lifecycle(self): step = PublishStep('parent') step.process = Mock() child_step = PublishStep('child') 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_publish_exception_still_removes_working_dir(self, mock_rmtree): step = PublishStep("foo") work_dir = os.path.join(self.working_dir, 'foo') step.working_dir = work_dir step.process_lifecycle = Mock(side_effect=Exception('foo')) step._build_final_report = Mock() self.assertRaises(Exception, step.publish) self.assertTrue(step.process_lifecycle.called) self.assertFalse(step._build_final_report.called) mock_rmtree.assert_called_once_with(work_dir, ignore_errors=True)
def test_publish(self, mock_rmtree): step = PublishStep("foo") work_dir = os.path.join(self.working_dir, 'foo') step.working_dir = work_dir step.process_lifecycle = Mock() step._build_final_report = Mock() step.publish() self.assertTrue(step.process_lifecycle.called) self.assertTrue(step._build_final_report.called) mock_rmtree.assert_called_once_with(work_dir, ignore_errors=True)
def test_publish(self): # just test that process_lifecycle got called, that is where the functionality lives now step = PublishStep("foo") step.process_lifecycle = Mock() step.publish() self.assertTrue(step.process_lifecycle.called)