#825: Allow passing explicit connection to '_BlobIterator.__init__'.

gcloud/storage/bucket.py

@@ -63,12 +63,18 @@ class _BlobIterator(Iterator):
 
     :type extra_params: dict or None
     :param extra_params: Extra query string parameters for the API call.
+
+    :type connection: :class:`gcloud.storage.connection.Connection`
+    :param connection: The connection to use when sending requests.  Defaults
+                       to the bucket's connection
     """
-    def __init__(self, bucket, extra_params=None):
+    def __init__(self, bucket, extra_params=None, connection=None):
+        if connection is None:
+            connection = bucket.connection
         self.bucket = bucket
         self.prefixes = ()
         super(_BlobIterator, self).__init__(
-            connection=bucket.connection, path=bucket.path + '/o',
+            connection=connection, path=bucket.path + '/o',
             extra_params=extra_params)
 
     def get_items_from_response(self, response):

gcloud/storage/test_bucket.py

@@ -37,6 +37,17 @@ def test_ctor(self):
         self.assertEqual(iterator.next_page_token, None)
         self.assertEqual(iterator.prefixes, ())
 
+    def test_ctor_w_explicit_connection(self):
+        connection = _Connection()
+        bucket = _Bucket(None)
+        iterator = self._makeOne(bucket, connection=connection)
+        self.assertTrue(iterator.bucket is bucket)
+        self.assertTrue(iterator.connection is connection)
+        self.assertEqual(iterator.path, '%s/o' % bucket.path)
+        self.assertEqual(iterator.page_number, 0)
+        self.assertEqual(iterator.next_page_token, None)
+        self.assertEqual(iterator.prefixes, ())
+
     def test_get_items_from_response_empty(self):
         connection = _Connection()
         bucket = _Bucket(connection)