def test__cleanup_direct(self, mock_rmtree, mock_mkdtemp):
self.setup_mkdtemp(mock_mkdtemp)
loader = BespinWorkflowLoader(self.direct_workflow_version)
loader._cleanup()
self.assertFalse(mock_rmtree.called)
def test__cleanup_zipped(self, mock_rmtree, mock_mkdtemp):
self.setup_mkdtemp(mock_mkdtemp)
loader = BespinWorkflowLoader(self.zipped_workflow_version)
loader._cleanup()
self.assertEqual(mock_rmtree.call_args, call(loader.download_dir))