[Pywebsocket PerformanceTests 2/2] Add blink_perf.pywebsocket

tools/perf/benchmarks/blink_perf.py

 # Copyright 2013 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
 
 from telemetry import benchmark
 from telemetry import page as page_module
 from telemetry.core import util
 from telemetry.page import page_set
@@ skipping 80 matching lines @@
       values = [float(v.replace(',', '')) for v in parts[1:-1]]
       units = parts[-1]
       metric = page.display_name.split('.')[0].replace('/', '_')
       results.AddValue(list_of_scalar_values.ListOfScalarValues(
           results.current_page, metric, units, values))
 
       break
 
     print log
 
+class _BlinkPerfWithPywebsocketMeasurement(_BlinkPerfMeasurement):
+  """Runs a Blink performance test with pywebsocket server."""
+  run_pywebsocket_server = True
+
+  def __init__(self):
+    super(_BlinkPerfWithPywebsocketMeasurement, self).__init__()
+    self.needs_pywebsocket_server = True
+
+  def CustomizeBrowserOptions(self, options):
+    super(_BlinkPerfWithPywebsocketMeasurement, self).CustomizeBrowserOptions(
+        options)
+    # Cross-origin accesses are needed to run benchmarks spanning two servers,
+    # the Telemetry's HTTP server and the pywebsocket server.
+    options.AppendExtraBrowserArgs([
+        '--disable-web-security'
+    ])
+
 
 class _BlinkPerfFullFrameMeasurement(_BlinkPerfMeasurement):
   def __init__(self):
     super(_BlinkPerfFullFrameMeasurement, self).__init__()
     self._blink_perf_js += '\nwindow.fullFrameMeasurement = true;'
 
   def CustomizeBrowserOptions(self, options):
     super(_BlinkPerfFullFrameMeasurement, self).CustomizeBrowserOptions(
         options)
     # Full layout measurement needs content_shell with internals testing API.
@@ skipping 114 matching lines @@
 
 # This benchmark is for local testing, doesn't need to run on bots.
 @benchmark.Disabled()
 class BlinkPerfXMLHttpRequest(benchmark.Benchmark):
   tag = 'xml_http_request'
   test = _BlinkPerfMeasurement
 
   def CreatePageSet(self, options):
     path = os.path.join(BLINK_PERF_BASE_DIR, 'XMLHttpRequest')
     return CreatePageSetFromPath(path, SKIPPED_FILE)
+
+
+class BlinkPerfPywebsocket(benchmark.Benchmark):
+  tag = 'pywebsocket'
+  test = _BlinkPerfWithPywebsocketMeasurement
+
+  def CreatePageSet(self, options):
+    path = os.path.join(BLINK_PERF_BASE_DIR, 'Pywebsocket')
+    return CreatePageSetFromPath(path, SKIPPED_FILE)