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
