def testWriteComponentToParallelUploadTrackerFile(self):
tracker_file_lock = parallelism_framework_util.CreateLock()
fpath = self.CreateTempFile(file_name='foo')
random_prefix = '123'
enc_key = '456'
objects = [ObjectFromTracker('obj1', '42'),
ObjectFromTracker('obj2', '314159')]
WriteParallelUploadTrackerFile(fpath, random_prefix, objects,
encryption_key_sha256=enc_key)
new_object = ObjectFromTracker('obj3', '43')
try:
WriteComponentToParallelUploadTrackerFile(
fpath, tracker_file_lock, new_object, self.logger,
encryption_key_sha256=None)
self.fail('Expected CommandException due to different encryption key')
except CommandException as e:
self.assertIn('does not match encryption key', str(e))
WriteComponentToParallelUploadTrackerFile(
fpath, tracker_file_lock, new_object, self.logger,
encryption_key_sha256='456')
(actual_key, actual_prefix, actual_objects) = ReadParallelUploadTrackerFile(
fpath, self.logger)
self.assertEqual(enc_key, actual_key)
self.assertEqual(random_prefix, actual_prefix)
self.assertEqual(objects + [new_object], actual_objects)
class TestTrackerFile(GsUtilUnitTestCase):
"""Unit tests for parallel upload functions in cp command."""
def test_HashFilename(self):
# Tests that _HashFilename function works for both string and unicode
# filenames (without raising any Unicode encode/decode errors).
_HashFilename('file1')
_HashFilename(u'file1')
def test_RewriteTrackerFile(self):
"""Tests Rewrite tracker file functions."""
tracker_file_name = GetRewriteTrackerFilePath('bk1', 'obj1', 'bk2',
'obj2', self.test_api)
# Should succeed regardless of whether it exists.
DeleteTrackerFile(tracker_file_name)
src_obj_metadata = apitools_messages.Object(bucket='bk1',
name='obj1',
etag='etag1',
md5Hash='12345')
src_obj2_metadata = apitools_messages.Object(bucket='bk1',
name='obj1',
etag='etag2',
md5Hash='67890')
dst_obj_metadata = apitools_messages.Object(bucket='bk2', name='obj2')
rewrite_token = 'token1'
self.assertIsNone(
ReadRewriteTrackerFile(tracker_file_name, src_obj_metadata))
rewrite_params_hash = HashRewriteParameters(src_obj_metadata,
dst_obj_metadata, 'full')
WriteRewriteTrackerFile(tracker_file_name, rewrite_params_hash,
rewrite_token)
self.assertEqual(
ReadRewriteTrackerFile(tracker_file_name, rewrite_params_hash),
rewrite_token)
# Tracker file for an updated source object (with non-matching etag/md5)
# should return None.
rewrite_params_hash2 = HashRewriteParameters(src_obj2_metadata,
dst_obj_metadata, 'full')
self.assertIsNone(
ReadRewriteTrackerFile(tracker_file_name, rewrite_params_hash2))
DeleteTrackerFile(tracker_file_name)
def testReadGsutil416ParallelUploadTrackerFile(self):
"""Tests the parallel upload tracker file format prior to gsutil 4.17."""
random_prefix = '123'
objects = ['obj1', '42', 'obj2', '314159']
contents = '\n'.join([random_prefix] + objects) + '\n'
fpath = self.CreateTempFile(file_name='foo', contents=contents)
expected_objects = [
ObjectFromTracker(objects[2 * i], objects[2 * i + 1])
for i in range(0,
len(objects) / 2)
]
(_, actual_prefix,
actual_objects) = ReadParallelUploadTrackerFile(fpath, self.logger)
self.assertEqual(random_prefix, actual_prefix)
self.assertEqual(expected_objects, actual_objects)
def testReadEmptyGsutil416ParallelUploadTrackerFile(self):
"""Tests reading an empty pre-gsutil 4.17 parallel upload tracker file."""
fpath = self.CreateTempFile(file_name='foo', contents='')
(_, actual_prefix,
actual_objects) = ReadParallelUploadTrackerFile(fpath, self.logger)
self.assertEqual(None, actual_prefix)
self.assertEqual([], actual_objects)
def testParallelUploadTrackerFileNoEncryption(self):
fpath = self.CreateTempFile(file_name='foo')
random_prefix = '123'
objects = [
ObjectFromTracker('obj1', '42'),
ObjectFromTracker('obj2', '314159')
]
WriteParallelUploadTrackerFile(fpath, random_prefix, objects)
(enc_key, actual_prefix,
actual_objects) = ReadParallelUploadTrackerFile(fpath, self.logger)
self.assertEqual(random_prefix, actual_prefix)
self.assertEqual(None, enc_key)
self.assertEqual(objects, actual_objects)
def testParallelUploadTrackerFileWithEncryption(self):
fpath = self.CreateTempFile(file_name='foo')
random_prefix = '123'
enc_key = '456'
objects = [
ObjectFromTracker('obj1', '42'),
ObjectFromTracker('obj2', '314159')
]
WriteParallelUploadTrackerFile(fpath,
random_prefix,
objects,
encryption_key_sha256=enc_key)
(actual_key, actual_prefix,
actual_objects) = ReadParallelUploadTrackerFile(fpath, self.logger)
self.assertEqual(enc_key, actual_key)
self.assertEqual(random_prefix, actual_prefix)
self.assertEqual(objects, actual_objects)
def testWriteComponentToParallelUploadTrackerFile(self):
tracker_file_lock = parallelism_framework_util.CreateLock()
fpath = self.CreateTempFile(file_name='foo')
random_prefix = '123'
enc_key = '456'
objects = [
ObjectFromTracker('obj1', '42'),
ObjectFromTracker('obj2', '314159')
]
WriteParallelUploadTrackerFile(fpath,
random_prefix,
objects,
encryption_key_sha256=enc_key)
new_object = ObjectFromTracker('obj3', '43')
try:
WriteComponentToParallelUploadTrackerFile(
fpath,
tracker_file_lock,
new_object,
self.logger,
encryption_key_sha256=None)
self.fail(
'Expected CommandException due to different encryption key')
except CommandException, e:
self.assertIn('does not match encryption key', str(e))
WriteComponentToParallelUploadTrackerFile(fpath,
tracker_file_lock,
new_object,
self.logger,
encryption_key_sha256='456')
(actual_key, actual_prefix,
actual_objects) = ReadParallelUploadTrackerFile(fpath, self.logger)
self.assertEqual(enc_key, actual_key)
self.assertEqual(random_prefix, actual_prefix)
self.assertEqual(objects + [new_object], actual_objects)