Update checked in version of gsutil to version 4.6

gslib/tests/test_Doption.py

+# -*- coding: utf-8 -*-
 # Copyright 2013 Google Inc. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+"""Integration tests for gsutil -D option."""
 
-import boto
+from __future__ import absolute_import
+
 import gslib
+from gslib.cs_api_map import ApiSelector
 import gslib.tests.testcase as testcase
+from gslib.tests.testcase.integration_testcase import SkipForS3
 from gslib.tests.util import ObjectToURI as suri
+from gslib.tests.util import SetBotoConfigForTest
+from gslib.util import ONE_KB
 
 
-class TestCat(testcase.GsUtilIntegrationTestCase):
+@SkipForS3('-D output is implementation-specific.')
+class TestDOption(testcase.GsUtilIntegrationTestCase):
   """Integration tests for gsutil -D option."""
 
+  def test_minus_D_multipart_upload(self):
+    """Tests that debug option does not output upload media body."""
+    # We want to ensure it works with and without a trailing newline.
+    for file_contents in ('a1b2c3d4', 'a1b2c3d4\n'):
+      fpath = self.CreateTempFile(contents=file_contents)
+      bucket_uri = self.CreateBucket()
+      with SetBotoConfigForTest(
+          [('GSUtil', 'resumable_threshold', str(ONE_KB))]):
+        stderr = self.RunGsUtil(
+            ['-D', 'cp', fpath, suri(bucket_uri)], return_stderr=True)
+        print 'command line:' + ' '.join(['-D', 'cp', fpath, suri(bucket_uri)])
+        if self.test_api == ApiSelector.JSON:
+          self.assertIn('media body', stderr)
+        self.assertNotIn('a1b2c3d4', stderr)
+        self.assertIn('Comparing local vs cloud md5-checksum for', stderr)
+        self.assertIn('total_bytes_transferred: %d' % len(file_contents),
+                      stderr)
+
+  def test_minus_D_resumable_upload(self):
+    fpath = self.CreateTempFile(contents='a1b2c3d4')
+    bucket_uri = self.CreateBucket()
+    with SetBotoConfigForTest([('GSUtil', 'resumable_threshold', '4')]):
+      stderr = self.RunGsUtil(
+          ['-D', 'cp', fpath, suri(bucket_uri)], return_stderr=True)
+      self.assertNotIn('a1b2c3d4', stderr)
+      self.assertIn('Comparing local vs cloud md5-checksum for', stderr)
+      self.assertIn('total_bytes_transferred: 8', stderr)
+
   def test_minus_D_cat(self):
+    """Tests cat command with debug option."""
     key_uri = self.CreateObject(contents='0123456789')
-    (stdout, stderr) = self.RunGsUtil(['-D', 'cat', suri(key_uri)],
-                                      return_stdout=True, return_stderr=True)
+    with SetBotoConfigForTest([('Boto', 'proxy_pass', 'secret')]):
+      (stdout, stderr) = self.RunGsUtil(
+          ['-D', 'cat', suri(key_uri)], return_stdout=True, return_stderr=True)
     self.assertIn('You are running gsutil with debug output enabled.', stderr)
-    self.assertIn('config: [', stderr)
-    self.assertRegexpMatches(
-        stderr, '.*HEAD /%s/%s.*Content-Length: 0.*User-Agent: .*gsutil/%s' %
-        (key_uri.bucket_name, key_uri.object_name, gslib.VERSION))
+    self.assertIn("reply: 'HTTP/1.1 200 OK", stderr)
+    self.assertIn('config:', stderr)
+    self.assertIn("('proxy_pass', 'REDACTED')", stderr)
     self.assertIn("reply: 'HTTP/1.1 200 OK", stderr)
     self.assertIn('header: Expires: ', stderr)
     self.assertIn('header: Date: ', stderr)
-    self.assertIn('header: Cache-Control: private, max-age=0',
-                  stderr)
-    self.assertIn('header: Last-Modified: ', stderr)
-    self.assertIn('header: ETag: "781e5e245d69b566979b86e28d23f2c7"', stderr)
-    self.assertIn('header: x-goog-generation: ', stderr)
-    self.assertIn('header: x-goog-metageneration: 1', stderr)
     self.assertIn('header: Content-Type: application/octet-stream', stderr)
     self.assertIn('header: Content-Length: 10', stderr)
-    self.assertIn('header: x-goog-hash: crc32c=KAwGng==', stderr)
-    self.assertIn('header: x-goog-hash: md5=eB5eJF1ptWaXm4bijSPyxw==', stderr)
-    self.assertIn('gsutil version %s' % gslib.VERSION, stdout)
+
+    if self.test_api == ApiSelector.XML:
+      self.assertRegexpMatches(
+          stderr, '.*HEAD /%s/%s.*Content-Length: 0.*User-Agent: .*gsutil/%s' %
+          (key_uri.bucket_name, key_uri.object_name, gslib.VERSION))
+
+      self.assertIn('header: Cache-Control: private, max-age=0',
+                    stderr)
+      self.assertIn('header: Last-Modified: ', stderr)
+      self.assertIn('header: ETag: "781e5e245d69b566979b86e28d23f2c7"', stderr)
+      self.assertIn('header: x-goog-generation: ', stderr)
+      self.assertIn('header: x-goog-metageneration: 1', stderr)
+      self.assertIn('header: x-goog-hash: crc32c=KAwGng==', stderr)
+      self.assertIn('header: x-goog-hash: md5=eB5eJF1ptWaXm4bijSPyxw==', stderr)
+    elif self.test_api == ApiSelector.JSON:
+      self.assertRegexpMatches(
+          stderr, '.*GET.*b/%s/o/%s.*user-agent:.*gsutil/%s' %
+          (key_uri.bucket_name, key_uri.object_name, gslib.VERSION))
+      self.assertIn(('header: Cache-Control: no-cache, no-store, max-age=0, '
+                     'must-revalidate'), stderr)
+      self.assertIn("md5Hash: u'eB5eJF1ptWaXm4bijSPyxw=='", stderr)
+
     if gslib.IS_PACKAGE_INSTALL:
       self.assertIn('PACKAGED_GSUTIL_INSTALLS_DO_NOT_HAVE_CHECKSUMS', stdout)
     else:
-      self.assertRegexpMatches(stdout, r'.*checksum [0-9a-f]{32}.*')
-    self.assertIn('boto version ', stdout)
-    self.assertIn('python version ', stdout)
+      self.assertRegexpMatches(stdout, r'.*checksum: [0-9a-f]{32}.*')
+    self.assertIn('gsutil version: %s' % gslib.VERSION, stdout)
+    self.assertIn('boto version: ', stdout)
+    self.assertIn('python version: ', stdout)
+    self.assertIn('OS: ', stdout)
+    self.assertIn('multiprocessing available: ', stdout)
+    self.assertIn('using cloud sdk: ', stdout)
     self.assertIn('config path: ', stdout)
     self.assertIn('gsutil path: ', stdout)
     self.assertIn('compiled crcmod: ', stdout)
     self.assertIn('installed via package manager: ', stdout)
     self.assertIn('editable install: ', stdout)