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

Unified Diff: tests/download_from_google_storage_unittests.py

Issue 1209033006: Revert of Add support for tar.gz archive files to download from download_from_google_storage (Closed) Base URL: http://src.chromium.org/svn/trunk/tools/depot_tools/
Patch Set: Created 5 years, 6 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
« no previous file with comments | « download_from_google_storage.py ('k') | tests/upload_to_google_storage_unittests.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: tests/download_from_google_storage_unittests.py
diff --git a/tests/download_from_google_storage_unittests.py b/tests/download_from_google_storage_unittests.py
index 0420ad94289def64e0d00836994eddce6cd79047..a8af63b0c74cbf9913c396103b8c04a3a0f230c8 100755
--- a/tests/download_from_google_storage_unittests.py
+++ b/tests/download_from_google_storage_unittests.py
@@ -11,7 +11,6 @@
import Queue
import shutil
import sys
-import tarfile
import tempfile
import threading
import unittest
@@ -60,21 +59,6 @@
return (0, '', '')
-class ChangedWorkingDirectory(object):
- def __init__(self, working_directory):
- self._old_cwd = ''
- self._working_directory = working_directory
-
- def __enter__(self):
- self._old_cwd = os.getcwd()
- print "Enter directory = ", self._working_directory
- os.chdir(self._working_directory)
-
- def __exit__(self, *_):
- print "Enter directory = ", self._old_cwd
- os.chdir(self._old_cwd)
-
-
class GstoolsUnitTests(unittest.TestCase):
def setUp(self):
self.temp_dir = tempfile.mkdtemp(prefix='gstools_test')
@@ -83,49 +67,6 @@
def cleanUp(self):
shutil.rmtree(self.temp_dir)
-
- def test_validate_tar_file(self):
- lorem_ipsum = os.path.join(self.base_path, 'lorem_ipsum.txt')
- with ChangedWorkingDirectory(self.base_path):
- # Sanity ok check.
- tar_dir = 'ok_dir'
- os.makedirs(os.path.join(self.base_path, tar_dir))
- tar = 'good.tar.gz'
- lorem_ipsum_copy = os.path.join(tar_dir, 'lorem_ipsum.txt')
- shutil.copyfile(lorem_ipsum, lorem_ipsum_copy)
- with tarfile.open(tar, 'w:gz') as tar:
- tar.add(lorem_ipsum_copy)
- self.assertTrue(
- download_from_google_storage._validate_tar_file(tar, tar_dir))
-
- # Test no links.
- tar_dir_link = 'for_tar_link'
- os.makedirs(tar_dir_link)
- link = os.path.join(tar_dir_link, 'link')
- os.symlink(lorem_ipsum, link)
- tar_with_links = 'with_links.tar.gz'
- with tarfile.open(tar_with_links, 'w:gz') as tar:
- tar.add(link)
- self.assertFalse(
- download_from_google_storage._validate_tar_file(tar, tar_dir_link))
-
- # Test not outside.
- tar_dir_outside = 'outside_tar'
- os.makedirs(tar_dir_outside)
- tar_with_outside = 'with_outside.tar.gz'
- with tarfile.open(tar_with_outside, 'w:gz') as tar:
- tar.add(lorem_ipsum)
- self.assertFalse(
- download_from_google_storage._validate_tar_file(tar,
- tar_dir_outside))
- # Test no ..
- tar_with_dotdot = 'with_dotdot.tar.gz'
- dotdot_file = os.path.join(tar_dir, '..', tar_dir, 'lorem_ipsum.txt')
- with tarfile.open(tar_with_dotdot, 'w:gz') as tar:
- tar.add(dotdot_file)
- self.assertFalse(
- download_from_google_storage._validate_tar_file(tar,
- tar_dir))
def test_gsutil(self):
gsutil = download_from_google_storage.Gsutil(GSUTIL_DEFAULT_PATH, None)
@@ -223,7 +164,7 @@
stdout_queue = Queue.Queue()
download_from_google_storage._downloader_worker_thread(
0, self.queue, False, self.base_url, self.gsutil,
- stdout_queue, self.ret_codes, True, False)
+ stdout_queue, self.ret_codes, True)
expected_calls = [
('check_call',
('ls', input_filename)),
@@ -249,52 +190,12 @@
stdout_queue = Queue.Queue()
download_from_google_storage._downloader_worker_thread(
0, self.queue, False, self.base_url, self.gsutil,
- stdout_queue, self.ret_codes, True, False)
+ stdout_queue, self.ret_codes, True)
expected_output = [
'0> File %s exists and SHA1 matches. Skipping.' % output_filename
]
self.assertEqual(list(stdout_queue.queue), expected_output)
self.assertEqual(self.gsutil.history, [])
-
- def test_download_extract_archive(self):
- # By design we make this not match
- sha1_hash = '61223e1ad3d86901a57629fee38313db5ec106ff'
- input_filename = '%s/%s' % (self.base_url, sha1_hash)
- # Generate a gzipped tarfile
- output_filename = os.path.join(self.base_path, 'subfolder.tar.gz')
- output_dirname = os.path.join(self.base_path, 'subfolder')
- extracted_filename = os.path.join(output_dirname, 'subfolder_text.txt')
- with tarfile.open(output_filename, 'w:gz') as tar:
- tar.add(output_dirname, arcname='subfolder')
- shutil.rmtree(output_dirname)
- print(output_dirname)
- self.queue.put((sha1_hash, output_filename))
- self.queue.put((None, None))
- stdout_queue = Queue.Queue()
- download_from_google_storage._downloader_worker_thread(
- 0, self.queue, False, self.base_url, self.gsutil,
- stdout_queue, self.ret_codes, True, True, delete=False)
- expected_calls = [
- ('check_call',
- ('ls', input_filename)),
- ('check_call',
- ('cp', input_filename, output_filename))]
- if sys.platform != 'win32':
- expected_calls.append(
- ('check_call',
- ('stat',
- 'gs://sometesturl/61223e1ad3d86901a57629fee38313db5ec106ff')))
- expected_output = [
- '0> Downloading %s...' % output_filename]
- expected_output.extend([
- '0> Extracting 3 entries from %s to %s' % (output_filename,
- output_dirname)])
- expected_ret_codes = []
- self.assertEqual(list(stdout_queue.queue), expected_output)
- self.assertEqual(self.gsutil.history, expected_calls)
- self.assertEqual(list(self.ret_codes.queue), expected_ret_codes)
- self.assertTrue(os.path.exists(output_dirname))
- self.assertTrue(os.path.exists(extracted_filename))
def test_download_worker_skips_not_found_file(self):
sha1_hash = '7871c8e24da15bad8b0be2c36edc9dc77e37727f'
@@ -306,7 +207,7 @@
self.gsutil.add_expected(1, '', '') # Return error when 'ls' is called.
download_from_google_storage._downloader_worker_thread(
0, self.queue, False, self.base_url, self.gsutil,
- stdout_queue, self.ret_codes, True, False)
+ stdout_queue, self.ret_codes, True)
expected_output = [
'0> Failed to fetch file %s for %s, skipping. [Err: ]' % (
input_filename, output_filename),
@@ -341,8 +242,7 @@
ignore_errors=False,
sha1_file=False,
verbose=True,
- auto_platform=False,
- extract=False)
+ auto_platform=False)
expected_calls = [
('check_call',
('ls', input_filename)),
@@ -373,8 +273,7 @@
ignore_errors=False,
sha1_file=False,
verbose=True,
- auto_platform=False,
- extract=False)
+ auto_platform=False)
expected_calls = [
('check_call',
('ls', input_filename)),
« no previous file with comments | « download_from_google_storage.py ('k') | tests/upload_to_google_storage_unittests.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698