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)