[Pywebsocket PerformanceTests 2/2] Add blink_perf.pywebsocket

tools/telemetry/telemetry/core/browser.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 os
 
 from telemetry import decorators
 from telemetry.core import app
 from telemetry.core import browser_credentials
 from telemetry.core import exceptions
 from telemetry.core import extension_dict
 from telemetry.core import local_server
 from telemetry.core import memory_cache_http_server
+from telemetry.core import pywebsocket_server
 from telemetry.core import tab_list
 from telemetry.core.backends import browser_backend
 
 
 class Browser(app.App):
   """A running browser instance that can be controlled in a limited way.
 
   To create a browser instance, use browser_finder.FindBrowser.
 
   Be sure to clean up after yourself by calling Close() when you are done with
@@ skipping 194 matching lines @@
 
     self._local_server_controller.Close()
     self._browser_backend.Close()
     self.credentials = None
 
   @property
   def http_server(self):
     return self._local_server_controller.GetRunningServer(
         memory_cache_http_server.MemoryCacheHTTPServer, None)
 
+  @property
+  def pywebsocket_server(self):
+    return self._local_server_controller.GetRunningServer(
+        pywebsocket_server.PywebsocketServer, None)
+
   def SetHTTPServerDirectories(self, paths):
     """Returns True if the HTTP server was started, False otherwise."""
     if isinstance(paths, basestring):
       paths = set([paths])
     paths = set(os.path.realpath(p) for p in paths)
 
     # If any path is in a subdirectory of another, remove the subdirectory.
     duplicates = set()
     for parent_path in paths:
       for sub_path in paths:
         if parent_path == sub_path:
           continue
         if os.path.commonprefix((parent_path, sub_path)) == parent_path:
           duplicates.add(sub_path)
     paths -= duplicates
 
     if self.http_server:
       if paths and self.http_server.paths == paths:
         return False
 
       self.http_server.Close()
 
     if not paths:
       return False
 
     server = memory_cache_http_server.MemoryCacheHTTPServer(paths)
     self.StartLocalServer(server)
     return True
 
+  def StartPywebsocketServer(self):
+    if self.pywebsocket_server:
+      return False
+
+    server = pywebsocket_server.PywebsocketServer()
+    self.StartLocalServer(server)
+    return True
+
   def StartLocalServer(self, server):
     """Starts a LocalServer and associates it with this browser.
 
     It will be closed when the browser closes.
     """
     self._local_server_controller.StartServer(server)
 
   @property
   def local_servers(self):
     """Returns the currently running local servers."""
     return self._local_server_controller.local_servers
 
   def GetStandardOutput(self):
     return self._browser_backend.GetStandardOutput()
 
   def GetStackTrace(self):
     return self._browser_backend.GetStackTrace()
 
   @property
   def supports_system_info(self):
     return self._browser_backend.supports_system_info
 
   def GetSystemInfo(self):
     """Returns low-level information about the system, if available.
 
        See the documentation of the SystemInfo class for more details."""
     return self._browser_backend.GetSystemInfo()