[Telemetry] Add API for updating ts_proxy_server's traffic settings

telemetry/telemetry/page/__init__.py

 # Copyright 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 import inspect
 import logging
 import os
 import urlparse
 
 from py_utils import cloud_storage  # pylint: disable=import-error
 
 from telemetry import story
+from telemetry.page import cache_temperature as cache_temperature_module
 from telemetry.page import shared_page_state
-from telemetry.page import cache_temperature as cache_temperature_module
+from telemetry.page import traffic_setting as traffic_setting_module
 from telemetry.internal.actions import action_runner as action_runner_module
 
 
 class Page(story.Story):
 
   def __init__(self, url, page_set=None, base_dir=None, name='',
                credentials_path=None,
                credentials_bucket=cloud_storage.PUBLIC_BUCKET, labels=None,
                startup_url='', make_javascript_deterministic=True,
                shared_page_state_class=shared_page_state.SharedPageState,
                grouping_keys=None,
-               cache_temperature=cache_temperature_module.ANY):
+               cache_temperature=cache_temperature_module.ANY,
+               traffic_setting=traffic_setting_module.NONE):
     self._url = url
 
     super(Page, self).__init__(
         shared_page_state_class, name=name, labels=labels,
         is_local=self._scheme in ['file', 'chrome', 'about'],
         make_javascript_deterministic=make_javascript_deterministic,
         grouping_keys=grouping_keys)
 
     self._page_set = page_set
     # Default value of base_dir is the directory of the file that defines the
     # class of this page instance.
     if base_dir is None:
       base_dir = os.path.dirname(inspect.getfile(self.__class__))
     self._base_dir = base_dir
     self._name = name
     if credentials_path:
       credentials_path = os.path.join(self._base_dir, credentials_path)
       cloud_storage.GetIfChanged(credentials_path, credentials_bucket)
       if not os.path.exists(credentials_path):
         logging.error('Invalid credentials path: %s' % credentials_path)
         credentials_path = None
     self._credentials_path = credentials_path
     self._cache_temperature = cache_temperature
     if cache_temperature != cache_temperature_module.ANY:
       self.grouping_keys['cache_temperature'] = cache_temperature
 
+    self._traffic_setting = traffic_setting
+
     # Whether to collect garbage on the page before navigating & performing
     # page actions.
     self._collect_garbage_before_run = True
 
     # These attributes can be set dynamically by the page.
     self.synthetic_delays = dict()
     self._startup_url = startup_url
     self.credentials = None
     self.skip_waits = False
     self.script_to_evaluate_on_commit = None
     self._SchemeErrorCheck()
 
   @property
   def credentials_path(self):
     return self._credentials_path
 
   @property
   def cache_temperature(self):
     return self._cache_temperature
 
   @property
+  def traffic_setting(self):
+    return self._traffic_setting
+
+  @property
   def startup_url(self):
     return self._startup_url
 
   def _SchemeErrorCheck(self):
     if not self._scheme:
       raise ValueError('Must prepend the URL with scheme (e.g. file://)')
 
     if self.startup_url:
       startup_url_scheme = urlparse.urlparse(self.startup_url).scheme
       if not startup_url_scheme:
@@ skipping 127 matching lines @@
 
   @property
   def display_name(self):
     if self.name:
       return self.name
     if self.page_set is None or not self.is_file:
       return self.url
     all_urls = [p.url.rstrip('/') for p in self.page_set if p.is_file]
     common_prefix = os.path.dirname(os.path.commonprefix(all_urls))
     return self.url[len(common_prefix):].strip('/')