def test_create_transfer_manager(self): client = object() config = TransferConfig() osutil = OSUtils() with mock.patch('boto3.s3.transfer.TransferManager') as manager: create_transfer_manager(client, config, osutil) assert manager.call_args == mock.call(client, config, osutil, None)
def test_open_file_chunk_reader(self): with mock.patch('boto3.s3.transfer.ReadFileChunk') as m: OSUtils().open_file_chunk_reader('myfile', 0, 100, None) m.from_filename.assert_called_with('myfile', 0, 100, None, enable_callback=False)
def test_get_file_size(self): with mock.patch('os.path.getsize') as m: OSUtils().get_file_size('myfile') m.assert_called_with('myfile')
def test_rename_file(self): with mock.patch('boto3.compat.rename_file') as rename_file: OSUtils().rename_file('foo', 'newfoo') rename_file.assert_called_with('foo', 'newfoo')
def test_remove_file_proxies_remove_file(self): with mock.patch('os.remove') as remove: OSUtils().remove_file('foo') remove.assert_called_with('foo')
def test_remove_file_ignores_errors(self): with mock.patch('os.remove') as remove: remove.side_effect = OSError('fake error') OSUtils().remove_file('foo') remove.assert_called_with('foo')
def test_open_file(self): fileobj = OSUtils().open(os.path.join(self.tempdir, 'foo'), 'w') self.assertTrue(hasattr(fileobj, 'write'))
def test_can_create_with_extra_configurations(self): transfer = S3Transfer(client=mock.Mock(), config=TransferConfig(), osutil=OSUtils()) self.assertIsInstance(transfer, S3Transfer)