Update TestStorageListFiles regression to use 'bucket.iterator'.

tseaver · tseaver · commit faf859566afd · 2014-11-11T19:07:07.000-05:00
See: #359 (comment)
diff --git a/regression/storage.py b/regression/storage.py
@@ -190,12 +190,11 @@ def test_list_files(self):
 
     def test_paginate_files(self):
         truncation_size = 1
-        extra_params = {'maxResults': len(self.FILENAMES) - truncation_size}
-        iterator = storage.key._KeyIterator(bucket=self.bucket,
-                                            extra_params=extra_params)
+        count = len(self.FILENAMES) - truncation_size
+        iterator = self.bucket.iterator(max_results=count)
         response = iterator.get_next_page_response()
         keys = list(iterator.get_items_from_response(response))
-        self.assertEqual(len(keys), extra_params['maxResults'])
+        self.assertEqual(len(keys), count)
         self.assertEqual(iterator.page_number, 1)
         self.assertTrue(iterator.next_page_token is not None)
 
