def _upload_manager(context): """ Instantiates and configures the upload manager. The context is used to access any necessary configuration. :return: initialized and ready to run upload manager instance :rtype: UploadManager """ upload_working_dir = os.path.join( context.config['filesystem']['upload_working_dir'], UPLOAD_SUBDIR) upload_working_dir = os.path.expanduser(upload_working_dir) chunk_size = int(context.config['server']['upload_chunk_size']) upload_manager = upload_lib.UploadManager(upload_working_dir, context.server, chunk_size) upload_manager.initialize() return upload_manager
def _upload_manager(context): """ Instantiates and configures the upload manager. The context is used to access any necessary configuration. :return: initialized and ready to run upload manager instance :rtype: UploadManager """ # Each upload_manager needs to be associated with a unique upload working directory. # Create a subdirectory for rpm uploads under the main upload_working_dir # to avoid interference with other types of uploads eg. iso uploads. upload_working_dir = os.path.join(context.config['filesystem']['upload_working_dir'], RPM_UPLOAD_SUBDIR) upload_working_dir = os.path.expanduser(upload_working_dir) chunk_size = int(context.config['server']['upload_chunk_size']) upload_manager = upload_lib.UploadManager(upload_working_dir, context.server, chunk_size) upload_manager.initialize() return upload_manager
def setUp(self): super(UploadManagerTests, self).setUp() # Temporary location to store tracker files self.upload_working_dir = '/tmp/pulp-upload-manager-test' if os.path.exists(self.upload_working_dir): shutil.rmtree(self.upload_working_dir) # Recreate for each test to minimize necessary cleanup self.mock_bindings = mock.Mock() self.mock_upload_bindings = mock.Mock() self.mock_bindings.uploads = self.mock_upload_bindings self._mock_initialize_upload() self._mock_upload_segment() self._mock_delete_upload() self._mock_import_upload() # Create the manager to test self.upload_manager = upload_util.UploadManager(self.upload_working_dir, self.mock_bindings)
def _get_upload_manager(context): """ Return a new UploadManager. :param context: ClientContext containing the CLI instance being configured :type context: pulp.client.extensions.core.ClientContext :return: An intialized UploadManager. :rtype: pulp.client.upload.manager.UploadManager """ # Each upload_manager needs to be associated with a unique upload working directory. # Create a subdirectory for iso uploads under the main upload_working_dir # to avoid interference with other types of uploads eg. rpm uploads. upload_working_dir = os.path.join( context.config['filesystem']['upload_working_dir'], ISO_UPLOAD_SUBDIR) upload_working_dir = os.path.expanduser(upload_working_dir) chunk_size = int(context.config['server']['upload_chunk_size']) upload_manager = upload_lib.UploadManager(upload_working_dir, context.server, chunk_size) upload_manager.initialize() return upload_manager