def __call__(self, client, result_queue):
"""Creates a S3TransferHandler instance
:type client: botocore.client.Client
:param client: The client to power the S3TransferHandler
:type result_queue: queue.Queue
:param result_queue: The result queue to be used to process results
for the S3TransferHandler
:returns: A S3TransferHandler instance
"""
transfer_config = create_transfer_config_from_runtime_config(
self._runtime_config)
transfer_config.max_in_memory_upload_chunks = self.MAX_IN_MEMORY_CHUNKS
transfer_config.max_in_memory_download_chunks = \
self.MAX_IN_MEMORY_CHUNKS
transfer_manager = TransferManager(client, transfer_config)
LOGGER.debug(
"Using a multipart threshold of %s and a part size of %s",
transfer_config.multipart_threshold,
transfer_config.multipart_chunksize
)
result_recorder = ResultRecorder()
result_processor_handlers = [result_recorder]
self._add_result_printer(result_recorder, result_processor_handlers)
result_processor = ResultProcessor(
result_queue, result_processor_handlers)
command_result_recorder = CommandResultRecorder(
result_queue, result_recorder, result_processor)
return S3TransferHandler(
transfer_manager, self._cli_params, command_result_recorder)
def setUp(self):
self.result_queue = queue.Queue()
self.result_recorder = ResultRecorder()
self.processed_results = []
self.result_processor = ResultProcessor(
self.result_queue,
[self.result_recorder, self.processed_results.append])
self.command_result_recorder = CommandResultRecorder(
self.result_queue, self.result_recorder, self.result_processor)
self.transfer_manager = mock.Mock(spec=TransferManager)
self.transfer_manager.__enter__ = mock.Mock()
self.transfer_manager.__exit__ = mock.Mock()
self.parameters = {}
self.s3_transfer_handler = S3TransferHandler(
self.transfer_manager, self.parameters,
self.command_result_recorder)