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)
