Update checked in version of gsutil to version 4.6

third_party/boto/tests/unit/s3/test_connection.py

 # Copyright (c) 2013 Amazon.com, Inc. or its affiliates.  All Rights Reserved
 #
 # 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.
 #
+import mock
+import time
+
 from tests.unit import unittest
 from tests.unit import AWSMockServiceTestCase
+from tests.unit import MockServiceWithConfigTestCase
 
-from boto.s3.connection import S3Connection
+from boto.s3.connection import S3Connection, HostRequiredError
+from boto.s3.connection import S3ResponseError, Bucket
 
 
 class TestSignatureAlteration(AWSMockServiceTestCase):
     connection_class = S3Connection
 
     def test_unchanged(self):
         self.assertEqual(
             self.service_connection._required_auth_capability(),
             ['s3']
         )
 
     def test_switched(self):
         conn = self.connection_class(
             aws_access_key_id='less',
             aws_secret_access_key='more',
             host='s3.cn-north-1.amazonaws.com.cn'
         )
         self.assertEqual(
             conn._required_auth_capability(),
             ['hmac-v4-s3']
         )
 
 
+class TestSigV4HostError(MockServiceWithConfigTestCase):
+    connection_class = S3Connection
+
+    def test_historical_behavior(self):
+        self.assertEqual(
+            self.service_connection._required_auth_capability(),
+            ['s3']
+        )
+        self.assertEqual(self.service_connection.host, 's3.amazonaws.com')
+
+    def test_sigv4_opt_in(self):
+        # Switch it at the config, so we can check to see how the host is
+        # handled.
+        self.config = {
+            's3': {
+                'use-sigv4': True,
+            }
+        }
+
+        with self.assertRaises(HostRequiredError):
+            # No host+SigV4 == KABOOM
+            self.connection_class(
+                aws_access_key_id='less',
+                aws_secret_access_key='more'
+            )
+
+        # Ensure passing a ``host`` still works.
+        conn = self.connection_class(
+            aws_access_key_id='less',
+            aws_secret_access_key='more',
+            host='s3.cn-north-1.amazonaws.com.cn'
+        )
+        self.assertEqual(
+            conn._required_auth_capability(),
+            ['hmac-v4-s3']
+        )
+        self.assertEqual(
+            conn.host,
+            's3.cn-north-1.amazonaws.com.cn'
+        )
+
+
+class TestSigV4Presigned(MockServiceWithConfigTestCase):
+    connection_class = S3Connection
+
+    def test_sigv4_presign(self):
+        self.config = {
+            's3': {
+                'use-sigv4': True,
+            }
+        }
+
+        conn = self.connection_class(
+            aws_access_key_id='less',
+            aws_secret_access_key='more',
+            host='s3.amazonaws.com'
+        )
+
+        # Here we force an input iso_date to ensure we always get the
+        # same signature.
+        url = conn.generate_url_sigv4(86400, 'GET', bucket='examplebucket',
+            key='test.txt', iso_date='20140625T000000Z')
+
+        self.assertIn('a937f5fbc125d98ac8f04c49e0204ea1526a7b8ca058000a54c192457be05b7d', url)
+
+    def test_sigv4_presign_optional_params(self):
+        self.config = {
+            's3': {
+                'use-sigv4': True,
+            }
+        }
+
+        conn = self.connection_class(
+            aws_access_key_id='less',
+            aws_secret_access_key='more',
+            security_token='token',
+            host='s3.amazonaws.com'
+        )
+
+        url = conn.generate_url_sigv4(86400, 'GET', bucket='examplebucket',
+            key='test.txt', version_id=2)
+
+        self.assertIn('VersionId=2', url)
+        self.assertIn('X-Amz-Security-Token=token', url)
+
+
 class TestUnicodeCallingFormat(AWSMockServiceTestCase):
     connection_class = S3Connection
 
     def default_body(self):
         return """<?xml version="1.0" encoding="UTF-8"?>
 <ListAllMyBucketsResult xmlns="http://doc.s3.amazonaws.com/2006-03-01">
   <Owner>
     <ID>bcaf1ffd86f461ca5fb16fd081034f</ID>
     <DisplayName>webfile</DisplayName>
   </Owner>
@@ skipping 12 matching lines @@
     def create_service_connection(self, **kwargs):
         kwargs['calling_format'] = u'boto.s3.connection.OrdinaryCallingFormat'
         return super(TestUnicodeCallingFormat,
                      self).create_service_connection(**kwargs)
 
     def test_unicode_calling_format(self):
         self.set_http_response(status_code=200)
         self.service_connection.get_all_buckets()
 
 
+class TestHeadBucket(AWSMockServiceTestCase):
+    connection_class = S3Connection
+
+    def default_body(self):
+        # HEAD requests always have an empty body.
+        return ""
+
+    def test_head_bucket_success(self):
+        self.set_http_response(status_code=200)
+        buck = self.service_connection.head_bucket('my-test-bucket')
+        self.assertTrue(isinstance(buck, Bucket))
+        self.assertEqual(buck.name, 'my-test-bucket')
+
+    def test_head_bucket_forbidden(self):
+        self.set_http_response(status_code=403)
+
+        with self.assertRaises(S3ResponseError) as cm:
+            self.service_connection.head_bucket('cant-touch-this')
+
+        err = cm.exception
+        self.assertEqual(err.status, 403)
+        self.assertEqual(err.error_code, 'AccessDenied')
+        self.assertEqual(err.message, 'Access Denied')
+
+    def test_head_bucket_notfound(self):
+        self.set_http_response(status_code=404)
+
+        with self.assertRaises(S3ResponseError) as cm:
+            self.service_connection.head_bucket('totally-doesnt-exist')
+
+        err = cm.exception
+        self.assertEqual(err.status, 404)
+        self.assertEqual(err.error_code, 'NoSuchBucket')
+        self.assertEqual(err.message, 'The specified bucket does not exist')
+
+    def test_head_bucket_other(self):
+        self.set_http_response(status_code=405)
+
+        with self.assertRaises(S3ResponseError) as cm:
+            self.service_connection.head_bucket('you-broke-it')
+
+        err = cm.exception
+        self.assertEqual(err.status, 405)
+        # We don't have special-cases for this error status.
+        self.assertEqual(err.error_code, None)
+        self.assertEqual(err.message, '')
+
+
 if __name__ == "__main__":
     unittest.main()