Adding test that covers explicit content type argument

cyluss · cyluss · commit 73c285c08d62 · 2016-02-16T10:28:38.000+09:00
diff --git a/gcloud/storage/test_blob.py b/gcloud/storage/test_blob.py
@@ -156,6 +156,38 @@ def _basic_generate_signed_url_helper(self, credentials=None):
     def test_generate_signed_url_w_default_method(self):
         self._basic_generate_signed_url_helper()
 
+    def test_generate_signed_url_w_content_type(self):
+        from gcloud._testing import _Monkey
+        from gcloud.storage import blob as MUT
+
+        BLOB_NAME = 'parent/child'
+        EXPIRATION = '2014-10-16T20:34:37.000Z'
+        connection = _Connection()
+        client = _Client(connection)
+        bucket = _Bucket(client)
+        blob = self._makeOne(BLOB_NAME, bucket=bucket)
+        URI = ('http://example.com/abucket/a-blob-name?Signature=DEADBEEF'
+               '&Expiration=2014-10-16T20:34:37.000Z')
+
+        SIGNER = _Signer()
+        with _Monkey(MUT, generate_signed_url=SIGNER):
+            signed_url = blob.generate_signed_url(EXPIRATION,
+                                                  content_type="text/html")
+            self.assertEqual(signed_url, URI)
+
+        EXPECTED_ARGS = (_Connection.credentials,)
+        EXPECTED_KWARGS = {
+            'api_access_endpoint': 'https://storage.googleapis.com',
+            'expiration': EXPIRATION,
+            'method': 'GET',
+            'resource': '/name/parent%2Fchild',
+            'content_type': "text/html",
+            'response_type': None,
+            'response_disposition': None,
+            'generation': None,
+        }
+        self.assertEqual(SIGNER._signed, [(EXPECTED_ARGS, EXPECTED_KWARGS)])
+
     def test_generate_signed_url_w_credentials(self):
         credentials = object()
         self._basic_generate_signed_url_helper(credentials=credentials)
