Index: telemetry/telemetry/page/__init__.py |
diff --git a/telemetry/telemetry/page/__init__.py b/telemetry/telemetry/page/__init__.py |
index 64de29b0b9b0fa8f229e5abcbd4a0d8cb7e51670..516bf2507b29b665d60730b0c991c666c38967c6 100644 |
--- a/telemetry/telemetry/page/__init__.py |
+++ b/telemetry/telemetry/page/__init__.py |
@@ -9,8 +9,9 @@ import urlparse |
from py_utils import cloud_storage # pylint: disable=import-error |
from telemetry import story |
-from telemetry.page import shared_page_state |
from telemetry.page import cache_temperature as cache_temperature_module |
+from telemetry.page import shared_page_state |
+from telemetry.page import traffic_setting as traffic_setting_module |
from telemetry.internal.actions import action_runner as action_runner_module |
@@ -22,7 +23,8 @@ class Page(story.Story): |
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__( |
@@ -49,6 +51,8 @@ class Page(story.Story): |
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 |
@@ -70,6 +74,10 @@ class Page(story.Story): |
return self._cache_temperature |
@property |
+ def traffic_setting(self): |
+ return self._traffic_setting |
+ |
+ @property |
def startup_url(self): |
return self._startup_url |