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_get_working_dir_already_calculated(self): step = PublishStep('foo_step') step.working_dir = 'foo' self.assertEquals('foo', step.get_working_dir())