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

telemetry/telemetry/page/shared_page_state.py

 # Copyright 2014 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 logging
 import os
 import sys
 
 from telemetry.core import exceptions
 from telemetry.core import util
 from telemetry import decorators
 from telemetry.internal.browser import browser_finder
 from telemetry.internal.browser import browser_finder_exceptions
 from telemetry.internal.browser import browser_info as browser_info_module
 from telemetry.internal.platform.profiler import profiler_finder
 from telemetry.internal.util import exception_formatter
 from telemetry.internal.util import file_handle
 from telemetry.page import cache_temperature
+from telemetry.page import traffic_setting
 from telemetry.page import legacy_page_test
 from telemetry import story
 from telemetry.util import screenshot
 from telemetry.util import wpr_modes
 from telemetry.web_perf import timeline_based_measurement
 
 
 def _PrepareFinderOptions(finder_options, test, device_type):
   browser_options = finder_options.browser_options
   # Set up user agent.
@@ skipping 210 matching lines @@
       # navigation has begun and RenderFrameHostManager::DidNavigateMainFrame()
       # will cancel the next navigation because it's pending. This manifests as
       # the first navigation in a PageSet freezing indefinitely because the
       # navigation was silently canceled when |self.browser.tabs[0]| was
       # committed. Only do this when we just started the browser, otherwise
       # there are cases where previous pages in a PageSet never complete
       # loading so we'll wait forever.
       if started_browser:
         self.browser.tabs[0].WaitForDocumentReadyStateToBeComplete()
 
+    # Reset traffic shaping to speed up cache temperature setup.
+    self.platform.network_controller.UpdateTrafficSettings(0, 0, 0)
     cache_temperature.EnsurePageCacheTemperature(
         self._current_page, self.browser, self._previous_page)
+    if self._current_page.traffic_setting != traffic_setting.NONE:
+      # TODO(kouhei): fill in appropriate traffic shaping data.
+      raise NotImplementedError()
 
     # Start profiling if needed.
     if self._finder_options.profiler:
       self._StartProfiling(self._current_page)
 
   def CanRunStory(self, page):
     return self.CanRunOnBrowser(browser_info_module.BrowserInfo(self.browser),
                                 page)
 
   def CanRunOnBrowser(self, browser_info,
@@ skipping 87 matching lines @@
 class SharedDesktopPageState(SharedPageState):
   _device_type = 'desktop'
 
 
 class SharedTabletPageState(SharedPageState):
   _device_type = 'tablet'
 
 
 class Shared10InchTabletPageState(SharedPageState):
   _device_type = 'tablet_10_inch'