def setUp(self):
self.local_file = os.path.abspath(".") + os.sep + "some_directory" + os.sep + "text1.txt"
self.local_dir = os.path.abspath(".") + os.sep + "some_directory" + os.sep
self.session = FakeSession()
self.service = self.session.get_service("s3")
self.endpoint = self.service.get_endpoint("us-east-1")
self.files = make_loc_files()
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.loc_files = make_loc_files()
self.bucket = make_s3_files(self.session)
parser = argparse.ArgumentParser()
parser.add_argument('--region')
self.parsed_globals = parser.parse_args(['--region', 'us-east-1'])
def setUp(self):
self.local_file = os.path.abspath('.') + os.sep + 'some_directory' \
+ os.sep + 'text1.txt'
self.local_dir = os.path.abspath('.') + os.sep + 'some_directory' \
+ os.sep
self.session = FakeSession()
self.files = make_loc_files()
def setUp(self):
super(S3HandlerTestDeleteList, self).setUp()
self.session = FakeSession()
params = {'region': 'us-east-1'}
self.s3_handler = S3Handler(self.session, params)
self.bucket = make_s3_files(self.session)
self.loc_files = make_loc_files()
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.service = self.session.get_service("s3")
self.endpoint = self.service.get_endpoint("us-east-1")
params = {"region": "us-east-1"}
self.s3_handler = S3Handler(self.session, params)
self.bucket = make_s3_files(self.session)
self.loc_files = make_loc_files()
def setUp(self):
self.environ = {}
self.environ_patch = patch('os.environ', self.environ)
self.environ_patch.start()
self.mock = MagicMock()
self.mock.get_config = MagicMock(return_value={'region': None})
self.loc_files = make_loc_files()
self.bucket = 's3testbucket'
def setUp(self):
super(S3HandlerTestDelete, self).setUp()
params = {'region': 'us-east-1'}
self.s3_handler = S3Handler(self.session, params,
runtime_config=runtime_config(
max_concurrent_requests=1))
self.loc_files = make_loc_files(self.file_creator)
self.bucket = 'mybucket'
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
params = {'region': 'us-east-1'}
self.s3_handler = S3Handler(self.session, params)
self.bucket = make_s3_files(self.session)
self.loc_files = make_loc_files()
def setUp(self):
super(CommandArchitectureTest, self).setUp()
self.session = FakeSession()
self.bucket = make_s3_files(self.session)
self.loc_files = make_loc_files()
self.output = StringIO()
self.saved_stdout = sys.stdout
sys.stdout = self.output
def setUp(self):
self.environ = {}
self.environ_patch = patch("os.environ", self.environ)
self.environ_patch.start()
self.session = FakeSession()
self.mock = MagicMock()
self.mock.get_config = MagicMock(return_value={"region": None})
self.loc_files = make_loc_files()
self.bucket = make_s3_files(self.session)
def setUp(self):
super(S3HandlerTestDeleteList, self).setUp()
self.session = FakeSession()
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
params = {'region': 'us-east-1'}
self.s3_handler = S3Handler(self.session, params)
self.bucket = make_s3_files(self.session)
self.loc_files = make_loc_files()
def setUp(self):
super(S3HandlerTestMvLocalS3, self).setUp()
self.session = FakeSession()
params = {'region': 'us-east-1', 'acl': ['private']}
self.s3_handler = S3Handler(self.session, params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
self.local_file = six.text_type(os.path.abspath('.') + os.sep + 'some_directory' \
+ os.sep + 'text1.txt')
self.local_dir = six.text_type(os.path.abspath('.') + os.sep + 'some_directory' \
+ os.sep)
self.session = FakeSession()
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
self.files = make_loc_files()
def setUp(self):
super(S3HandlerExceptionSingleTaskTest, self).setUp()
self.session = FakeSession(True, True)
params = {'region': 'us-east-1'}
self.s3_handler = S3Handler(self.session, params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.client = self.session.create_client('s3', 'us-west-2')
self.source_client = self.session.create_client('s3', 'us-west-2')
params = {'region': 'us-west-2'}
self.s3_handler = S3Handler(self.session, params)
self.bucket = make_s3_files(self.session)
self.file_creator = FileCreator()
self.loc_files = make_loc_files(self.file_creator)
def setUp(self):
super(S3HandlerTestMvLocalS3, self).setUp()
params = {'region': 'us-east-1', 'acl': ['private'], 'quiet': True}
self.s3_handler = S3Handler(self.session, params,
runtime_config=runtime_config(
max_concurrent_requests=1))
self.bucket = 'mybucket'
self.loc_files = make_loc_files(self.file_creator)
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
super(S3HandlerExceptionMultiTaskTest, self).setUp()
self.session = FakeSession(True, True)
params = {'region': 'us-east-1'}
self.s3_handler_multi = S3Handler(self.session, params,
multi_threshold=10, chunksize=2)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
super(S3HandlerTestMvLocalS3, self).setUp()
self.session = FakeSession()
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
params = {'region': 'us-east-1', 'acl': ['private'], 'quiet': True}
self.s3_handler = S3Handler(self.session, params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
super(CommandArchitectureTest, self).setUp()
self.session = self.driver.session
self.bucket = 'mybucket'
self.file_creator = FileCreator()
self.loc_files = make_loc_files(self.file_creator)
self.output = StringIO()
self.err_output = StringIO()
self.saved_stdout = sys.stdout
self.saved_stderr = sys.stderr
sys.stdout = self.output
sys.stderr = self.err_output
def setUp(self):
super(S3HandlerTestUpload, self).setUp()
self.session = FakeSession()
params = {'region': 'us-east-1', 'acl': ['private']}
self.s3_handler = S3Handler(self.session, params)
self.s3_handler_multi = S3Handler(self.session, multi_threshold=10,
chunksize=2,
params=params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.service = self.session.get_service("s3")
self.endpoint = self.service.get_endpoint("us-east-1")
params = {"region": "us-east-1", "acl": ["private"]}
self.s3_handler = S3Handler(self.session, params)
self.s3_handler_multi = S3Handler(self.session, multi_threshold=10, chunksize=2, params=params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + "/text1.txt", self.bucket + "/another_directory/text2.txt"]
self.output = StringIO()
self.saved_stdout = sys.stdout
sys.stdout = self.output
def setUp(self):
super(S3HandlerExceptionMultiTaskTest, self).setUp()
self.session = FakeSession(True, True)
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
params = {'region': 'us-east-1', 'quiet': True}
self.s3_handler_multi = S3Handler(
self.session, params,
runtime_config=runtime_config(
multipart_threshold=10, multipart_chunksize=2))
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
params = {'region': 'us-east-1', 'acl': ['private']}
self.s3_handler = S3Handler(self.session, params)
self.s3_handler_multi = S3Handler(self.session, multi_threshold=10,
chunksize=2,
params=params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
self.output = StringIO()
self.saved_stdout = sys.stdout
sys.stdout = self.output
def setUp(self):
super(S3HandlerTestUpload, self).setUp()
params = {'region': 'us-east-1', 'acl': 'private', 'quiet': True}
self.s3_handler = S3Handler(
self.session, params, runtime_config=runtime_config(
max_concurrent_requests=1))
self.s3_handler_multi = S3Handler(
self.session, params=params,
runtime_config=runtime_config(
multipart_threshold=10, multipart_chunksize=10,
max_concurrent_requests=1))
self.bucket = 'mybucket'
self.loc_files = make_loc_files(self.file_creator)
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
def setUp(self):
super(S3HandlerExceptionMultiTaskTest, self).setUp()
self.session = FakeSession(True, True)
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
params = {'region': 'us-east-1'}
self.s3_handler_multi = S3Handler(self.session,
params,
multi_threshold=10,
chunksize=2)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [
self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt'
]
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
params = {'region': 'us-east-1', 'acl': ['private']}
self.s3_handler = S3Handler(self.session, params)
self.s3_handler_multi = S3Handler(self.session, multi_threshold=10,
chunksize=2,
params=params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
self.output = StringIO()
self.saved_stdout = sys.stdout
sys.stdout = self.output
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.client = self.session.create_client('s3', 'us-west-2')
params = {'region': 'us-west-2', 'acl': ['private']}
self.s3_handler = S3Handler(self.session, params)
self.s3_handler_multi = S3Handler(
self.session, params=params,
runtime_config=runtime_config(
multipart_threshold=10, multipart_chunksize=2))
self.bucket = create_bucket(self.session)
self.file_creator = FileCreator()
self.loc_files = make_loc_files(self.file_creator)
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
self.output = StringIO()
self.saved_stderr = sys.stderr
sys.stderr = self.output
def setUp(self):
super(S3HandlerTestUpload, self).setUp()
self.session = FakeSession()
self.service = self.session.get_service('s3')
self.endpoint = self.service.get_endpoint('us-east-1')
params = {'region': 'us-east-1', 'acl': ['private']}
self.s3_handler = S3Handler(self.session, params)
self.s3_handler_multi = S3Handler(self.session,
multi_threshold=10,
chunksize=2,
params=params)
self.bucket = create_bucket(self.session)
self.loc_files = make_loc_files()
self.s3_files = [
self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt'
]
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
self.client = self.session.create_client('s3', 'us-west-2')
params = {'region': 'us-west-2', 'acl': ['private']}
self.s3_handler = S3Handler(self.session, params)
self.s3_handler_multi = S3Handler(
self.session, params=params,
runtime_config=runtime_config(
multipart_threshold=10, multipart_chunksize=2))
self.bucket = create_bucket(self.session)
self.file_creator = FileCreator()
self.loc_files = make_loc_files(self.file_creator)
self.s3_files = [self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt']
self.output = StringIO()
self.saved_stderr = sys.stderr
sys.stderr = self.output
def setUp(self):
super(S3HandlerTestUpload, self).setUp()
params = {'region': 'us-east-1', 'acl': 'private', 'quiet': True}
self.s3_handler = S3Handler(
self.session,
params,
runtime_config=runtime_config(max_concurrent_requests=1))
self.s3_handler_multi = S3Handler(self.session,
params=params,
runtime_config=runtime_config(
multipart_threshold=10,
multipart_chunksize=10,
max_concurrent_requests=1))
self.bucket = 'mybucket'
self.loc_files = make_loc_files(self.file_creator)
self.s3_files = [
self.bucket + '/text1.txt',
self.bucket + '/another_directory/text2.txt'
]
def test_upload(self):
self.loc_files = make_loc_files(self.file_creator)
# Confirm there are no objects in the bucket.
response = self.client.list_objects(Bucket=self.bucket)
self.assertEqual(len(response.get('Contents', [])), 0)
# Create file info objects to perform upload.
files = [self.loc_files[0], self.loc_files[1]]
tasks = []
for i in range(len(files)):
tasks.append(FileInfo(
src=self.loc_files[i],
dest=self.s3_files[i],
operation_name='upload', size=0,
client=self.client,
))
# Perform the upload.
self.s3_handler.call(tasks)
# Confirm the files were uploaded.
response = self.client.list_objects(Bucket=self.bucket)
self.assertEqual(len(response.get('Contents', [])), 2)
def test_upload(self):
self.loc_files = make_loc_files(self.file_creator)
# Confirm there are no objects in the bucket.
response = self.client.list_objects(Bucket=self.bucket)
self.assertEqual(len(response.get('Contents', [])), 0)
# Create file info objects to perform upload.
files = [self.loc_files[0], self.loc_files[1]]
tasks = []
for i in range(len(files)):
tasks.append(FileInfo(
src=self.loc_files[i],
dest=self.s3_files[i],
operation_name='upload', size=0,
client=self.client,
))
# Perform the upload.
self.s3_handler.call(tasks)
# Confirm the files were uploaded.
response = self.client.list_objects(Bucket=self.bucket)
self.assertEqual(len(response.get('Contents', [])), 2)
def test_multi_upload(self):
self.loc_files = make_loc_files(self.file_creator, self.threshold+1)
files = [self.loc_files[0], self.loc_files[1]]
tasks = []
for i in range(len(files)):
tasks.append(FileInfo(
src=self.loc_files[i],
dest=self.s3_files[i],
size=self.threshold+1,
operation_name='upload',
client=self.client,
))
self.s3_handler_multi.call(tasks)
# Confirm UploadPart was called
self.assertIn("Completed 4 of 4 part(s)", self.output.getvalue())
# Confirm the files were uploaded.
response = self.client.list_objects(Bucket=self.bucket)
self.assertEqual(len(response.get('Contents', [])), 2)
def test_multi_upload(self):
self.loc_files = make_loc_files(self.file_creator, self.threshold+1)
files = [self.loc_files[0], self.loc_files[1]]
tasks = []
for i in range(len(files)):
tasks.append(FileInfo(
src=self.loc_files[i],
dest=self.s3_files[i],
size=self.threshold+1,
operation_name='upload',
client=self.client,
))
self.s3_handler_multi.call(tasks)
# Confirm UploadPart was called
self.assertIn("Completed 4 of 4 part(s)", self.output.getvalue())
# Confirm the files were uploaded.
response = self.client.list_objects(Bucket=self.bucket)
self.assertEqual(len(response.get('Contents', [])), 2)
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
params = {'region': 'us-east-1'}
self.s3_handler = S3Handler(self.session, params)
self.bucket = make_s3_files(self.session)
self.loc_files = make_loc_files()
def setUp(self):
self.session = botocore.session.get_session(EnvironmentVariables)
params = {'region': 'us-east-1'}
self.s3_handler = S3Handler(self.session, params)
self.bucket = make_s3_files(self.session, key1='a+b/foo', key2=None)
self.loc_files = make_loc_files()
def setUp(self):
self.client = None
self.file_creator = FileCreator()
self.files = make_loc_files(self.file_creator)
self.local_file = self.files[0]
self.local_dir = self.files[3] + os.sep
def setUp(self):
self.client = None
self.file_creator = FileCreator()
self.files = make_loc_files(self.file_creator)
self.local_file = self.files[0]
self.local_dir = self.files[3] + os.sep
