def test_delete_with_plugins(self):
"""
Tests that deleting a repo that has importers and distributors configured deletes them as
well.
"""
# Setup
self.manager.create_repo('doomed')
importer_manager = manager_factory.repo_importer_manager()
distributor_manager = manager_factory.repo_distributor_manager()
importer_manager.set_importer('doomed', 'mock-importer', {})
distributor_manager.add_distributor('doomed', 'mock-distributor', {}, True,
distributor_id='dist-1')
distributor_manager.add_distributor('doomed', 'mock-distributor', {}, True,
distributor_id='dist-2')
self.assertEqual(1, len(list(RepoImporter.get_collection().find({'repo_id': 'doomed'}))))
self.assertEqual(2, len(list(RepoDistributor.get_collection().find({'repo_id': 'doomed'}))))
# Test
self.manager.delete_repo('doomed')
# Verify
self.assertEqual(0, len(list(Repo.get_collection().find())))
self.assertEqual(0, len(list(RepoImporter.get_collection().find({'repo_id': 'doomed'}))))
self.assertEqual(0, len(list(RepoDistributor.get_collection().find({'repo_id': 'doomed'}))))
self.assertEqual(1, mock_plugins.MOCK_IMPORTER.importer_removed.call_count)
self.assertEqual(2, mock_plugins.MOCK_DISTRIBUTOR.distributor_removed.call_count)
repo_working_dir = common_utils.repository_working_dir('doomed', mkdir=False)
self.assertTrue(not os.path.exists(repo_working_dir))
def test_delete_with_plugins(self):
"""
Tests that deleting a repo that has importers and distributors configured deletes them as well.
"""
# Setup
self.manager.create_repo('doomed')
importer_manager = manager_factory.repo_importer_manager()
distributor_manager = manager_factory.repo_distributor_manager()
importer_manager.set_importer('doomed', 'mock-importer', {})
distributor_manager.add_distributor('doomed', 'mock-distributor', {}, True, distributor_id='dist-1')
distributor_manager.add_distributor('doomed', 'mock-distributor', {}, True, distributor_id='dist-2')
self.assertEqual(1, len(list(RepoImporter.get_collection().find({'repo_id' : 'doomed'}))))
self.assertEqual(2, len(list(RepoDistributor.get_collection().find({'repo_id' : 'doomed'}))))
# Test
self.manager.delete_repo('doomed')
# Verify
self.assertEqual(0, len(list(Repo.get_collection().find())))
self.assertEqual(0, len(list(RepoImporter.get_collection().find({'repo_id' : 'doomed'}))))
self.assertEqual(0, len(list(RepoDistributor.get_collection().find({'repo_id' : 'doomed'}))))
self.assertEqual(1, mock_plugins.MOCK_IMPORTER.importer_removed.call_count)
self.assertEqual(2, mock_plugins.MOCK_DISTRIBUTOR.distributor_removed.call_count)
repo_working_dir = common_utils.repository_working_dir('doomed', mkdir=False)
self.assertTrue(not os.path.exists(repo_working_dir))
repo_importer['importer_type_id'], repo_id))
error_tuples.append(e)
# Inform all distributors
distributor_coll = RepoDistributor.get_collection()
repo_distributors = list(distributor_coll.find({'repo_id' : repo_id}))
for repo_distributor in repo_distributors:
try:
distributor_manager.remove_distributor(repo_id, repo_distributor['id'])
except Exception, e:
logger.exception('Error received removing distributor [%s] from repo [%s]' % (
repo_distributor['id'], repo_id))
error_tuples.append(e)
# Delete the repository working directory
repo_working_dir = common_utils.repository_working_dir(repo_id, mkdir=False)
if os.path.exists(repo_working_dir):
try:
shutil.rmtree(repo_working_dir)
except Exception, e:
logger.exception('Error while deleting repo working dir [%s] for repo [%s]' % (
repo_working_dir, repo_id))
error_tuples.append(e)
# Database Updates
try:
Repo.get_collection().remove({'id' : repo_id}, safe=True)
# Remove all importers and distributors from the repo
# This is likely already done by the calls to other methods in
# this manager, but in case those failed we still want to attempt
# Inform all distributors
distributor_coll = RepoDistributor.get_collection()
repo_distributors = list(distributor_coll.find({'repo_id': repo_id}))
for repo_distributor in repo_distributors:
try:
distributor_manager.remove_distributor(repo_id,
repo_distributor['id'])
except Exception, e:
_LOG.exception(
'Error received removing distributor [%s] from repo [%s]' %
(repo_distributor['id'], repo_id))
error_tuples.append((_('Distributor Delete Error'), e.args))
# Delete the repository working directory
repo_working_dir = common_utils.repository_working_dir(repo_id,
mkdir=False)
if os.path.exists(repo_working_dir):
try:
shutil.rmtree(repo_working_dir)
except Exception, e:
_LOG.exception(
'Error while deleting repo working dir [%s] for repo [%s]'
% (repo_working_dir, repo_id))
error_tuples.append((_('Filesystem Cleanup Error'), e.args))
# Database Updates
try:
Repo.get_collection().remove({'id': repo_id}, safe=True)
# Remove all importers and distributors from the repo
# This is likely already done by the calls to other methods in