def test_uses_provided_osutil(self):
osutil = RecordingOSUtils()
# Use the recording os utility for the transfer manager
self._manager = TransferManager(self.client, self.config, osutil)
self.add_put_object_response_with_default_expected_params()
future = self.manager.upload(self.filename, self.bucket, self.key)
future.result()
# The upload should have used the os utility. We check this by making
# sure that the recorded opens are as expected.
self.assertEqual(osutil.open_records, [(self.filename, 'rb')])
def test_uses_provided_osutil(self):
osutil = RecordingOSUtils()
# Use the recording os utility for the transfer manager
self._manager = TransferManager(self.client, self.config, osutil)
self.add_head_object_response()
self.add_successful_get_object_responses()
future = self.manager.download(**self.create_call_kwargs())
future.result()
# The osutil should have had its open() method invoked when opening
# a temporary file and its rename_file() method invoked when the
# the temporary file was moved to its final location.
self.assertEqual(len(osutil.open_records), 1)
self.assertEqual(len(osutil.rename_records), 1)