Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(36)

Unified Diff: tools/telemetry/third_party/gsutil/third_party/boto/tests/integration/gs/test_resumable_downloads.py

Issue 1260493004: Revert "Add gsutil 4.13 to telemetry/third_party" (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: tools/telemetry/third_party/gsutil/third_party/boto/tests/integration/gs/test_resumable_downloads.py
diff --git a/tools/telemetry/third_party/gsutil/third_party/boto/tests/integration/gs/test_resumable_downloads.py b/tools/telemetry/third_party/gsutil/third_party/boto/tests/integration/gs/test_resumable_downloads.py
deleted file mode 100644
index ba5d9830833e1284f7fed9c497304a175bb43e1b..0000000000000000000000000000000000000000
--- a/tools/telemetry/third_party/gsutil/third_party/boto/tests/integration/gs/test_resumable_downloads.py
+++ /dev/null
@@ -1,354 +0,0 @@
-# Copyright 2010 Google Inc.
-#
-# Permission is hereby granted, free of charge, to any person obtaining a
-# copy of this software and associated documentation files (the
-# "Software"), to deal in the Software without restriction, including
-# without limitation the rights to use, copy, modify, merge, publish, dis-
-# tribute, sublicense, and/or sell copies of the Software, and to permit
-# persons to whom the Software is furnished to do so, subject to the fol-
-# lowing conditions:
-#
-# The above copyright notice and this permission notice shall be included
-# in all copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
-# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
-# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
-# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
-# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
-# IN THE SOFTWARE.
-
-"""
-Tests of resumable downloads.
-"""
-
-import errno
-import os
-import re
-
-import boto
-from boto.s3.resumable_download_handler import get_cur_file_size
-from boto.s3.resumable_download_handler import ResumableDownloadHandler
-from boto.exception import ResumableTransferDisposition
-from boto.exception import ResumableDownloadException
-from cb_test_harness import CallbackTestHarness
-from tests.integration.gs.testcase import GSTestCase
-
-
-SMALL_KEY_SIZE = 2 * 1024 # 2 KB.
-LARGE_KEY_SIZE = 500 * 1024 # 500 KB.
-
-
-class ResumableDownloadTests(GSTestCase):
- """Resumable download test suite."""
-
- def make_small_key(self):
- small_src_key_as_string = os.urandom(SMALL_KEY_SIZE)
- small_src_key = self._MakeKey(data=small_src_key_as_string)
- return small_src_key_as_string, small_src_key
-
- def make_tracker_file(self, tmpdir=None):
- if not tmpdir:
- tmpdir = self._MakeTempDir()
- tracker_file = os.path.join(tmpdir, 'tracker')
- return tracker_file
-
- def make_dst_fp(self, tmpdir=None):
- if not tmpdir:
- tmpdir = self._MakeTempDir()
- dst_file = os.path.join(tmpdir, 'dstfile')
- return open(dst_file, 'w')
-
- def test_non_resumable_download(self):
- """
- Tests that non-resumable downloads work
- """
- dst_fp = self.make_dst_fp()
- small_src_key_as_string, small_src_key = self.make_small_key()
- small_src_key.get_contents_to_file(dst_fp)
- self.assertEqual(SMALL_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
-
- def test_download_without_persistent_tracker(self):
- """
- Tests a single resumable download, with no tracker persistence
- """
- res_download_handler = ResumableDownloadHandler()
- dst_fp = self.make_dst_fp()
- small_src_key_as_string, small_src_key = self.make_small_key()
- small_src_key.get_contents_to_file(
- dst_fp, res_download_handler=res_download_handler)
- self.assertEqual(SMALL_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
-
- def test_failed_download_with_persistent_tracker(self):
- """
- Tests that failed resumable download leaves a correct tracker file
- """
- harness = CallbackTestHarness()
- tmpdir = self._MakeTempDir()
- tracker_file_name = self.make_tracker_file(tmpdir)
- dst_fp = self.make_dst_fp(tmpdir)
- res_download_handler = ResumableDownloadHandler(
- tracker_file_name=tracker_file_name, num_retries=0)
- small_src_key_as_string, small_src_key = self.make_small_key()
- try:
- small_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- self.fail('Did not get expected ResumableDownloadException')
- except ResumableDownloadException, e:
- # We'll get a ResumableDownloadException at this point because
- # of CallbackTestHarness (above). Check that the tracker file was
- # created correctly.
- self.assertEqual(e.disposition,
- ResumableTransferDisposition.ABORT_CUR_PROCESS)
- self.assertTrue(os.path.exists(tracker_file_name))
- f = open(tracker_file_name)
- etag_line = f.readline()
- self.assertEquals(etag_line.rstrip('\n'), small_src_key.etag.strip('"\''))
-
- def test_retryable_exception_recovery(self):
- """
- Tests handling of a retryable exception
- """
- # Test one of the RETRYABLE_EXCEPTIONS.
- exception = ResumableDownloadHandler.RETRYABLE_EXCEPTIONS[0]
- harness = CallbackTestHarness(exception=exception)
- res_download_handler = ResumableDownloadHandler(num_retries=1)
- dst_fp = self.make_dst_fp()
- small_src_key_as_string, small_src_key = self.make_small_key()
- small_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- # Ensure downloaded object has correct content.
- self.assertEqual(SMALL_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
-
- def test_broken_pipe_recovery(self):
- """
- Tests handling of a Broken Pipe (which interacts with an httplib bug)
- """
- exception = IOError(errno.EPIPE, "Broken pipe")
- harness = CallbackTestHarness(exception=exception)
- res_download_handler = ResumableDownloadHandler(num_retries=1)
- dst_fp = self.make_dst_fp()
- small_src_key_as_string, small_src_key = self.make_small_key()
- small_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- # Ensure downloaded object has correct content.
- self.assertEqual(SMALL_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
-
- def test_non_retryable_exception_handling(self):
- """
- Tests resumable download that fails with a non-retryable exception
- """
- harness = CallbackTestHarness(
- exception=OSError(errno.EACCES, 'Permission denied'))
- res_download_handler = ResumableDownloadHandler(num_retries=1)
- dst_fp = self.make_dst_fp()
- small_src_key_as_string, small_src_key = self.make_small_key()
- try:
- small_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- self.fail('Did not get expected OSError')
- except OSError, e:
- # Ensure the error was re-raised.
- self.assertEqual(e.errno, 13)
-
- def test_failed_and_restarted_download_with_persistent_tracker(self):
- """
- Tests resumable download that fails once and then completes,
- with tracker file
- """
- harness = CallbackTestHarness()
- tmpdir = self._MakeTempDir()
- tracker_file_name = self.make_tracker_file(tmpdir)
- dst_fp = self.make_dst_fp(tmpdir)
- small_src_key_as_string, small_src_key = self.make_small_key()
- res_download_handler = ResumableDownloadHandler(
- tracker_file_name=tracker_file_name, num_retries=1)
- small_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- # Ensure downloaded object has correct content.
- self.assertEqual(SMALL_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
- # Ensure tracker file deleted.
- self.assertFalse(os.path.exists(tracker_file_name))
-
- def test_multiple_in_process_failures_then_succeed(self):
- """
- Tests resumable download that fails twice in one process, then completes
- """
- res_download_handler = ResumableDownloadHandler(num_retries=3)
- dst_fp = self.make_dst_fp()
- small_src_key_as_string, small_src_key = self.make_small_key()
- small_src_key.get_contents_to_file(
- dst_fp, res_download_handler=res_download_handler)
- # Ensure downloaded object has correct content.
- self.assertEqual(SMALL_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
-
- def test_multiple_in_process_failures_then_succeed_with_tracker_file(self):
- """
- Tests resumable download that fails completely in one process,
- then when restarted completes, using a tracker file
- """
- # Set up test harness that causes more failures than a single
- # ResumableDownloadHandler instance will handle, writing enough data
- # before the first failure that some of it survives that process run.
- harness = CallbackTestHarness(
- fail_after_n_bytes=LARGE_KEY_SIZE/2, num_times_to_fail=2)
- larger_src_key_as_string = os.urandom(LARGE_KEY_SIZE)
- larger_src_key = self._MakeKey(data=larger_src_key_as_string)
- tmpdir = self._MakeTempDir()
- tracker_file_name = self.make_tracker_file(tmpdir)
- dst_fp = self.make_dst_fp(tmpdir)
- res_download_handler = ResumableDownloadHandler(
- tracker_file_name=tracker_file_name, num_retries=0)
- try:
- larger_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- self.fail('Did not get expected ResumableDownloadException')
- except ResumableDownloadException, e:
- self.assertEqual(e.disposition,
- ResumableTransferDisposition.ABORT_CUR_PROCESS)
- # Ensure a tracker file survived.
- self.assertTrue(os.path.exists(tracker_file_name))
- # Try it one more time; this time should succeed.
- larger_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- self.assertEqual(LARGE_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(larger_src_key_as_string,
- larger_src_key.get_contents_as_string())
- self.assertFalse(os.path.exists(tracker_file_name))
- # Ensure some of the file was downloaded both before and after failure.
- self.assertTrue(
- len(harness.transferred_seq_before_first_failure) > 1 and
- len(harness.transferred_seq_after_first_failure) > 1)
-
- def test_download_with_inital_partial_download_before_failure(self):
- """
- Tests resumable download that successfully downloads some content
- before it fails, then restarts and completes
- """
- # Set up harness to fail download after several hundred KB so download
- # server will have saved something before we retry.
- harness = CallbackTestHarness(
- fail_after_n_bytes=LARGE_KEY_SIZE/2)
- larger_src_key_as_string = os.urandom(LARGE_KEY_SIZE)
- larger_src_key = self._MakeKey(data=larger_src_key_as_string)
- res_download_handler = ResumableDownloadHandler(num_retries=1)
- dst_fp = self.make_dst_fp()
- larger_src_key.get_contents_to_file(
- dst_fp, cb=harness.call,
- res_download_handler=res_download_handler)
- # Ensure downloaded object has correct content.
- self.assertEqual(LARGE_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(larger_src_key_as_string,
- larger_src_key.get_contents_as_string())
- # Ensure some of the file was downloaded both before and after failure.
- self.assertTrue(
- len(harness.transferred_seq_before_first_failure) > 1 and
- len(harness.transferred_seq_after_first_failure) > 1)
-
- def test_zero_length_object_download(self):
- """
- Tests downloading a zero-length object (exercises boundary conditions).
- """
- res_download_handler = ResumableDownloadHandler()
- dst_fp = self.make_dst_fp()
- k = self._MakeKey()
- k.get_contents_to_file(dst_fp,
- res_download_handler=res_download_handler)
- self.assertEqual(0, get_cur_file_size(dst_fp))
-
- def test_download_with_invalid_tracker_etag(self):
- """
- Tests resumable download with a tracker file containing an invalid etag
- """
- tmp_dir = self._MakeTempDir()
- dst_fp = self.make_dst_fp(tmp_dir)
- small_src_key_as_string, small_src_key = self.make_small_key()
- invalid_etag_tracker_file_name = os.path.join(tmp_dir,
- 'invalid_etag_tracker')
- f = open(invalid_etag_tracker_file_name, 'w')
- f.write('3.14159\n')
- f.close()
- res_download_handler = ResumableDownloadHandler(
- tracker_file_name=invalid_etag_tracker_file_name)
- # An error should be printed about the invalid tracker, but then it
- # should run the update successfully.
- small_src_key.get_contents_to_file(
- dst_fp, res_download_handler=res_download_handler)
- self.assertEqual(SMALL_KEY_SIZE, get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
-
- def test_download_with_inconsistent_etag_in_tracker(self):
- """
- Tests resumable download with an inconsistent etag in tracker file
- """
- tmp_dir = self._MakeTempDir()
- dst_fp = self.make_dst_fp(tmp_dir)
- small_src_key_as_string, small_src_key = self.make_small_key()
- inconsistent_etag_tracker_file_name = os.path.join(tmp_dir,
- 'inconsistent_etag_tracker')
- f = open(inconsistent_etag_tracker_file_name, 'w')
- good_etag = small_src_key.etag.strip('"\'')
- new_val_as_list = []
- for c in reversed(good_etag):
- new_val_as_list.append(c)
- f.write('%s\n' % ''.join(new_val_as_list))
- f.close()
- res_download_handler = ResumableDownloadHandler(
- tracker_file_name=inconsistent_etag_tracker_file_name)
- # An error should be printed about the expired tracker, but then it
- # should run the update successfully.
- small_src_key.get_contents_to_file(
- dst_fp, res_download_handler=res_download_handler)
- self.assertEqual(SMALL_KEY_SIZE,
- get_cur_file_size(dst_fp))
- self.assertEqual(small_src_key_as_string,
- small_src_key.get_contents_as_string())
-
- def test_download_with_unwritable_tracker_file(self):
- """
- Tests resumable download with an unwritable tracker file
- """
- # Make dir where tracker_file lives temporarily unwritable.
- tmp_dir = self._MakeTempDir()
- tracker_file_name = os.path.join(tmp_dir, 'tracker')
- save_mod = os.stat(tmp_dir).st_mode
- try:
- os.chmod(tmp_dir, 0)
- res_download_handler = ResumableDownloadHandler(
- tracker_file_name=tracker_file_name)
- except ResumableDownloadException, e:
- self.assertEqual(e.disposition, ResumableTransferDisposition.ABORT)
- self.assertNotEqual(
- e.message.find('Couldn\'t write URI tracker file'), -1)
- finally:
- # Restore original protection of dir where tracker_file lives.
- os.chmod(tmp_dir, save_mod)

Powered by Google App Engine
This is Rietveld 408576698