def setUp(self):
super(ImporterScratchPadMixinTests, self).setUp()
mock_plugins.install()
self.repo_manager = RepoManager()
self.importer_manager = RepoImporterManager()
self.repo_id = 'repo-1'
self.repo_manager.create_repo(self.repo_id)
self.conduit = ImporterScratchPadMixin(self.repo_id, 'test-importer')
def setUp(self, mock_get_working_directory):
super(DistributorScratchpadMixinTests, self).setUp()
mock_plugins.install()
self.repo_manager = RepoManager()
self.distributor_manager = RepoDistributorManager()
repo_id = 'repo-1'
self.repo_manager.create_repo(repo_id)
self.distributor_manager.add_distributor(repo_id, 'mock-distributor', {}, True,
distributor_id='test-distributor')
self.conduit = mixins.DistributorScratchPadMixin(repo_id, 'test-distributor')
def migrate(*args, **kwargs):
"""
Regenerates the 'content_unit_counts' attribute of each repository, and
removes the obsolete attribute 'content_unit_count'. The normal use case
will be that the 'content_unit_counts' attribute does not yet exist, but
this migration is idempotent just in case.
"""
RepoManager().rebuild_content_unit_counts()
repo_collection = Repo.get_collection()
repo_collection.update({}, {'$unset': {
'content_unit_count': 1
}},
safe=True)
def setUp(self):
super(ImporterScratchPadMixinTests, self).setUp()
mock_plugins.install()
self.repo_manager = RepoManager()
self.importer_manager = RepoImporterManager()
self.repo_id = 'repo-1'
self.repo_manager.create_repo(self.repo_id)
self.conduit = mixins.ImporterScratchPadMixin(self.repo_id, 'test-importer')
manager_factory.initialize()
self.importer_id = 'isp-importer'
self.mixin = mixins.ImporterScratchPadMixin(self.repo_id, self.importer_id)
