Add optional cloud bucket permission property to page and page_set

tools/telemetry/telemetry/page/page_set_unittest.py

Index: tools/telemetry/telemetry/page/page_set_unittest.py
diff --git a/tools/telemetry/telemetry/page/page_set_unittest.py b/tools/telemetry/telemetry/page/page_set_unittest.py
index 444aa03f763d7037e71dbfe83f13dc3e7446c7f0..0f863edcafdb4b6ccf485251298be30c76fa72d8 100644
--- a/tools/telemetry/telemetry/page/page_set_unittest.py
+++ b/tools/telemetry/telemetry/page/page_set_unittest.py
@@ -7,9 +7,9 @@ import tempfile
 import unittest
 
 from telemetry.core import util
+from telemetry.page import cloud_storage
 from telemetry.page import page_set
 
-
 class TestPageSet(unittest.TestCase):
 
   def testServingDirs(self):
@@ -50,6 +50,7 @@ class TestPageSet(unittest.TestCase):
     self.assertEqual('data/credential', pps.credentials_path)
     self.assertEqual('desktop', pps.user_agent_type)
     self.assertEqual(test_pps_dir, pps.file_path)
+    self.assertEqual(page_set.PUBLIC_BUCKET, pps.privacy_bucket)
     self.assertEqual(3, len(pps.pages))
     google_page = pps.pages[0]
     self.assertEqual('https://www.google.com', google_page.url)
@@ -78,3 +79,27 @@ class TestPageSet(unittest.TestCase):
     self.assertEqual(
       os.path.normpath(os.path.join(
         util.GetUnittestDataDir(), 'pages/foo.html')), external_page.file_path)
+
+  def testCloudBucket(self):
+    blank_ps = page_set.PageSet()
+    expected_privacy_bucket = None
+    self.assertEqual(blank_ps.privacy_bucket, expected_privacy_bucket)
+
+    public_ps = page_set.PageSet(
+                         privacy_bucket=page_set.PUBLIC_BUCKET)
+    expected_privacy_bucket = cloud_storage.PUBLIC_BUCKET
+    self.assertEqual(public_ps.privacy_bucket, expected_privacy_bucket)
+
+    partner_ps = page_set.PageSet(
+                          privacy_bucket=page_set.PARTNER_BUCKET)
+    expected_privacy_bucket = cloud_storage.PARTNER_BUCKET
+    self.assertEqual(partner_ps.privacy_bucket, expected_privacy_bucket)
+
+    internal_ps = page_set.PageSet(
+                           privacy_bucket=page_set.INTERNAL_BUCKET)
+    expected_privacy_bucket = cloud_storage.INTERNAL_BUCKET
+    self.assertEqual(internal_ps.privacy_bucket, expected_privacy_bucket)
+
+    self.assertRaises(ValueError,
+        page_set.PageSet,
+        privacy_bucket='garbage_bucket')