class TestCPCommand(BaseAWSCommandParamsTest):
prefix = 's3 cp '
def setUp(self):
super(TestCPCommand, self).setUp()
self.files = FileCreator()
def tearDown(self):
super(TestCPCommand, self).tearDown()
self.files.remove_all()
def test_operations_used_in_upload(self):
full_path = self.files.create_file('foo.txt', 'mycontent')
cmdline = '%s %s s3://bucket/key.txt' % (self.prefix, full_path)
self.parsed_responses = [{'ETag': '"c8afdb36c52cf4727836669019e69222"'}]
self.run_cmd(cmdline, expected_rc=0)
# The only operation we should have called is PutObject.
self.assertEqual(len(self.operations_called), 1, self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'PutObject')
def test_operations_used_in_download_file(self):
self.parsed_responses = [
{"ContentLength": "100", "LastModified": "00:00:00Z"},
{'ETag': '"foo-1"', 'Body': six.BytesIO(b'foo')},
]
cmdline = '%s s3://bucket/key.txt %s' % (self.prefix,
self.files.rootdir)
self.run_cmd(cmdline, expected_rc=0)
# The only operations we should have called are HeadObject/GetObject.
self.assertEqual(len(self.operations_called), 2, self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'HeadObject')
self.assertEqual(self.operations_called[1][0].name, 'GetObject')
def test_operations_used_in_recursive_download(self):
self.parsed_responses = [
{'ETag': '"foo-1"', 'Contents': [], 'CommonPrefixes': []},
]
cmdline = '%s s3://bucket/key.txt %s --recursive' % (
self.prefix, self.files.rootdir)
self.run_cmd(cmdline, expected_rc=0)
# We called ListObjects but had no objects to download, so
# we only have a single ListObjects operation being called.
self.assertEqual(len(self.operations_called), 1, self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'ListObjects')
class TestCPCommand(BaseAWSCommandParamsTest):
prefix = 's3 cp '
def setUp(self):
super(TestCPCommand, self).setUp()
self.files = FileCreator()
def tearDown(self):
super(TestCPCommand, self).tearDown()
self.files.remove_all()
def test_operations_used_in_upload(self):
full_path = self.files.create_file('foo.txt', 'mycontent')
cmdline = '%s %s s3://bucket/key.txt' % (self.prefix, full_path)
self.parsed_responses = [{
'ETag': '"c8afdb36c52cf4727836669019e69222"'
}]
self.run_cmd(cmdline, expected_rc=0)
# The only operation we should have called is PutObject.
self.assertEqual(len(self.operations_called), 1,
self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'PutObject')
def test_key_name_added_when_only_bucket_provided(self):
full_path = self.files.create_file('foo.txt', 'mycontent')
cmdline = '%s %s s3://bucket/' % (self.prefix, full_path)
self.parsed_responses = [{
'ETag': '"c8afdb36c52cf4727836669019e69222"'
}]
self.run_cmd(cmdline, expected_rc=0)
# The only operation we should have called is PutObject.
self.assertEqual(len(self.operations_called), 1,
self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'PutObject')
self.assertEqual(self.operations_called[0][1]['key'], 'foo.txt')
self.assertEqual(self.operations_called[0][1]['bucket'], 'bucket')
def test_trailing_slash_appended(self):
full_path = self.files.create_file('foo.txt', 'mycontent')
# Here we're saying s3://bucket instead of s3://bucket/
# This should still work the same as if we added the trailing slash.
cmdline = '%s %s s3://bucket' % (self.prefix, full_path)
self.parsed_responses = [{
'ETag': '"c8afdb36c52cf4727836669019e69222"'
}]
self.run_cmd(cmdline, expected_rc=0)
# The only operation we should have called is PutObject.
self.assertEqual(len(self.operations_called), 1,
self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'PutObject')
self.assertEqual(self.operations_called[0][1]['key'], 'foo.txt')
self.assertEqual(self.operations_called[0][1]['bucket'], 'bucket')
def test_operations_used_in_download_file(self):
self.parsed_responses = [
{
"ContentLength": "100",
"LastModified": "00:00:00Z"
},
{
'ETag': '"foo-1"',
'Body': six.BytesIO(b'foo')
},
]
cmdline = '%s s3://bucket/key.txt %s' % (self.prefix,
self.files.rootdir)
self.run_cmd(cmdline, expected_rc=0)
# The only operations we should have called are HeadObject/GetObject.
self.assertEqual(len(self.operations_called), 2,
self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'HeadObject')
self.assertEqual(self.operations_called[1][0].name, 'GetObject')
def test_operations_used_in_recursive_download(self):
self.parsed_responses = [
{
'ETag': '"foo-1"',
'Contents': [],
'CommonPrefixes': []
},
]
cmdline = '%s s3://bucket/key.txt %s --recursive' % (
self.prefix, self.files.rootdir)
self.run_cmd(cmdline, expected_rc=0)
# We called ListObjects but had no objects to download, so
# we only have a single ListObjects operation being called.
self.assertEqual(len(self.operations_called), 1,
self.operations_called)
self.assertEqual(self.operations_called[0][0].name, 'ListObjects')
