Add netconfigs for different network type

telemetry/telemetry/page/page_run_end_to_end_unittest.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 os
 import re
 import shutil
 import sys
 import StringIO
 import tempfile
+import time
 import unittest
 
 from telemetry import benchmark
 from telemetry import story
 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 user_agent
 from telemetry.internal.results import results_options
 from telemetry.internal import story_runner
 from telemetry.internal.testing.test_page_sets import example_domain
 from telemetry.internal.util import exception_formatter
 from telemetry.page import page as page_module
 from telemetry.page import legacy_page_test
 from telemetry.page import shared_page_state
+from telemetry.page import traffic_setting as traffic_setting_module
 from telemetry.util import image_util
 from telemetry.testing import fakes
 from telemetry.testing import options_for_unittests
 from telemetry.testing import system_stub
 
 
 SIMPLE_CREDENTIALS_STRING = """
 {
   "test": {
     "username": "example",
@@ skipping 276 matching lines @@
         assert len(tab.browser.tabs) == 1
 
     test = TestOneTab()
     options = options_for_unittests.GetCopy()
     options.output_formats = ['none']
     options.suppress_gtest_report = True
     SetUpStoryRunnerArguments(options)
     results = results_options.CreateResults(EmptyMetadataForTest(), options)
     story_runner.Run(test, story_set, options, results)
 
+  def testTrafficSettings(self):
+    story_set = story.StorySet()
+    slow_page = page_module.Page(
+        'file://green_rect.html', story_set, base_dir=util.GetUnittestDataDir(),
+        name='slow',
+        traffic_setting=traffic_setting_module.REGULAR_2G)
+    fast_page = page_module.Page(
+        'file://green_rect.html', story_set, base_dir=util.GetUnittestDataDir(),
+        name='fast',
+        traffic_setting=traffic_setting_module.WIFI)
+    story_set.AddStory(slow_page)
+    story_set.AddStory(fast_page)
+
+    latencies_by_page_in_ms = {}
+
+    class MeasureLatency(legacy_page_test.LegacyPageTest):
+      def __init__(self):
+        super(MeasureLatency, self).__init__()
+        self._will_navigate_time = None
+
+      def WillNavigateToPage(self, page, tab):
+        del page, tab # unused
+        self._will_navigate_time = time.time() * 1000
+
+      def ValidateAndMeasurePage(self, page, tab, results):
+        del results  # unused
+        latencies_by_page_in_ms[page.name] = (
+            time.time() * 1000 - self._will_navigate_time)
+
+    test = MeasureLatency()
+    options = options_for_unittests.GetCopy()
+    options.output_formats = ['none']
+    options.suppress_gtest_report = True
+    SetUpStoryRunnerArguments(options)
+    results = results_options.CreateResults(EmptyMetadataForTest(), options)
+    story_runner.Run(test, story_set, options, results)
+    # Slow page should be slower than fast page by at least 300 ms (roundtrip
+    # time of 2G) - 2 ms (roundtrip time of Wifi)
+    self.assertGreater(latencies_by_page_in_ms['slow'],
+                       latencies_by_page_in_ms['fast'] + 300 - 2)
+
   # Ensure that story_runner allows >1 tab for multi-tab test.
   @decorators.Enabled('has tabs')
   def testMultipleTabsOkayForMultiTabTest(self):
     story_set = story.StorySet()
     page = page_module.Page(
         'file://blank.html', story_set, base_dir=util.GetUnittestDataDir())
     story_set.AddStory(page)
 
     class TestMultiTabs(legacy_page_test.LegacyPageTest):
       def TabForPage(self, page, browser):
@@ skipping 396 matching lines @@
                   results.pages_to_profiling_files)
     screenshot_file_path = (
         results.pages_to_profiling_files[failing_page][0].GetAbsPath())
     try:
       actual_screenshot_img = image_util.FromPngFile(screenshot_file_path)
       self.assertTrue(image_util.AreEqual(
                       image_util.FromBase64Png(expected_png_base64),
                       actual_screenshot_img))
     finally:  # Must clean up screenshot file if exists.
       os.remove(screenshot_file_path)